general: Added logging.

1 files changed, 13 insertions, 1 deletions

diff --git a/pywal/util.py b/pywal/util.py
index 65d4557..0c678c9 100644
--- a/pywal/util.py
+++ b/pywal/util.py
@@ -3,6 +3,7 @@ Misc helper functions.
 """
 import colorsys
 import json
+import logging
 import os
 import subprocess
 
@@ -75,7 +76,7 @@ def save_file(data, export_file):
         with open(export_file, "w") as file:
             file.write(data)
     except PermissionError:
-        print("warning: Couldn't write to %s." % export_file)
+        logging.warning("Couldn't write to %s.", export_file)
 
 
 def save_file_json(data, export_file):
@@ -91,6 +92,17 @@ def create_dir(directory):
     os.makedirs(directory, exist_ok=True)
 
 
+def setup_logging():
+    """Logging config."""
+    logging.basicConfig(format=("[%(levelname)s\033[0m] "
+                                "\033[1;31m%(module)s\033[0m: "
+                                "%(message)s"),
+                        level=logging.INFO)
+    logging.addLevelName(logging.ERROR, '\033[1;31mE')
+    logging.addLevelName(logging.INFO, '\033[1;32mI')
+    logging.addLevelName(logging.WARNING, '\033[1;33mW')
+
+
 def hex_to_rgb(color):
     """Convert a hex color to rgb."""
     return tuple(bytes.fromhex(color.strip("#")))