[2/n] click-ify - add 2_dump_detail command

run with
poetry run python main.py --step dump_detail
This commit is contained in:
Viktor Barzin 2025-05-11 19:02:23 +00:00
parent 90b531f5d9
commit c2196c15c1
No known key found for this signature in database
GPG key ID: 4056458DBDBF8863
2 changed files with 31 additions and 21 deletions

View file

@ -4,12 +4,12 @@ from tqdm import tqdm
from data_access import Listing from data_access import Listing
incremental = True
def dump_detail():
listings = Listing.get_all_listings() incremental = True
filtered_listings = [] listings = Listing.get_all_listings()
for listing in listings: filtered_listings = []
for listing in listings:
# We introduced last_seen later, so not all entries have it. # We introduced last_seen later, so not all entries have it.
# If it doesnt exist then its on the platform anymore. So skip # If it doesnt exist then its on the platform anymore. So skip
last_seen = listing.last_seen last_seen = listing.last_seen
@ -22,10 +22,18 @@ for listing in listings:
if incremental and not listing.path_detail_json().exists(): if incremental and not listing.path_detail_json().exists():
filtered_listings.append(listing) filtered_listings.append(listing)
for listing in tqdm(filtered_listings): for listing in tqdm(filtered_listings):
try: try:
d = detail_query(listing.identifier) d = detail_query(listing.identifier)
with open(listing.path_detail_json(), "w") as f: with open(listing.path_detail_json(), "w") as f:
json.dump(d, f) json.dump(d, f)
except Exception as e: except Exception as e:
print(e) print(e)
def main():
dump_detail()
if __name__ == "__main__":
main()

View file

@ -2,9 +2,11 @@ import click
import importlib import importlib
dump_listings_module = importlib.import_module('1_dump_listings') dump_listings_module = importlib.import_module('1_dump_listings')
dump_detail_module = importlib.import_module('2_dump_detail')
steps_to_handlers = { steps_to_handlers = {
'dump_listings': dump_listings_module.dump_listings, 'dump_listings': dump_listings_module.dump_listings,
'dump_detail': dump_detail_module.dump_detail,
} }