diff --git a/src/entry.py b/src/entry.py index 6e35853..fc7a46e 100644 --- a/src/entry.py +++ b/src/entry.py @@ -5,6 +5,6 @@ def entry(): initialize() # import should happen after the `initialize` function is called - from src.interface.choose_action import choose_action + from src.interface.menu import menu - choose_action() + menu() diff --git a/src/interface/menu.py b/src/interface/menu.py new file mode 100644 index 0000000..d707540 --- /dev/null +++ b/src/interface/menu.py @@ -0,0 +1,45 @@ +import inquirer + + +class Choices: + RUN_SETUP_SCRIPTS = "Run setup scripts" + INSTALL_ARCH_LINUX = "Install Arch Linux" + QUIT = "Quit" + + +def menu(): + """Show menu screen.""" + + print("\n") + + questions = [ + inquirer.List( + "menu", + message="What would you like to do?", + choices=[ + Choices.RUN_SETUP_SCRIPTS, + Choices.INSTALL_ARCH_LINUX, + Choices.QUIT, + ], + ), + ] + + choice = inquirer.prompt(questions)["menu"] + + if choice == Choices.RUN_SETUP_SCRIPTS: + from src.interface.choose_action import choose_action + + choose_action() + input("\nSetup complete! (press Enter to return to menu)") + menu() + elif choice == Choices.INSTALL_ARCH_LINUX: + input("Work In Progress (press Enter to return to menu)") + menu() + elif choice == Choices.QUIT: + exit(0) + else: + print( + """You should not have been able to reach this side of the code, +yet here you are. Consider this an easter egg.""" + ) + exit(1)