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()