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)