diff options
author | Todd Leonhardt <todd.leonhardt@gmail.com> | 2020-02-07 06:12:14 -0500 |
---|---|---|
committer | Todd Leonhardt <todd.leonhardt@gmail.com> | 2020-02-07 06:12:14 -0500 |
commit | 09aeb21dbdbc7b6016daa3d9ee277d7050abb7d4 (patch) | |
tree | f4c98d9873e9885098bb3bb0d6ead7f70e5043c6 /cmd2/ansi.py | |
parent | 4307a20ea2e6d3df8f3ca628aced7fdaaf3c89c1 (diff) | |
download | cmd2-git-09aeb21dbdbc7b6016daa3d9ee277d7050abb7d4.tar.gz |
Removed unnecessary methods from fg and bg color enums
Diffstat (limited to 'cmd2/ansi.py')
-rw-r--r-- | cmd2/ansi.py | 22 |
1 files changed, 2 insertions, 20 deletions
diff --git a/cmd2/ansi.py b/cmd2/ansi.py index 013aa107..98b98726 100644 --- a/cmd2/ansi.py +++ b/cmd2/ansi.py @@ -50,20 +50,11 @@ class fg(Enum): bright_white = Fore.LIGHTWHITE_EX reset = Fore.RESET - def __str__(self) -> str: - """Make the value the string representation instead of the enum name.""" - return self.value - @staticmethod def colors() -> List[str]: """Return a list of color names.""" return [color.name for color in fg] - @staticmethod - def get_value(name: str) -> str: - """Retrieve color code by name string.""" - return fg[name].value - # Background colors # noinspection PyPep8Naming,DuplicatedCode @@ -88,20 +79,11 @@ class bg(Enum): bright_white = Back.LIGHTWHITE_EX reset = Back.RESET - def __str__(self) -> str: - """Make the value the string representation instead of the enum name.""" - return self.value - @staticmethod def colors() -> List[str]: """Return a list of color names.""" return [color.name for color in bg] - @staticmethod - def get_value(name: str) -> str: - """Retrieve color code by name string.""" - return bg[name].value - FG_RESET = fg.reset.value BG_RESET = bg.reset.value @@ -162,7 +144,7 @@ def fg_lookup(fg_name: Union[str, fg]) -> str: return fg_name.value try: - ansi_escape = fg.get_value(fg_name.lower()) + ansi_escape = fg[fg_name.lower()].value except KeyError: raise ValueError('Foreground color {!r} does not exist; must be one of: {}'.format(fg_name, fg.colors())) return ansi_escape @@ -180,7 +162,7 @@ def bg_lookup(bg_name: Union[str, bg]) -> str: return bg_name.value try: - ansi_escape = bg.get_value(bg_name.lower()) + ansi_escape = bg[bg_name.lower()].value except KeyError: raise ValueError('Background color {!r} does not exist; must be one of: {}'.format(bg_name, bg.colors())) return ansi_escape |