summaryrefslogtreecommitdiffstats
path: root/src/ansiblelint/text.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/ansiblelint/text.py')
-rw-r--r--src/ansiblelint/text.py50
1 files changed, 50 insertions, 0 deletions
diff --git a/src/ansiblelint/text.py b/src/ansiblelint/text.py
new file mode 100644
index 0000000..12ec9ad
--- /dev/null
+++ b/src/ansiblelint/text.py
@@ -0,0 +1,50 @@
+"""Text utils."""
+from __future__ import annotations
+
+import re
+from functools import lru_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():
+ raise RuntimeError(
+ f"Unable to convert role name '{text}' to valid variable name."
+ )
+ 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[:]
+
+
+@lru_cache(maxsize=None)
+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))
+
+
+@lru_cache(maxsize=None)
+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))