From a8e56413815123b4bb10f3ddd597f42d11c8ad46 Mon Sep 17 00:00:00 2001
From: neodarz <neodarz@neodarz.net>
Date: Sat, 25 May 2019 23:33:13 +0200
Subject: Move init_colorama function to external file
---
utils/__init__.py | 0
utils/utils.py | 23 +++++++++++++++++++++++
2 files changed, 23 insertions(+)
create mode 100644 utils/__init__.py
create mode 100644 utils/utils.py
(limited to 'utils')
diff --git a/utils/__init__.py b/utils/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/utils/utils.py b/utils/utils.py
new file mode 100644
index 00000000..a76beb5d
--- /dev/null
+++ b/utils/utils.py
@@ -0,0 +1,23 @@
+#!/bin/python3
+
+from contextlib import contextmanager
+import colorama
+
+
+@contextmanager
+def init_colorama():
+ """Set global foreground modifying ANSI codes.
+
+ BLACK, BLUE, CYAN, GREEN, MAGENTA, RED, WHITE, YELLOW, and RESET.
+
+ """
+
+ # pylint: disable=exec-used,invalid-name
+
+ colorama.init()
+ for color, ansi in colorama.Fore.__dict__.items():
+ exec("global {0}; {0} = '{1}'".format(color, ansi))
+ yield
+ for color in colorama.Fore.__dict__:
+ exec("global {0}; {0} = ''".format(color))
+ colorama.deinit()
--
cgit v1.2.1