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
---
pyblog | 23 +++--------------------
utils/__init__.py | 0
utils/utils.py | 23 +++++++++++++++++++++++
3 files changed, 26 insertions(+), 20 deletions(-)
create mode 100644 utils/__init__.py
create mode 100644 utils/utils.py
diff --git a/pyblog b/pyblog
index f4ae79ab..1511dce4 100755
--- a/pyblog
+++ b/pyblog
@@ -44,6 +44,8 @@ import toml
from rss import *
+from utils import utils
+
ROOTDIR = os.path.dirname(os.path.realpath(__file__))
@@ -96,25 +98,6 @@ WHITE = ""
YELLOW = ""
RESET = ""
-@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()
-
-
def current_datetime():
"""Return the current datetime, complete with tzinfo.
@@ -1723,7 +1706,7 @@ def main():
description="Bring up post selector to select post for editing.")
parser_new_post.set_defaults(func=edit_existing_post)
- with init_colorama():
+ with utils.init_colorama():
args = parser.parse_args()
returncode = args.func(args)
exit(returncode)
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