Adding initial walking time and identifier to the information

This commit is contained in:
Kadir 2024-03-25 20:47:31 +00:00
parent ce632c795d
commit 47f7b2b672
2 changed files with 8 additions and 1 deletions

View file

@ -11,7 +11,6 @@ _DATA_DIR = pathlib.Path('data/rs/')
class Listing(): class Listing():
identifier: int identifier: int
_cached: Dict = None _cached: Dict = None
@staticmethod @staticmethod
def get_all_listings() -> List['Listing']: def get_all_listings() -> List['Listing']:
@ -187,6 +186,7 @@ class Listing():
def dict_nicely(self): def dict_nicely(self):
return { return {
'identifier': self.identifier,
'sqm_ocr': self.sqm_ocr, 'sqm_ocr': self.sqm_ocr,
'price': self.price, 'price': self.price,
'price_per_sqm': self.price_per_sqm, 'price_per_sqm': self.price_per_sqm,

View file

@ -59,11 +59,17 @@ def extract_time(d, limit:int=2):
duration_static = int(route['staticDuration'].strip('s')) duration_static = int(route['staticDuration'].strip('s'))
steps = route['legs'][0]['steps'] steps = route['legs'][0]['steps']
initial_walk_duration = 0
used_transit = False
duration_per_transit = defaultdict(lambda: 0) duration_per_transit = defaultdict(lambda: 0)
distance_per_transit = defaultdict(lambda: 0) distance_per_transit = defaultdict(lambda: 0)
number_of_transit_stops = 0 number_of_transit_stops = 0
for step in steps: for step in steps:
if used_transit == False and step['travelMode'] == 'WALK':
initial_walk_duration += int(step['staticDuration'].strip('s'))
else:
used_transit = True
duration_per_transit[step['travelMode']] += int(step['staticDuration'].strip('s')) duration_per_transit[step['travelMode']] += int(step['staticDuration'].strip('s'))
distance_per_transit[step['travelMode']] += step.get('distanceMeters', 0) distance_per_transit[step['travelMode']] += step.get('distanceMeters', 0)
if step['travelMode'] == 'TRANSIT': if step['travelMode'] == 'TRANSIT':
@ -73,6 +79,7 @@ def extract_time(d, limit:int=2):
'duration': duration, 'duration': duration,
'distance': distance, 'distance': distance,
'duration_static': duration_static, 'duration_static': duration_static,
'initial_walk_duration': initial_walk_duration,
'duration_per_transit': dict(duration_per_transit), 'duration_per_transit': dict(duration_per_transit),
'distance_per_transit': dict(distance_per_transit), 'distance_per_transit': dict(distance_per_transit),
'number_of_transit_stops': number_of_transit_stops, 'number_of_transit_stops': number_of_transit_stops,