Download as docx, pdf, or txt
Download as docx, pdf, or txt
You are on page 1of 3

class Flight:

def __init__(self, flight_number, departure_city, destination, departure_time, arrival_time):

self.flight_number = flight_number

self.departure_city = departure_city

self.destination = destination

self.departure_time = departure_time

self.arrival_time = arrival_time

class FlightManager:

def __init__(self):

self.flights = []

def add_flight(self, flight):

self.flights.append(flight)

print("Flight added successfully.")

def display_all_flights(self):

if self.flights:

print("All Flights:")

for flight in self.flights:

print(f"Flight Number: {flight.flight_number}, Departure City: {flight.departure_city},


Destination: {flight.destination}, Departure Time: {flight.departure_time}, Arrival Time:
{flight.arrival_time}")

else:

print("No flights available.")

def search_flights_by_destination(self, destination):

found_flights = [flight for flight in self.flights if flight.destination == destination]

if found_flights:

print(f"Flights to {destination}:")

for flight in found_flights:


print(f"Flight Number: {flight.flight_number}, Departure City: {flight.departure_city},
Departure Time: {flight.departure_time}, Arrival Time: {flight.arrival_time}")

else:

print(f"No flights found to {destination}.")

def search_flights_by_departure_city(self, departure_city):

found_flights = [flight for flight in self.flights if flight.departure_city == departure_city]

if found_flights:

print(f"Flights from {departure_city}:")

for flight in found_flights:

print(f"Flight Number: {flight.flight_number}, Destination: {flight.destination}, Departure


Time: {flight.departure_time}, Arrival Time: {flight.arrival_time}")

else:

print(f"No flights found from {departure_city}.")

# Sample Usage

if __name__ == "__main__":

manager = FlightManager()

# Adding sample flights

flight1 = Flight("ABC123", "New York", "Los Angeles", "10:00", "13:00")

flight2 = Flight("DEF456", "Los Angeles", "Chicago", "14:00", "16:30")

flight3 = Flight("GHI789", "Chicago", "New York", "17:00", "19:30")

manager.add_flight(flight1)

manager.add_flight(flight2)

manager.add_flight(flight3)

# Display all flights

manager.display_all_flights()

# Search flights by destination


manager.search_flights_by_destination("Los Angeles")

# Search flights by departure city

manager.search_flights_by_departure_city("Chicago")

You might also like