diff options
Diffstat (limited to 'menu.py')
| -rw-r--r-- | menu.py | 114 |
1 files changed, 114 insertions, 0 deletions
@@ -0,0 +1,114 @@ +# check if the entry.py and time modules are being used. Remove import if not. +from msvcrt import * +from entry import * +from time import * + +class Menu(): + def __init__(self, prompt, entries, color_int=7, side_bar=None): + self.prompt = prompt + self.entries = entries + self.iter_entries = iter(entries) + self.color_int = color_int + self.next_item = None + self.side_bar = side_bar + + @property + def text_color(self): + if self.color_int > 0 and self.color_int <=7: + return f'\033[9{self.color_int}m' + + @property + def indicate_color(self): + if self.color_int > 0 and self.color_int <=7: + return f'\033[10{self.color_int}m\033[30m' + + def refresh(self): + if self.color_int == 7: color_int = None + else: color_int = self.color_int + self.text = '\n' + if color_int: self.text += f'\033[9{color_int}m' + self.text += self.prompt + self.text += '\n' * 2 + for entry in self.entries: + if hasattr(entry, 'get_options'): + entry.get_options() + self.text += '\t' + entry.indicate(color_int) + if color_int: self.text += f'\033[9{color_int}m' + self.text += '\n' + self.text += '\n' + + def open(self, prev_menu): + self.clear() + self.prev_menu = prev_menu + self.next_item = None + self.entries.set_select() + self.refresh() + self.print_side_bar() + print(self.text) + self.keyboard_nav() + + def clear(self): + print('\033[0m', '\033[?25l', '\033[2J', '\033[0;0f', sep='', end='') + + def reprint(self): + self.clear() + self.print_side_bar() + print(self.text) + + def print_side_bar(self): + if self.side_bar: + lines = self.side_bar.split('\n') + l = 2 + for line in lines: + print(f'\033[{l};40f', f'\033[9{self.color_int}m', line, '\033[0m', sep='', end='') + l+=1 + print('\033[0;0f', end='') + + + def nav_entries(self, next='NEXT'): + self.entries.nav(next) + self.refresh() + self.reprint() + + + def edit_param(self, param): + param.edit = True + self.refresh() + self.reprint() + param.keyboard_edit() + param.edit = False + self.refresh() + self.reprint() + + def keyboard_nav(self): + while True: + cmd = getch() + if cmd == b'\x1b' or cmd == b'\x08': + self.next_item = self.prev_menu + break + elif cmd == b'\x00' or cmd == b'\xe0': + cmd = getch() + if cmd == b'M' or cmd == b'P': + self.nav_entries(next='NEXT') + elif cmd == b'K' or cmd == b'H': + self.nav_entries(next='PREV') + elif cmd == b'\t': + self.nav_entries(next='NEXT') + elif cmd == b'\r': + for entry in self.entries: + if entry.select and type(entry) is Param: + self.edit_param(entry) + elif entry.select: + self.next_item = entry.item + if self.next_item != None: + break + + + + + + + + + +
\ No newline at end of file |