blob: 038fde15b8346b7ced11466014bb3fd0cd05d930 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
"""Text utils."""
from __future__ import annotations
import re
from functools import cache
RE_HAS_JINJA = re.compile(r"{[{%#].*[%#}]}", re.DOTALL)
RE_HAS_GLOB = re.compile("[][*?]")
def strip_ansi_escape(data: str | bytes) -> str:
"""Remove all ANSI escapes from string or bytes.
If bytes is passed instead of string, it will be converted to string
using UTF-8.
"""
if isinstance(data, bytes): # pragma: no branch
data = data.decode("utf-8")
return re.sub(r"\x1b[^m]*m", "", data)
def toidentifier(text: str) -> str:
"""Convert unsafe chars to ones allowed in variables."""
result = re.sub(r"[\s-]+", "_", text)
if not result.isidentifier():
msg = f"Unable to convert role name '{text}' to valid variable name."
raise RuntimeError(msg)
return result
# https://www.python.org/dev/peps/pep-0616/
def removeprefix(self: str, prefix: str) -> str:
"""Remove prefix from string."""
if self.startswith(prefix):
return self[len(prefix) :]
return self[:]
@cache
def has_jinja(value: str) -> bool:
"""Return true if a string seems to contain jinja templating."""
return bool(isinstance(value, str) and RE_HAS_JINJA.search(value))
@cache
def has_glob(value: str) -> bool:
"""Return true if a string looks like having a glob pattern."""
return bool(isinstance(value, str) and RE_HAS_GLOB.search(value))
|
