all tools: Creating a shared group of argument options. Processing this

shared group too, changing the log level when logging options are
specified.

1 files changed, 25 insertions, 4 deletions

diff --git a/src/ee/logging.py b/src/ee/logging.py
index c8fd48a..aadfdd3 100644
--- a/src/ee/logging.py
+++ b/src/ee/logging.py
@@ -4,17 +4,38 @@ __all__ = ["Log", "log"]
 
 
 class Log(object):
+    LEVEL_WARN = 3
+    LEVEL_INFO = 2
+    LEVEL_DEBUG = 1
+
     def __init__(self):
-        pass
+        self.level = Log.LEVEL_INFO
 
     def warn(self, msg):
-        print(color(msg, "orange"))
+        self._p(Log.LEVEL_WARN, color(msg, "orange"))
 
     def info(self, msg):
-        print(color(msg, "white"))
+        self._p(Log.LEVEL_INFO, color(msg, "white"))
+
+    def summary(self, msg):
+        self._p(Log.LEVEL_INFO, color(msg, "white"))
 
     def debug(self, msg):
-        print(color(msg, "grey"))
+        self._p(Log.LEVEL_DEBUG, color(msg, "grey"))
+
+    def _p(self, level, msg):
+        if level >= self.level:
+            print(msg)
+
+    def set_level(self, level):
+        if level == "warn":
+            self.level = Log.LEVEL_WARN
+        elif level == "info":
+            self.level = Log.LEVEL_INFO
+        elif level == "debug":
+            self.level = Log.LEVEL_DEBUG
+        else:
+            self.level = Log.LEVEL_DEBUG
 
 
 log = Log()