Fixed up record generators + new ones from Goldblaze

2025-06-10 05:30:00 -05:00 · 2022-10-13 01:34:54 -07:00
parent 532fd39d40
commit 6537d392ac
7 changed files with 422 additions and 99 deletions
--- a/recordGenerators/AirportDelays.py
+++ b/recordGenerators/AirportDelays.py
@ -0,0 +1,96 @@
+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.")