WIP2

logprep/util/helper.py

@@ -435,4 +435,34 @@ def get_ansi_code(color, type):
         'lightwhite': 67
     }
 
-    return f"\x1b[{str(base+colors.get(color))}m"
+    return f"\x1b[{str(base+colors.get(color))}m"
+
+
+class AnsiFore:
+    _colors = {
+        'RED': 31,
+        'GREEN': 32,
+        'BLUE': 34,
+        'RESET': 39
+    }
+
+    def __class_getitem__(cls, name):
+        if name in cls._colors:
+            return f'\033[{cls._colors[name]}m'
+        raise AttributeError(f"No Ansi code found: {name}")
+
+
+class AnsiBack:
+    _colors = {
+        'RED': 41,
+        'GREEN': 42,
+        'BLUE': 44,
+        'RESET': 49
+    }
+
+    def __class_getitem__(cls, name):
+        if name in cls._colors:
+            return f'\033[{cls._colors[name]}m'
+        raise AttributeError(f"No Ansi code found: {name}")
+
+

logprep/util/rule_dry_runner.py

@@ -50,6 +50,7 @@
 from logprep.util.configuration import Configuration
 from logprep.util.getter import GetterFactory
 from logprep.util.helper import color_print_line, color_print_title, recursive_compare, get_ansi_code
+from logprep.util.helper import AnsiBack, AnsiFore
 
 yaml = YAML(typ="safe", pure=True)