viktor/wrongmove
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor detect floorplan to use model listings

Browse source
This commit is contained in:
Viktor Barzin 2025-06-07 14:30:32 +00:00
parent ba87d07cd2
commit 325823e631
No known key found for this signature in database
GPG key ID: 4056458DBDBF8863
3 changed files with 37 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

8
crawler/main.py
View file

@ -165,7 +165,7 @@ def dump_listings(
@click.pass_context
def dump_images(ctx: click.core.Context):
data_dir = ctx.obj["data_dir"]
click.echo(f"Running dump_images stored in {data_dir}")
click.echo(f"Running dump_images for listings stored in {engine.url}")
repository = ListingRepository(engine=engine)
asyncio.run(dump_images_module.dump_images(repository, image_base_path=data_dir))
@ -174,9 +174,9 @@ def dump_images(ctx: click.core.Context):
@click.pass_context
def detect_floorplan(ctx: click.core.Context):
data_dir = ctx.obj["data_dir"]
click.echo(f"Running detect_floorplan in {data_dir}")
listing_paths = sorted(list(pathlib.Path(data_dir).glob("*/listing.json")))
asyncio.run(detect_floorplan_module.detect_floorplan(listing_paths))
click.echo(f"Running detect_floorplan for listings stored in {engine.url}")
repository = ListingRepository(engine=engine)
asyncio.run(detect_floorplan_module.detect_floorplan(repository))
@cli.command()