Compare commits

...

2 Commits

Author SHA1 Message Date
Sebastian Lohff e0177d0e10 Fix inline header indentation 2019-02-01 18:17:57 +01:00
Sebastian Lohff b18372e3b8 Allow passing an item to each option 2019-02-01 18:17:57 +01:00
1 changed files with 16 additions and 7 deletions

View File

@ -19,9 +19,10 @@ class CliMenuHeader:
class CliMenuOption:
"""Hold a menu option"""
def __init__(self, text, num):
def __init__(self, text, num, item=None):
self.text = text
self.num = num
self.item = item
self.focusable = True
@ -96,19 +97,27 @@ class CliMenu:
for option in options:
self.add_option(option)
def add_header(self, title, indent=False):
def add_header(self, title, indent=True):
for text in title.split('\n'):
self._items.append(CliMenuHeader(text))
self._items.append(CliMenuHeader(text, indent=indent))
def add_option(self, text):
self._items.append(CliMenuOption(text, self._item_num))
def add_option(self, text, item=None):
self._items.append(CliMenuOption(text, self._item_num, item=item))
self._item_num += 1
def get_selection(self):
if not self._ran:
self._run()
return self._items[self._pos].num
item = self._items[self._pos]
return (item.num, item.item)
def get_selection_num(self):
return self.get_selection()[0]
def get_selection_item(self):
return self.get_selection()[1]
def cursor(self):
return '{} '.format(self._cursor)
@ -137,7 +146,7 @@ class CliMenu:
style = s.option_style
else:
if item.indent:
indent += ' ' * self._header_indent
indent += ' ' * (self._header_indent + len(self._cursor) + 1)
style = s.header_style
return Transformation([('', indent), (style, prefix + text)])