main.py

import os
import json
import time
from datetime import datetime, timedelta
from dotenv import load_dotenv
from twilio.rest import Client

from sheety_data_manager import SheetyDataManager
from flight_club_ui import FlightClubApp
from flight_search import FlightSearch
from flight_search_token import GetDataFlight, GetToken
from email_sender import SendEmail
from flight_price_search import find_cheapest_flight

# Launch the user interface
FlightClubApp()

# Load environment variables
load_dotenv()

# Set up environment and communication data
EMAIL = os.getenv("SENDER_EMAIL")
TO_EMAIL = os.getenv("TO_EMAIL")
FROM_CITY = "LON"

# Set up Twilio for communication
account_sid = os.getenv("TWILIO_ACCOUNT_SID")
auth_token = os.getenv("TWILIO_AUTH_TOKEN")
client = Client(account_sid, auth_token)

# Set up data management and flight search services
data_manager = SheetyDataManager()
sheet_data = data_manager.get_sheet_data()
token = GetToken()  # Creating an instance of GetToken

# Get flight data
flight_data = GetDataFlight(token_id=token, data=data_manager)
flight_search = FlightSearch(token=token, data=data_manager)

# -------------------- Handle missing IATA codes --------------------
for row in sheet_data:
    if row["iataCode"] == "TESTING":
        data_manager.delete_sheet_data()
    elif row["iataCode"] == "":
        row["iataCode"] = flight_search.search_iata(row["city"])
        data_manager.update_sheet_data()

# -------------------- Prepare flight data -------------------------
file_name = "flight_data.json"

if os.path.exists(file_name):
    try:
        with open(file_name, mode="r") as file:
            data = json.load(file)
    except json.JSONDecodeError:
        data = []

    if isinstance(data, list):
        for i, flight in enumerate(data):
            if isinstance(flight, dict):
                flight["iataCode_to_city"] = []
                to_city = flight.get("to_city", [])
                for city in to_city:
                    iata_code = flight_search.search_iata(city)
                    flight["iataCode_to_city"].append(iata_code)

                from_city = flight.get("from_city", "")
                iata_code_from_city = flight_search.search_iata(from_city)
                flight["iataCode_from_city"] = iata_code_from_city
            else:
                print(f"Error: Invalid data at index {i}")

        with open(file_name, mode="w") as file:
            json.dump(data, file, indent=4)
            print("Flight data updated.")
else:
    print("Flight data file not found.")

# -------------------- Find the cheapest flights --------------------
all_messages = []
today = datetime.now() + timedelta(days=3)
six_months_later = today + timedelta(days=3 * 30)
tomorrow = today.strftime("%Y-%m-%d")
after_six_month = six_months_later.strftime("%Y-%m-%d")

with open(file_name, mode="r") as file:
    json_data = json.load(file)

    if isinstance(json_data, list):
        for data in json_data:
            if "iataCode_to_city" in data:
                for to_city in data["iataCode_to_city"]:
                    print(f"Getting flights for {data['name']} from {data['from_city']}...")

                    flight = flight_data.check_flights(
                        from_city=data["iataCode_from_city"],
                        to_city=to_city,
                        tomorrow=tomorrow,
                        after_six_month=after_six_month
                    )

                    row_data = flight
                    cheapest_flight = find_cheapest_flight(row_data)

                    if all([cheapest_flight.price != "N/A",
                            cheapest_flight.out_date != "N/A",
                            cheapest_flight.return_date != "N/A",
                            cheapest_flight.from_airport != "N/A",
                            cheapest_flight.to_airport != "N/A"]):
                        message_body = (
                            f"Destination: {data['from_city']} \n"
                            f"Price: £{cheapest_flight.price}\n"
                            f"From: {cheapest_flight.from_airport}\n"
                            f"To: {cheapest_flight.to_airport}\n"
                            f"Out Date: {cheapest_flight.out_date}\n"
                            f"Return Date: {cheapest_flight.return_date}\n"
                            "---------------------------------\n"
                        )
                        all_messages.append(message_body)
                        time.sleep(1)

        with open(file_name, mode="w") as file:
            for i, data in enumerate(json_data):
                if i < len(all_messages):
                    data["message"] = all_messages[i]

            json.dump(json_data, file, indent=4)
            print("Messages added and data updated.")

# -------------------- Send messages --------------------
with open(file_name, mode="r") as file:
    messages = json.load(file)
    for message in messages:
        try:
            # Send email if email is provided
            if message.get("email") and message["email"] != "@gmail.com":
                send_email = SendEmail(message=f"Hello {message['name']}\n{message['message']}")
                send_email.send_email(email=EMAIL, to_add=message["email"], password=os.getenv("EMAIL_PASSWORD"))

            # Send SMS if phone number is provided
            if message.get("phone_number") and message["phone_number"] != "+20":
                client.messages.create(
                    body=f"Hello {message['name']}\n{message['message']}",
                    from_="+13083652686",
                    to=message["phone_number"],
                )
                print("SMS sent.")
            else:
                print(f"{message['name']} does not have a valid phone number.")

            # Clear message field after sending
            message["message"] = ""

        except Exception as e:
            print(f"Error while sending message to {message['name']}: {e}")

    # Save updated messages with cleared "message" field
    with open(file_name, mode="w") as file:
        json.dump(messages, file, indent=4)

    print("All messages processed.")