diff --git a/.gitignore b/.gitignore index af6bfdc..061461e 100644 --- a/.gitignore +++ b/.gitignore @@ -1,6 +1,8 @@ +# PyCharm +.idea .vscode/settings.json pythonenv3.8/ __pycache__ dependencies testing -lookup_route.py \ No newline at end of file +lookup_route.py diff --git a/defAirport.py b/defAirport.py index ed64da8..073ad46 100644 --- a/defAirport.py +++ b/defAirport.py @@ -1,41 +1,45 @@ import csv import math + + def add_airport_region(airport_dict): - #Get full region/state name from iso region name - with open('./dependencies/regions.csv', 'r', encoding='utf-8') as regions_csv: - regions_csv = csv.DictReader(filter(lambda row: row[0]!='#', regions_csv)) - for region in regions_csv: - if region['code'] == airport_dict['iso_region']: - airport_dict['region'] = region['name'] - return airport_dict + # Get full region/state name from iso region name + with open('./dependencies/regions.csv', 'r', encoding='utf-8') as regions_csv: + regions_csv = csv.DictReader(filter(lambda row: row[0] != '#', regions_csv)) + for region in regions_csv: + if region['code'] == airport_dict['iso_region']: + airport_dict['region'] = region['name'] + return airport_dict + + def getClosestAirport(latitude, longitude, allowed_types): - from geopy.distance import geodesic - plane = (latitude, longitude) - with open('./dependencies/airports.csv', 'r', encoding='utf-8') as airport_csv: - airport_csv_reader = csv.DictReader(filter(lambda row: row[0]!='#', airport_csv)) - for airport in airport_csv_reader: - if airport['type'] in allowed_types: - airport_coord = float(airport['latitude_deg']), float(airport['longitude_deg']) - airport_dist = float((geodesic(plane, airport_coord).mi)) - if "closest_airport_dict" not in locals(): - closest_airport_dict = airport - closest_airport_dist = airport_dist - elif airport_dist < closest_airport_dist: - closest_airport_dict = airport - closest_airport_dist = airport_dist - closest_airport_dict['distance_mi'] = closest_airport_dist - #Convert indent key to icao key as its labeled icao in other places not ident - closest_airport_dict['icao'] = closest_airport_dict.pop('gps_code') - closest_airport_dict = add_airport_region(closest_airport_dict) - return closest_airport_dict + from geopy.distance import geodesic + plane = (latitude, longitude) + closest_airport_dict = dict() + with open('./dependencies/airports.csv', 'r', encoding='utf-8') as airport_csv: + airport_csv_reader = csv.DictReader(filter(lambda row: row[0] != '#', airport_csv)) + for airport in airport_csv_reader: + if airport['type'] in allowed_types: + airport_coord = float(airport['latitude_deg']), float(airport['longitude_deg']) + airport_dist = float((geodesic(plane, airport_coord).mi)) + if not closest_airport_dict or airport_dist < closest_airport_dist: + closest_airport_dict = airport + closest_airport_dist = airport_dist + closest_airport_dict['distance_mi'] = closest_airport_dist + # Convert indent key to icao key as its labeled icao in other places not ident + closest_airport_dict['icao'] = closest_airport_dict.pop('gps_code') + closest_airport_dict = add_airport_region(closest_airport_dict) + return closest_airport_dict + + def get_airport_by_icao(icao): - with open('./dependencies/airports.csv', 'r', encoding='utf-8') as airport_csv: - airport_csv_reader = csv.DictReader(filter(lambda row: row[0]!='#', airport_csv)) - for airport in airport_csv_reader: - if airport['gps_code'] == icao: - matching_airport = airport - #Convert indent key to icao key as its labeled icao in other places not ident - matching_airport['icao'] = matching_airport.pop('gps_code') - break - matching_airport = add_airport_region(matching_airport) - return matching_airport \ No newline at end of file + with open('./dependencies/airports.csv', 'r', encoding='utf-8') as airport_csv: + airport_csv_reader = csv.DictReader(filter(lambda row: row[0] != '#', airport_csv)) + for airport in airport_csv_reader: + if airport['gps_code'] == icao: + matching_airport = airport + # Convert indent key to icao key as its labeled icao in other places not ident + matching_airport['icao'] = matching_airport.pop('gps_code') + break + matching_airport = add_airport_region(matching_airport) + return matching_airport