Skip to content

Commit 1ca3ce9

Browse files
committed
Handling alternate cases of allow_ansi values
1 parent e73661c commit 1ca3ce9

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

cmd2/cmd2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -564,7 +564,7 @@ def allow_ansi(self) -> str:
564564
@allow_ansi.setter
565565
def allow_ansi(self, new_val: str) -> None:
566566
"""Read-only property needed to support do_set when it sets allow_ansi"""
567-
if new_val not in (ansi.ANSI_TERMINAL, ansi.ANSI_ALWAYS, ansi.ANSI_NEVER):
567+
if new_val.lower() not in (ansi.ANSI_TERMINAL.lower(), ansi.ANSI_ALWAYS.lower(), ansi.ANSI_NEVER.lower()):
568568
self.perror('Invalid value: {} (valid values: {}, {}, {})'.format(new_val, ansi.ANSI_TERMINAL,
569569
ansi.ANSI_ALWAYS, ansi.ANSI_NEVER))
570570
else:

tests/test_cmd2.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,9 @@ def test_set_quiet(base_app):
185185
(ansi.ANSI_NEVER, False),
186186
(ansi.ANSI_TERMINAL, False),
187187
(ansi.ANSI_ALWAYS, False),
188+
('neVeR', False),
189+
('TeRMInal', False),
190+
('AlWaYs', False),
188191
('invalid', True),
189192
])
190193
def test_set_allow_ansi(base_app, new_val, is_valid):

0 commit comments

Comments
 (0)