예제 #1
0
from notification_manager import NotificationManager

sheety_data = DataManager()
search = FlightSearch()
notif = NotificationManager()
flight_data = FlightData()

# search for IATA codes
cities = flight_data.list_cities()
city_codes = search.get_codes(cities)

# # write city codes into google sheet
# sheety_data.put_data(city_codes)

# find cheapest flights for all cities
data = flight_data.data(city_codes)

min_prices = data["prices"]
out_date = data["out date"]
return_date = data["return date"]

# find original price of flights
orig_prices = flight_data.orig_prices()

# if min price < orig price send alert
alert_city_num = [
    num for num in range(len(min_prices)) if min_prices[num] < orig_prices[num]
]
email_data = sheety_data.email_data()["users"]
emails = [da["email"] for da in email_data]