expose rightmove query parameters as cli options
This commit is contained in:
parent
96562c0895
commit
bb9afc76fe
2 changed files with 70 additions and 15 deletions
|
|
@ -41,6 +41,30 @@ def cli(ctx, data_dir: str):
|
|||
|
||||
|
||||
@cli.command()
|
||||
@click.option(
|
||||
'--min-bedrooms',
|
||||
default=1,
|
||||
help='Minimum number of bedrooms',
|
||||
type=click.IntRange(min=1),
|
||||
)
|
||||
@click.option(
|
||||
'--max-bedrooms',
|
||||
default=5,
|
||||
help='Maximum number of bedrooms',
|
||||
type=click.IntRange(min=1),
|
||||
)
|
||||
@click.option(
|
||||
'--min-price',
|
||||
default=0,
|
||||
help='Minimum price',
|
||||
type=click.IntRange(min=0),
|
||||
)
|
||||
@click.option(
|
||||
'--max-price',
|
||||
default=1000000,
|
||||
help='Maximum price',
|
||||
type=click.IntRange(min=0),
|
||||
)
|
||||
@click.option(
|
||||
'--district',
|
||||
default=None,
|
||||
|
|
@ -49,13 +73,28 @@ def cli(ctx, data_dir: str):
|
|||
multiple=True,
|
||||
)
|
||||
@click.pass_context
|
||||
def dump_listings(ctx: click.core.Context, district: list[str]):
|
||||
def dump_listings(
|
||||
ctx: click.core.Context,
|
||||
district: list[str],
|
||||
min_bedrooms: int,
|
||||
max_bedrooms: int,
|
||||
min_price: int,
|
||||
max_price: int,
|
||||
):
|
||||
data_dir: str = ctx.obj['data_dir']
|
||||
query_parameters = dump_listings_module.QueryParameters(
|
||||
district_names=set(district),
|
||||
min_bedrooms=min_bedrooms,
|
||||
max_bedrooms=max_bedrooms,
|
||||
min_price=min_price,
|
||||
max_price=max_price,
|
||||
)
|
||||
click.echo(
|
||||
f'Running dump_listings for districts {district} and data dir {data_dir}'
|
||||
f'Running dump_listings for districts {district}, data dir {data_dir} and parameters: '
|
||||
f'{query_parameters}'
|
||||
)
|
||||
data_dir_path = pathlib.Path(data_dir)
|
||||
dump_listings_module.dump_listings(set(district), data_dir_path)
|
||||
dump_listings_module.dump_listings(query_parameters, data_dir_path)
|
||||
|
||||
|
||||
@cli.command()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue