140 lines
3.5 KiB
Python
140 lines
3.5 KiB
Python
'''Wizard Kit: Functions - Advanced Menu'''
|
|
# vim: sts=2 sw=2 ts=2
|
|
|
|
import os
|
|
from collections import OrderedDict
|
|
|
|
# STATIC VARIABLES
|
|
COLORS = {
|
|
'CLEAR': '\033[0m',
|
|
'RED': '\033[31m',
|
|
'ORANGE': '\033[31;1m',
|
|
'GREEN': '\033[32m',
|
|
'YELLOW': '\033[33m',
|
|
'BLUE': '\033[34m',
|
|
'PURPLE': '\033[35m',
|
|
'CYAN': '\033[36m',
|
|
}
|
|
VALID_ENTRY_TYPES = (
|
|
'Action',
|
|
'Option',
|
|
'Set',
|
|
'Toggle',
|
|
)
|
|
|
|
|
|
# Classes
|
|
class MenuEntry():
|
|
"""Class for advanced menu entries"""
|
|
# pylint: disable=too-few-public-methods
|
|
def __init__(self, **kwargs):
|
|
kwargs = {str(k).lower(): v for k, v in kwargs.items()}
|
|
self.name = kwargs.pop('name', None)
|
|
self.type = kwargs.pop('type', None)
|
|
self.disabled = kwargs.pop('disabled', False)
|
|
self.hidden = kwargs.pop('hidden', False)
|
|
self.selected = kwargs.pop('selected', False)
|
|
self.separator = kwargs.pop('separator', False)
|
|
|
|
# Other attributes
|
|
for _key, _value in kwargs.items():
|
|
setattr(self, _key, kwargs.get(_key))
|
|
del kwargs
|
|
|
|
# Check attributes
|
|
self.check()
|
|
|
|
def check(self):
|
|
"""Check for invalid or missing attributes."""
|
|
assert self.name, 'Invalid menu entry name.'
|
|
assert self.type in VALID_ENTRY_TYPES, 'Invalid menu entry type.'
|
|
|
|
|
|
class MenuState():
|
|
"""Class to track various parts of the advanced menu."""
|
|
def __init__(self, title):
|
|
self.checkmark = '✓' if 'DISPLAY' in os.environ else '*'
|
|
self.entries = OrderedDict({})
|
|
self.last_sel = None
|
|
self.sep = '─'
|
|
self.sep_len = 0
|
|
self.title = title
|
|
|
|
def add_entry(self, **kwargs):
|
|
"""Add entry and update state."""
|
|
_e = MenuEntry(**kwargs)
|
|
assert _e.name not in self.entries, 'Duplicate menu entry.'
|
|
|
|
# Add to entries
|
|
self.entries[_e.name] = _e
|
|
|
|
# Update sep length
|
|
self.sep_len = max(len(_e.name), self.sep_len)
|
|
|
|
def make_single_selection(self):
|
|
"""Select single entry."""
|
|
_sep_len = self.sep_len + 3
|
|
display_list = [self.title, '']
|
|
valid_answers = {}
|
|
|
|
# Safety Check
|
|
assert self.entries, 'No menu entries defined.'
|
|
|
|
# Build Menu
|
|
i = 1
|
|
for name, entry in self.entries.items():
|
|
# Skip sets
|
|
if entry.type in ('Set', 'Toggle'):
|
|
continue
|
|
|
|
# Separators
|
|
if entry.separator:
|
|
display_list.append(self.sep * _sep_len)
|
|
|
|
# Entries
|
|
_prefix = None
|
|
if entry.type == 'Option':
|
|
_prefix = str(i)
|
|
i += 1
|
|
elif entry.type == 'Action':
|
|
_prefix = name[0:1].upper()
|
|
display_list.append('{}: {}'.format(_prefix, name))
|
|
valid_answers[_prefix] = name
|
|
|
|
# Disable entry if necessary
|
|
if entry.disabled:
|
|
display_list[-1] = '{ORANGE}{text}{CLEAR}'.format(
|
|
text=display_list[-1],
|
|
**COLORS,
|
|
)
|
|
valid_answers.pop(_prefix)
|
|
|
|
# Hide action entry if necessary
|
|
if entry.type == 'Action' and entry.hidden:
|
|
display_list.pop()
|
|
|
|
# Show Menu and make selection
|
|
_answer = ''
|
|
while _answer.upper() not in valid_answers:
|
|
clear()
|
|
print('\n'.join(display_list))
|
|
_answer = input('Please make a selection: ')
|
|
|
|
# Save last selection
|
|
self.last_sel = valid_answers[_answer.upper()]
|
|
|
|
def make_multiple_selections(self):
|
|
"""Select one or more entries."""
|
|
_sep_len = self.sep_len + 7
|
|
# TODO
|
|
|
|
# Functions
|
|
def clear():
|
|
"""Simple wrapper for cls/clear."""
|
|
if os.name == 'nt':
|
|
os.system('cls')
|
|
else:
|
|
os.system('clear')
|
|
|
|
if __name__ == '__main__':
|
|
print('This file is not meant to be called directly.')
|