from pydantic import BaseModel
from datetime import datetime
from typing import Literal


# ======================
# FLIGHT
# ======================
class Flight(BaseModel):
    flight_id: str
    airline: str
    origin: str
    destination: str
    departure_time: datetime
    arrival_time: datetime
    duration_minutes: int
    seat_class: Literal["FIRST", "BUSINESS", "ECONOMY"]
    price: float


# ======================
# BOOKING
# ======================
class BookingRequest(BaseModel):
    customer_name: str
    flight_id: str
    customer_id: str


# ======================
# REFUND
# ======================
class RefundRequest(BaseModel):
    booking_id: str


# ======================
# RESCHEDULE
# ======================
class RescheduleRequest(BaseModel):
    booking_id: str
    new_flight_id: str