expose rightmove query parameters as cli options

This commit is contained in:
Viktor Barzin 2025-05-17 20:40:44 +00:00
parent 96562c0895
commit bb9afc76fe
No known key found for this signature in database
GPG key ID: 4056458DBDBF8863
2 changed files with 70 additions and 15 deletions

View file

@ -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()