airport_database.py

import numpy as np
import math
import pickle


class Airport:  # Not used. Was only used to create the database
    def __init__(self, icao: str, latitude: float, longitude: float, name: str):
        self.icao = icao
        self.latitude = latitude
        self.longitude = longitude
        self.name = name

    def __str__(self):
        print(f'ICAO: {self.icao} ({self.name})\tLatitude: {self.latitude}\tLongitude: {self.longitude}')


class AirportDatabase:
    def __init__(self):
        self.database = {}

    def add_airport(self, icao: str, latitude: float, longitude: float, name: str):
        """Not used by the program. Was only used to create the database"""
        self.database[icao] = Airport(icao, latitude, longitude, name)

    def get_lat_and_long(self, icao: str):
        """Returns the latitude and longitude of an airport"""
        return self.database[icao].latitude, self.database[icao].longitude

    def is_in_list(self, icao: str):
        """Checks if an airport is in the database (airport_database.dat)"""
        if icao in self.database:
            return True
        return False

    def get_track(self, departure: str, arrival: str):
        """Finds the track (angle) between the departure and arrival airports"""
        lat1, lat2, lon1, lon2 = self.database[departure].latitude, self.database[arrival].latitude, self.database[departure].longitude, self.database[arrival].longitude
        delta_lon = lon2 - lon1
        x = math.cos(math.radians(lat2)) * math.sin(math.radians(delta_lon))
        y = math.cos(math.radians(lat1)) * math.sin(math.radians(lat2)) - math.sin(math.radians(lat1)) * math.cos(math.radians(lat2)) * math.cos(math.radians(delta_lon))
        track = np.degrees(np.arctan2(x, y))
        return track


def get_database():
    """Loads the database from the .dat file which holds most world airports"""
    with open('airport_database.dat', 'rb') as f:
        return pickle.load(f)