44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from os import system
|
|
import keyboard
|
|
from colorama import Fore
|
|
class MenuManager:
|
|
'''
|
|
Creates text menus controllable with arrow keys
|
|
'''
|
|
def __init__(self,selections:list,additional:str):
|
|
self.selected = 0 # current selection
|
|
self.selections = selections # available selections
|
|
self.additional = additional # additional text to display above the menu
|
|
keyboard.add_hotkey("up",self.up)
|
|
keyboard.add_hotkey("down",self.down)
|
|
keyboard.add_hotkey("enter",self.make_selection)
|
|
self.show_menu()
|
|
input()
|
|
|
|
def make_selection(self) -> int:
|
|
keyboard.remove_all_hotkeys()
|
|
|
|
def up(self):
|
|
if self.selected == 0:
|
|
self.selected = len(self.selections)-1
|
|
else:
|
|
self.selected -= 1
|
|
system("cls||clear")
|
|
self.show_menu()
|
|
|
|
def down(self):
|
|
if self.selected == len(self.selections)-1:
|
|
self.selected = 0
|
|
else:
|
|
self.selected += 1
|
|
system("cls||clear")
|
|
self.show_menu()
|
|
|
|
def show_menu(self):
|
|
print(self.additional)
|
|
for selection in self.selections:
|
|
if(self.selected == self.selections.index(selection)):
|
|
print(f"{Fore.RED}->{Fore.RESET} {selection}")
|
|
else:
|
|
print(f" {selection}")
|