34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
import requests
|
|
import datetime
|
|
import os
|
|
import json
|
|
|
|
# Move to script directory in case it was run elsewhere.
|
|
scriptdir = os.path.dirname(os.path.abspath(__file__))
|
|
os.chdir(scriptdir)
|
|
|
|
# Import API key from .env file
|
|
api_key=os.environ['API_KEY']
|
|
|
|
# Get latest URL for each route, source for routes:https://www.transitchicago.com/developers/ttdocs/
|
|
routes= ['Red', 'Blue', 'Brn', 'G', 'Org', 'P', 'Pink', 'Y']
|
|
|
|
for route in routes:
|
|
response = requests.get("https://lapi.transitchicago.com/api/1.0/ttpositions.aspx?key="+api_key+"&rt="+route+"&outputType=JSON")
|
|
|
|
trains = response.json()
|
|
#trains_query_time = datetime.datetime.fromisoformat(trains['ctatt']['tmst'])
|
|
trains_query_time = trains['ctatt']['tmst']
|
|
|
|
outdir = "./output/"+route+"/"
|
|
if not os.path.exists(outdir):
|
|
os.makedirs(outdir)
|
|
|
|
outfile = outdir+trains_query_time+"_location_"+route+".json"
|
|
# Lazy format simplify
|
|
outfile = outfile.replace(":","")
|
|
outfile = outfile.replace("-","")
|
|
|
|
with open(outfile,mode='w') as f:
|
|
json.dump(trains, f)
|