Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2024-12-22 19:23:34 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Replace rich formatting stripper (#706)

Browse Source 
* Revert "Fix overly greedy _strip_rich_formatting (#703)"

This reverts commit ae6a31463e.

* Improve rich formatter stripper

reverts and replaces #703
This commit is contained in:
J. Nick Koston 2024-01-24 21:49:26 -10:00 committed by GitHub
parent 8947ffbc94
commit fa6bc59b29
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
kasa/cli.py
View File

@ -37,9 +37,11 @@ except ImportError:
try:
from rich import print as _do_echo
except ImportError:
# Remove 7-bit C1 ANSI sequences
# https://stackoverflow.com/questions/14693701/how-can-i-remove-the-ansi-escape-sequences-from-a-string-in-python
ansi_escape = re.compile(r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])")
# Strip out rich formatting if rich is not installed
# but only lower case tags to avoid stripping out
# raw data from the device that is printed from
# the device state.
rich_formatting = re.compile(r"\[/?[a-z]+]")
def _strip_rich_formatting(echo_func):
"""Strip rich formatting from messages."""
@ -47,7 +49,7 @@ except ImportError:
@wraps(echo_func)
def wrapper(message=None, *args, **kwargs):
if message is not None:
message = ansi_escape.sub("", message)
message = rich_formatting.sub("", message)
echo_func(message, *args, **kwargs)
return wrapper