return ret
+def strip_ansi_sequences(in_str: str) -> str:
+ """Strips ANSI sequences out of strings.
+
+ >>> import ansi as a
+ >>> s = a.fg('blue') + 'blue!' + a.reset()
+ >>> len(s) # '\x1b[38;5;21mblue!\x1b[m'
+ 18
+ >>> len(strip_ansi_sequences(s))
+ 5
+ >>> strip_ansi_sequences(s)
+ 'blue!'
+
+ """
+ return re.sub(r'\x1b\[[\d+;]*[a-z]', '', in_str)
+
+
class SprintfStdout(contextlib.AbstractContextManager):
"""
A context manager that captures outputs to stdout.
return in_str
-def replace_nth(string: str, source: str, target: str, nth: int):
+def replace_nth(in_str: str, source: str, target: str, nth: int):
"""Replaces the nth occurrance of a substring within a string.
>>> replace_nth('this is a test', ' ', '-', 3)
'this is a-test'
"""
- where = [m.start() for m in re.finditer(source, string)][nth - 1]
- before = string[:where]
- after = string[where:]
+ where = [m.start() for m in re.finditer(source, in_str)][nth - 1]
+ before = in_str[:where]
+ after = in_str[where:]
after = after.replace(source, target, 1)
return before + after