added error AirportNotFoundError

python/FlightRadar24/__init__.py

@@ -13,7 +13,7 @@
 """
 
 __author__ = "Jean Loui Bernard Silva de Jesus"
-__version__ = "1.3.15"
+__version__ = "1.3.16"
 
 from .api import FlightRadar24API, FlightTrackerConfig
 from .entities import Airport, Entity, Flight

python/FlightRadar24/api.py

@@ -8,7 +8,7 @@
 from .core import Core
 from .entities.airport import Airport
 from .entities.flight import Flight
-from .errors import LoginError
+from .errors import AirportNotFoundError, LoginError
 from .request import APIRequest
 
 
@@ -88,7 +88,12 @@ def get_airport(self, code: str) -> Airport:
         :param code: ICAO or IATA of the airport
         """
         response = APIRequest(Core.airport_data_url.format(code), headers = Core.json_headers)
-        return Airport(details=response.get_content()["details"])
+        content = response.get_content()
+
+        if not content or not isinstance(content, dict) or not content.get("details"):
+            raise AirportNotFoundError(f"Could not find an airport by the code '{code}'.");
+
+        return Airport(details=content["details"])
 
     def get_airport_details(self, code: str, flight_limit: int = 100, page: int = 1) -> Dict:
         """

python/FlightRadar24/errors.py

@@ -1,5 +1,9 @@
 # -*- coding: utf-8 -*-
 
+class AirportNotFoundError(Exception):
+    pass
+
+
 class CloudflareError(Exception):
     def __init__(self, message, response):
         self.message = message