From 68e6ecbe4feedc17f35a87cac4d3116568cb638a Mon Sep 17 00:00:00 2001 From: Sebastian Lohff Date: Wed, 26 Aug 2020 23:46:35 +0200 Subject: [PATCH] Make CliMenuHeader and CliMenuOption private These classes don't need to be used by the user, only via the API, so we can mark them private. --- clintermission/climenu.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/clintermission/climenu.py b/clintermission/climenu.py index c45cd6a..eeb9f93 100644 --- a/clintermission/climenu.py +++ b/clintermission/climenu.py @@ -12,7 +12,7 @@ from prompt_toolkit import search from prompt_toolkit.widgets import SearchToolbar -class CliMenuHeader: +class _CliMenuHeader: """Hold a menu header""" def __init__(self, text, indent=False): self.text = text @@ -20,7 +20,7 @@ class CliMenuHeader: self.focusable = False -class CliMenuOption: +class _CliMenuOption: """Hold a menu option""" def __init__(self, text, num, item=None): self.text = text @@ -117,10 +117,10 @@ class CliMenu: def add_text(self, title, indent=True): for text in title.split('\n'): - self._items.append(CliMenuHeader(text, indent=indent)) + self._items.append(_CliMenuHeader(text, indent=indent)) def add_option(self, text, item=None): - self._items.append(CliMenuOption(text, self._item_num, item=item)) + self._items.append(_CliMenuOption(text, self._item_num, item=item)) self._item_num += 1 @property @@ -131,7 +131,7 @@ class CliMenu: return self._success def get_options(self): - return [_item for _item in self._items if isinstance(_item, CliMenuOption)] + return [_item for _item in self._items if isinstance(_item, _CliMenuOption)] @property def num_options(self):