nos0s/menu.py

54 lines
1.4 KiB
Python
Raw Normal View History

class MenuState:
def __init__(self, title, desc):
self.title = title
self.description = desc
self.parent = None
self.subs = []
def add_submenu(self, menu):
self.subs.append(menu)
menu.parent = self
def display_CLI(self):
display_text = f'{self.title}\n'
if self.parent != None:
display_text += f' b. Back'
for i in range(0, len(self.subs)):
sub = self.subs[i]
display_text += f' {i}. {sub.title}\n'
display_text += f' - {sub.description}\n'
return display_text
def nav_to_sub(self, sub_index):
if sub_index <= len(self.subs):
return self.subs[sub_index]
else:
return self
def nav_to_parent(self):
if self.parent != None:
return self.parent
else:
return self
class LeafMenu(MenuState):
def __init__(self, title, desc):
super.__init__(title, desc)
self.subs = None
self.actions = []
def add_submenu(self, menu):
return
def add_action(self, action):
self.actions.append(action)
def execute_action(self, act_index):
if act_index <= len(self.actions):
# what
self.actions[act_index]()
else:
return