RomanNum3ral
/
plane-notify
mirror of https://github.com/RomanNum3ral/plane-notify.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #27 from willnoonan/multi 

Improvements to defAirport.py
This commit is contained in:
Jack Sweeney 2022-03-29 09:55:35 -04:00 committed by GitHub
parent 9e2113c7c0 c8e4b382b7
commit c6aaff10df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 43 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -1,3 +1,5 @@
# PyCharm
.idea
.vscode/settings.json .vscode/settings.json
pythonenv3.8/ pythonenv3.8/
__pycache__ __pycache__

12
defAirport.py
View File

@ -1,5 +1,7 @@
import csv import csv
import math import math
def add_airport_region(airport_dict): def add_airport_region(airport_dict):
# Get full region/state name from iso region name # Get full region/state name from iso region name
with open('./dependencies/regions.csv', 'r', encoding='utf-8') as regions_csv: with open('./dependencies/regions.csv', 'r', encoding='utf-8') as regions_csv:
@ -8,19 +10,19 @@ def add_airport_region(airport_dict):
if region['code'] == airport_dict['iso_region']: if region['code'] == airport_dict['iso_region']:
airport_dict['region'] = region['name'] airport_dict['region'] = region['name']
return airport_dict return airport_dict
def getClosestAirport(latitude, longitude, allowed_types): def getClosestAirport(latitude, longitude, allowed_types):
from geopy.distance import geodesic from geopy.distance import geodesic
plane = (latitude, longitude) plane = (latitude, longitude)
closest_airport_dict = dict()
with open('./dependencies/airports.csv', 'r', encoding='utf-8') as airport_csv: with open('./dependencies/airports.csv', 'r', encoding='utf-8') as airport_csv:
airport_csv_reader = csv.DictReader(filter(lambda row: row[0] != '#', airport_csv)) airport_csv_reader = csv.DictReader(filter(lambda row: row[0] != '#', airport_csv))
for airport in airport_csv_reader: for airport in airport_csv_reader:
if airport['type'] in allowed_types: if airport['type'] in allowed_types:
airport_coord = float(airport['latitude_deg']), float(airport['longitude_deg']) airport_coord = float(airport['latitude_deg']), float(airport['longitude_deg'])
airport_dist = float((geodesic(plane, airport_coord).mi)) airport_dist = float((geodesic(plane, airport_coord).mi))
if "closest_airport_dict" not in locals(): if not closest_airport_dict or airport_dist < closest_airport_dist:
closest_airport_dict = airport
closest_airport_dist = airport_dist
elif airport_dist < closest_airport_dist:
closest_airport_dict = airport closest_airport_dict = airport
closest_airport_dist = airport_dist closest_airport_dist = airport_dist
closest_airport_dict['distance_mi'] = closest_airport_dist closest_airport_dict['distance_mi'] = closest_airport_dist
@ -28,6 +30,8 @@ def getClosestAirport(latitude, longitude, allowed_types):
closest_airport_dict['icao'] = closest_airport_dict.pop('gps_code') closest_airport_dict['icao'] = closest_airport_dict.pop('gps_code')
closest_airport_dict = add_airport_region(closest_airport_dict) closest_airport_dict = add_airport_region(closest_airport_dict)
return closest_airport_dict return closest_airport_dict
def get_airport_by_icao(icao): def get_airport_by_icao(icao):
with open('./dependencies/airports.csv', 'r', encoding='utf-8') as airport_csv: with open('./dependencies/airports.csv', 'r', encoding='utf-8') as airport_csv:
airport_csv_reader = csv.DictReader(filter(lambda row: row[0] != '#', airport_csv)) airport_csv_reader = csv.DictReader(filter(lambda row: row[0] != '#', airport_csv))