def _setup_flight_menu():
menu = Menu("Flights Menu")
menu.new_option("n", "New Flight", new_flight)
menu.new_option("e", "Edit Flight", edit_flight)
menu.new_option("l", "List Flights", list_flights)
menu.new_option("d", "Delete Flight", delete_flight)
return menu
def setup_reservation_menu():
menu = Menu("Reservations Menu")
menu.new_option("n", "New Reservation", new_reservation)
menu.new_option("e", "Edit Reservation", edit_reservation)
menu.new_option("l", "List Reservations", list_reservations)
menu.new_option("d", "Delete Reservation", delete_reservation)
return menu
from lib.menu import Menu
from lib.cli.reservations.menu import res_menu
from lib.cli.flights.menu import flight_menu
menu = Menu("Main Menu")
menu.new_option("r", "Reservations", res_menu.prompt)
menu.new_option("f", "Flights", flight_menu.prompt)
def main():
print("Welcome to the flight reservation system.")
try:
menu.prompt()
except KeyboardInterrupt:
pass
print("\n\nExiting...\n\n")
if __name__ == "__main__":
main()
