update flight_data class

Author: rublaman

flight-deals-start/data_manager.py FlightDealsStart/data_manager.pyflight-deals-start/data_manager.py renamed to FlightDealsStart/data_manager.py

@@ -1,6 +1,4 @@
 import requests
-import os
-from datetime import datetime
 
 
 class DataManager:
@@ -11,7 +9,7 @@ def __init__(self, api_key: str):
             'Authorization': f'Basic {api_key}'
         }
 
-    def get_flights_from_google_sheet(self):
+    def get_flights_from_google_sheet(self) -> dict:
         try:
             response = requests.get(self.__API_URL, headers=self.__HEADERS)
             response.raise_for_status()
@@ -21,5 +19,5 @@ def get_flights_from_google_sheet(self):
             print(f"Error in request: {err}")
 
 
-test = DataManager(api_key="")
+test = DataManager(api_key="dGVzdFJ1YmVuOlBhc3N3MHJk")
 print(test.get_flights_from_google_sheet())

FlightDealsStart/flight_data.py

@@ -0,0 +1,10 @@
+from FlightDealsStart.data_manager import DataManager
+
+class FlightData:
+    #This class is responsible for structuring the flight data.
+
+    def __init__(self, api_key: str):
+        self.__flights = DataManager(api_key).get_flights_from_google_sheet()
+
+    def struct_flights(self):
+        flights = self.__flights["prices"]