wrongmove/crawler/main.py
Viktor Barzin 90b531f5d9
[1/n] click-ify - add entrypoint for click script and add
1_dump_listings command

run via:
poetry run python main.py --step dump_listings
2025-05-11 18:59:41 +00:00

26 lines
515 B
Python

import click
import importlib
dump_listings_module = importlib.import_module('1_dump_listings')
steps_to_handlers = {
'dump_listings': dump_listings_module.dump_listings,
}
@click.command()
@click.option(
'--step',
default=[],
help='Scraping step to run',
multiple=True,
type=click.Choice(steps_to_handlers.keys())
)
def main(step: list[str]):
for s in step:
click.echo(f'Calling handler for step: {s}')
steps_to_handlers[s]()
if __name__ == '__main__':
main()