diff --git a/recordGenerators/AirportDelays.py b/recordGenerators/AirportDelays.py
index 4edae1c..7da8b04 100644
--- a/recordGenerators/AirportDelays.py
+++ b/recordGenerators/AirportDelays.py
@@ -4,6 +4,7 @@ import os
import shutil
import xml.dom.minidom
import logging,coloredlogs
+import aiohttp, aiofiles
import sys
sys.path.append("./py2lib")
@@ -29,56 +30,56 @@ l.debug(airports)
apiKey = '21d8a80b3d6b444998a80b3d6b1449d3'
-def getData(airport):
+async def getData(airport):
url = f"https://api.weather.com/v1/airportcode/{airport}/airport/delays.xml?language=en-US&apiKey={apiKey}"
+ data = ""
- res = requests.get(url=url)
+ async with aiohttp.ClientSession() as s:
+ async with s.get(url) as r:
+ data = await r.text()
- data = res.text
newData = data[48:-11]
# Write to i2doc file
i2Doc = f'' + '' + newData + f'{airport}'
- f = open("./.temp/AirportDelays.i2m", 'a')
- f.write(i2Doc)
- f.close()
+ async with aiofiles.open("./.temp/AirportDelays.i2m", 'a') as f:
+ await f.write(i2Doc)
+ await f.close()
-def writeData():
+async 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):
- l.debug(f"[AIRPORT DELAYS] No delays for {x} found, skipping..")
- else:
- airportsWithDelays.append(x)
- l.debug(f"[AIRPORT DELAYS] {x} has a delay! Writing a file..")
- useData = True
+ async with aiohttp.ClientSession() as s:
+ async with s.get(f"https://api.weather.com/v1/airportcode/{x}/airport/delays.xml?language=en-US&apiKey={apiKey}") as r:
+ if r.status != 200:
+ l.debug(f"No delay for {x} found, skipping..")
+ else:
+ airportsWithDelays.append(x)
+ useData = True
if (useData):
l.info("Writing an AirportDelays record.")
header = ''
footer = ""
- with open("./.temp/AirportDelays.i2m", 'w') as doc:
- doc.write(header)
+ async with aiofiles.open("./.temp/AirportDelays.i2m", 'w') as doc:
+ await doc.write(header)
for x in airportsWithDelays:
- getData(x)
+ await getData(x)
- with open("./.temp/AirportDelays.i2m", 'a') as end:
- end.write(footer)
+ async with aiofiles.open("./.temp/AirportDelays.i2m", 'a') as end:
+ await end.write(footer)
dom = xml.dom.minidom.parse("./.temp/AirportDelays.i2m")
prettyXml = dom.toprettyxml(indent=" ")
- with open("./.temp/AirportDelays.i2m", 'w') as g:
- g.write(prettyXml)
- g.close()
+ async with aiofiles.open("./.temp/AirportDelays.i2m", 'w') as g:
+ await g.write(prettyXml)
+ await g.close()
files = []
commands = []