-
Notifications
You must be signed in to change notification settings - Fork 0
/
nearby_stops.py
76 lines (55 loc) · 2.08 KB
/
nearby_stops.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import requests
import math
from datetime import datetime
import os
from dotenv import load_dotenv
from geopy.geocoders import Nominatim
# Load environment variables from .env file
load_dotenv()
API_KEY = os.environ.get("API_KEY")
MAX_DISTANCE = os.environ.get("RADIUS")
# calling the Nominatim tool and create Nominatim class
loc = Nominatim(user_agent="Geopy Library")
# entering the location name
getLoc = loc.geocode(input("Enter location: "))
# printing address
print(getLoc.address)
LAT = getLoc.latitude
LON = getLoc.longitude
currentTime = datetime.now().timestamp()
localTime = datetime.now()
print(f"\nDepartures for {localTime}")
payload = {
'lat' : LAT,
'lon' : LON,
'max_distance' : MAX_DISTANCE
}
header = {"apiKey" : API_KEY}
nearby_stops = requests.get(
"https://external.transitapp.com/v3/public/nearby_stops", params=payload, headers=header
)
for x in nearby_stops.json()['stops']:
stopId = [x][0]['global_stop_id']
stopName = [x][0]['stop_name']
stopCode = [x][0]['stop_code']
stopInfo = {
'global_stop_id' : stopId
}
departures = requests.get(
"https://external.transitapp.com/v3/public/stop_departures", params=stopInfo, headers=header
)
print("\n" + stopName + " [" + stopCode + "] " + stopId)
for i in range(3):
headsign = departures.json()['route_departures'][0]['itineraries'][0]['headsign'] # destination
route_short_name = departures.json()['route_departures'][0]['route_short_name'] # route name or number
# add extra detail if iON
if route_short_name == "301":
route_short_name = "301 iON"
currentDeparture = math.floor((departures.json()['route_departures'][0]['itineraries'][0]['schedule_items'][i]['departure_time']) - currentTime)
timeUnit = "min"
if currentDeparture > 60:
currentDeparture = math.floor(currentDeparture/60)
else:
timeUnit ="s"
print(f"{route_short_name} to {headsign}\t\t{currentDeparture} {timeUnit}")
print("\n=========================================================")