From a28c76e0019d9a675608d0a6dfd500caf5a0f0d0 Mon Sep 17 00:00:00 2001 From: Sebastian Lohff Date: Wed, 5 Aug 2020 02:32:15 +0200 Subject: [PATCH] Allow a default theme and cursor to be set --- clintermission/climenu.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/clintermission/climenu.py b/clintermission/climenu.py index 68ff1a0..9bc3cea 100644 --- a/clintermission/climenu.py +++ b/clintermission/climenu.py @@ -75,6 +75,14 @@ class CliMenu: default_style = CliMenuTheme.BASIC default_cursor = CliMenuCursor.TRIANGLE + @classmethod + def set_default_style(cls, style): + cls.default_style = style + + @classmethod + def set_default_cursor(cls, cursor): + cls.default_cursor = cursor + def __init__(self, options=None, header=None, cursor=None, style=None, indent=2, dedent_selection=False): self._items = [] @@ -86,13 +94,8 @@ class CliMenu: self._header_indent = indent self._dedent_selection = dedent_selection - self._cursor = cursor - if not self._cursor: - self._cursor = self.default_cursor - - self._style = style - if not self._style: - self._style = self.default_style + self._cursor = cursor or self.default_cursor + self._style = style or self.default_style if header: self.add_text(header, indent=False)