100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
import requests
|
|
from rec.utils import nextMonday
|
|
from collections import defaultdict
|
|
|
|
API_KEY = 'AIzaSyBoBHzeQFgR7O-NlNsuHXQcC1B7ccEHpl8'
|
|
url = "https://routes.googleapis.com/directions/v2:computeRoutes"
|
|
|
|
|
|
def transit_route(origin_lat:float, origin_lon:float, dest_lat:float, dest_lon:float, compute_alternative_routes=True):
|
|
monday9am = nextMonday()
|
|
|
|
header = {
|
|
"X-Goog-Api-Key": API_KEY,
|
|
"Content-Type": "application/json",
|
|
"X-Goog-FieldMask": "routes.distanceMeters,routes.duration,routes.staticDuration,routes.legs.steps.distanceMeters,routes.legs.steps.staticDuration,routes.legs.steps.travelMode",
|
|
}
|
|
|
|
body = {
|
|
"origin":{
|
|
"location":{
|
|
"latLng":{
|
|
"latitude": origin_lat,
|
|
"longitude": origin_lon
|
|
}
|
|
}
|
|
},
|
|
"destination":{
|
|
"location":{
|
|
"latLng":{
|
|
"latitude": dest_lat,
|
|
"longitude": dest_lon
|
|
}
|
|
}
|
|
},
|
|
"travelMode": "TRANSIT",
|
|
# "2023-10-15T15:01:23.045123456Z"
|
|
"departureTime": monday9am.strftime("%Y-%m-%dT%H:%M:%S.%fZ"),
|
|
"computeAlternativeRoutes": compute_alternative_routes,
|
|
# "routeModifiers": {
|
|
# "avoidTolls": false,
|
|
# "avoidHighways": false,
|
|
# "avoidFerries": false
|
|
# },
|
|
"languageCode": "en-US",
|
|
"units": "METRIC"
|
|
}
|
|
|
|
r = requests.post(url, json=body, headers=header)
|
|
if r.status_code == 200:
|
|
return r.json()
|
|
|
|
raise Exception(r.json())
|
|
|
|
def extract_time(d, limit:int=2):
|
|
res = []
|
|
for route in d['routes']:
|
|
distance = route['distanceMeters']
|
|
duration = int(route['duration'].strip('s'))
|
|
duration_static = int(route['staticDuration'].strip('s'))
|
|
|
|
steps = route['legs'][0]['steps']
|
|
duration_per_transit = defaultdict(lambda: 0)
|
|
distance_per_transit = defaultdict(lambda: 0)
|
|
number_of_transit_stops = 0
|
|
|
|
for step in steps:
|
|
duration_per_transit[step['travelMode']] += int(step['staticDuration'].strip('s'))
|
|
distance_per_transit[step['travelMode']] += step.get('distanceMeters', 0)
|
|
if step['travelMode'] == 'TRANSIT':
|
|
number_of_transit_stops += 1
|
|
|
|
res.append({
|
|
'duration': duration,
|
|
'distance': distance,
|
|
'duration_static': duration_static,
|
|
'duration_per_transit': dict(duration_per_transit),
|
|
'distance_per_transit': dict(distance_per_transit),
|
|
'number_of_transit_stops': number_of_transit_stops,
|
|
})
|
|
|
|
return res[:limit]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import json
|
|
with open('code/json/routing_routeapi.json', 'r') as f:
|
|
d = json.load(f)
|
|
|
|
extract_time(d)
|
|
|
|
|
|
|
|
# if __name__ == "__main__":
|
|
# origin = 51.5635664310333, -0.1107173751570373 # home
|
|
# dest = 51.50475678313417, 0.04915321000190009 # london city airport
|
|
# d = travel_time(origin[0], origin[1], dest[0], dest[1])
|
|
# import json
|
|
# with open('code/json/routing_routeapi.json', 'w') as f:
|
|
# json.dump(d, f)
|
|
|