import requests
import gzip
import os
import shutil
import xml.dom.minidom

import sys
sys.path.append("./py2lib")
sys.path.append("./Util")
sys.path.append("./records")
import bit
import MachineProductCfg as MPC
import LFRecord as LFR

locationIds = []
zipCodes = []
airports = []

for i in MPC.getPrimaryLocations():
    locationIds.append(LFR.getCoopId(i))
    zipCodes.append(LFR.getZip(i))

airports = MPC.getAirportCodes()
print(airports)

apiKey = '21d8a80b3d6b444998a80b3d6b1449d3'

def getData(airport):
    url = f"https://api.weather.com/v1/airportcode/{airport}/airport/delays.xml?language=en-US&apiKey={apiKey}"

    res = requests.get(url=url)

    data = res.text
    newData = data[48:-11]

    # Write to i2doc file
    i2Doc = f'<AirportDelays id="000000000" locationKey="{airport}" isWxScan="0">' + '' + newData + f'<clientKey>{airport}</clientKey></AirportDelays>' 
    print(f"[AIRPORT DELAYS] Writing airport delay data for {airport}")

    f = open("D:\\AirportDelays.i2m", 'a')
    f.write(i2Doc)
    f.close()

def writeData():
    useData = False
    airportsWithDelays = []

    for x in airports:
        # Do a quick check to see if the airport in question has a delay or not
        res = requests.get(f"https://api.weather.com/v1/airportcode/{x}/airport/delays.xml?language=en-US&apiKey={apiKey}")

        if (res.status_code != 200):
            print(f"[AIRPORT DELAYS] No delays for {x} found, skipping..")
        else:
            airportsWithDelays.append(x)
            print(f"[AIRPORT DELAYS] {x} has a delay! Writing a file..")
            useData = True

    if (useData):
        header = '<Data type="AirportDelays">'
        footer = "</Data>"

        with open("D:\\AirportDelays.i2m", 'w') as doc:
            doc.write(header)

        for x in airportsWithDelays:
            getData(x)

        with open("D:\\AirportDelays.i2m", 'a') as end:
            end.write(footer)

        dom = xml.dom.minidom.parse("D:\\AirportDelays.i2m")
        prettyXml = dom.toprettyxml(indent="  ")

        with open("D:\\AirportDelays.i2m", 'w') as g:
            g.write(prettyXml)
            g.close()

        files = []
        commands = []
        with open("D:\\AirportDelays.i2m", 'rb') as f_in:
            with gzip.open("D:\\AirportDelays.gz", 'wb') as f_out:
                shutil.copyfileobj(f_in, f_out)

        gZipFile = "D:\\AirportDelays.gz"

        files.append(gZipFile)
        comand = commands.append('<MSG><Exec workRequest="storeData(File={0},QGROUP=__AirportDelays__,Feed=AirportDelays)" /><GzipCompressedMsg fname="AirportDelays" /></MSG>')
        numFiles = len(files)

        bit.sendFile(files, commands, numFiles, 0)

        os.remove("D:\\AirportDelays.i2m")
        os.remove("D:\\AirportDelays.gz")
    else:
        print("[AIRPORT DELAYS] Not writing AirportDelays -- Either no delays found, or the API is broken.")