fix types and format

This commit is contained in:
Viktor Barzin 2025-05-18 12:27:26 +00:00
parent 91d3237516
commit b873eaf203
No known key found for this signature in database
GPG key ID: 4056458DBDBF8863
8 changed files with 117 additions and 172 deletions

View file

@ -1,4 +1,3 @@
import pathlib
from data_access import Listing
from tqdm import tqdm
from geopy.distance import geodesic
@ -15,30 +14,30 @@ def calculate_route(listing_paths: list[str]):
# reduce listings to everything within 7 miles
filtered_listings = []
for listing in listings:
miles = geodesic(
BROCK_STREET_LAT_LONG, (listing.latitude, listing.longitude)
).miles
miles = geodesic(BROCK_STREET_LAT_LONG,
(listing.latitude, listing.longitude)).miles
if listing.isRemoved:
log.info(f"Removed-Skip: Skipping {listing.identifier} is already removed.")
log.info(f"Removed-Skip: Skipping {listing.identifier} "
"is already removed.")
continue
if miles > 7:
log.info(
f"Miles-Skip: Skipping {listing.identifier} as it is {miles} miles away"
)
log.info(f"Miles-Skip: Skipping {listing.identifier} as it is "
f"{miles} miles away")
continue
if listing.path_routing_json().exists():
log.info(
f"Path-Skip: Skipping {listing.identifier} as path routing already exists"
)
(f"Path-Skip: Skipping {listing.identifier} as path routing "
"already exists"))
continue
if listing.sqm_ocr is None or listing.sqm_ocr < 30 or listing.sqm_ocr > 200:
log.info(
f"Floorplan-Skip: Skipping {listing.identifier} as sqm_ocr is {listing.sqm_ocr}"
)
if (listing.sqm_ocr is None or listing.sqm_ocr < 30
or listing.sqm_ocr > 200):
log.info((f"Floorplan-Skip: Skipping {listing.identifier} as "
f"sqm_ocr is {listing.sqm_ocr}"))
continue
filtered_listings.append(listing)
print(f"Filtered listings from {len(listings)} to {len(filtered_listings)}")
print(
f"Filtered listings from {len(listings)} to {len(filtered_listings)}")
for listing in tqdm(filtered_listings):
lat, long = BROCK_STREET_LAT_LONG
@ -47,12 +46,3 @@ def calculate_route(listing_paths: list[str]):
duration_minutes = traveltime["duration"] / 60.0
tqdm.write(f"{listing.identifier} {duration_minutes}")
def main():
listing_paths = sorted(list(pathlib.Path("data/rs").glob("*/listing.json")))
calculate_route(listing_paths)
if __name__ == "__main__":
main()