Skip to content

Commit 83e4188

Browse files
committed
Refactored allow_ansi setter
1 parent 3bf3bf6 commit 83e4188

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

cmd2/cmd2.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -564,12 +564,16 @@ def allow_ansi(self) -> str:
564564
@allow_ansi.setter
565565
def allow_ansi(self, new_val: str) -> None:
566566
"""Setter property needed to support do_set when it updates allow_ansi"""
567-
new_val = new_val.capitalize()
568-
if new_val not in (ansi.ANSI_TERMINAL, ansi.ANSI_ALWAYS, ansi.ANSI_NEVER):
567+
new_val = new_val.lower()
568+
if new_val == ansi.ANSI_TERMINAL.lower():
569+
ansi.allow_ansi = ansi.ANSI_TERMINAL
570+
elif new_val == ansi.ANSI_ALWAYS.lower():
571+
ansi.allow_ansi = ansi.ANSI_ALWAYS
572+
elif new_val == ansi.ANSI_NEVER.lower():
573+
ansi.allow_ansi = ansi.ANSI_NEVER
574+
else:
569575
self.perror('Invalid value: {} (valid values: {}, {}, {})'.format(new_val, ansi.ANSI_TERMINAL,
570576
ansi.ANSI_ALWAYS, ansi.ANSI_NEVER))
571-
else:
572-
ansi.allow_ansi = new_val
573577

574578
@property
575579
def visible_prompt(self) -> str:

0 commit comments

Comments
 (0)