put_user_data.put_data()
# THIS VARIABLE CONTAIN IATA CODE OF OUR CURRENT CITY FROM WHERE WE WANT TO TRAVEL ITS BOM NOW for MUMBAI
ORIGIN_CITY_IATA = "BOM"
# WHEN WE 1st crate sheet there will be no iata code for cities so if its not their then this block will execute
if return_data[0]["iataCode"] == "":
# WE WILL FEATCH get_flight_data function form flightsearch class to get iata code for each city
from_flight_search = FlightSearch()
for data in return_data:
data["iataCode"] = from_flight_search.get_flight_data(data["city"])
# after execution of this we are replacing our data_manager.data variable with return data from above function
# with iata code for each country
data_manager.data = return_data
# AFTER THAT CALLING put_data function from data_manager for storing data into sheet
data_manager.put_data()
# Getting todays date and price for next 6 month using timedelta function
tomorrow = datetime.now() + timedelta(days=1)
next_6_month = datetime.now() + timedelta(days=(6 * 30))
# This will pass each city iata and time to search flight_fuction
for city in return_data:
flight = flight_search.search_flight(
ORIGIN_CITY_IATA,
city["iataCode"],
from_time=tomorrow,
to_time=next_6_month
#This file will need to use the DataManager,FlightSearch, FlightData, NotificationManager classes to achieve the program requirements.
from data_manager import DataManager
from flight_search import FlightSearch
from pprint import pprint
data_manager = DataManager()
sheet_data=data_manager.get_data()
flight_search=FlightSearch()
for data in sheet_data["prices"]:
code=flight_search.get_iata_code(data["city"])
data["iataCode"]=code
id=data["id"]
body={
"price":{
"iataCode": code
}
}
data_manager.update(id,body)
data_manager.data=sheet_data
print(data_manager.data)
