first commit

dotdrop/logger.py

@@ -0,0 +1,65 @@
+"""
+author: deadc0de6 (https://github.com/deadc0de6)
+handle logging to stdout/stderr
+"""
+
+import sys
+
+
+class Logger:
+
+    RED = '\033[91m'
+    GREEN = '\033[92m'
+    YELLOW = '\033[93m'
+    BLUE = '\033[94m'
+    MAGENTA = '\033[95m'
+    RESET = '\033[0m'
+    EMPH = '\033[33m'
+
+    def __init__(self):
+        pass
+
+    def log(self, string, end='\n', pre=''):
+        cs = self._color(self.BLUE)
+        ce = self._color(self.RESET)
+        sys.stdout.write('%s%s%s%s%s' % (pre, cs, string, end, ce))
+
+    def sub(self, string):
+        cs = self._color(self.BLUE)
+        ce = self._color(self.RESET)
+        sys.stdout.write('\t%s->%s %s\n' % (cs, ce, string))
+
+    def emph(self, string):
+        cs = self._color(self.EMPH)
+        ce = self._color(self.RESET)
+        sys.stderr.write('%s%s%s' % (cs, string, ce))
+
+    def err(self, string, end='\n'):
+        cs = self._color(self.RED)
+        ce = self._color(self.RESET)
+        sys.stderr.write('%s[ERR]%s %s%s' % (cs, string, end, ce))
+
+    def warn(self, string, end='\n'):
+        cs = self._color(self.YELLOW)
+        ce = self._color(self.RESET)
+        sys.stderr.write('%s[WARN]%s %s%s' % (cs, string, end, ce))
+
+    def dry(self, string, end='\n'):
+        cs = self._color(self.GREEN)
+        ce = self._color(self.RESET)
+        sys.stdout.write('%s[DRY]%s %s%s' % (cs, string, end, ce))
+
+    def raw(self, string, end='\n'):
+        sys.stdout.write('%s%s' % (string, end))
+
+    def ask(self, query):
+        cs = self._color(self.BLUE)
+        ce = self._color(self.RESET)
+        q = '%s%s%s' % (cs, query + ' [y/N] ? ', ce)
+        r = input(q)
+        return r == 'y'
+
+    def _color(self, col):
+        if not sys.stdout.isatty():
+            return ''
+        return col