Rename bfxapi.utils to _utils (and update references).

bfxapi/_utils/logger.py

@@ -0,0 +1,51 @@
+import logging, sys
+
+BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = range(8)
+
+COLOR_SEQ, ITALIC_COLOR_SEQ = "\033[1;%dm", "\033[3;%dm"
+
+COLORS = {
+    "DEBUG": CYAN,
+    "INFO": BLUE,
+    "WARNING": YELLOW,
+    "ERROR": RED
+}
+
+RESET_SEQ = "\033[0m"
+
+class _ColorFormatter(logging.Formatter):
+    def __init__(self, msg, use_color = True):
+        logging.Formatter.__init__(self, msg, "%d-%m-%Y %H:%M:%S")
+
+        self.use_color = use_color
+
+    def format(self, record):
+        levelname = record.levelname
+        if self.use_color and levelname in COLORS:
+            record.name = ITALIC_COLOR_SEQ % (30 + BLACK) + record.name + RESET_SEQ
+            record.levelname = COLOR_SEQ % (30 + COLORS[levelname]) + levelname + RESET_SEQ
+        return logging.Formatter.format(self, record)
+
+class ColorLogger(logging.Logger):
+    FORMAT = "[%(name)s] [%(levelname)s] [%(asctime)s] %(message)s"
+
+    def __init__(self, name, level):
+        logging.Logger.__init__(self, name, level)
+
+        colored_formatter = _ColorFormatter(self.FORMAT, use_color=True)
+        handler = logging.StreamHandler(stream=sys.stderr)
+        handler.setFormatter(fmt=colored_formatter)
+
+        self.addHandler(hdlr=handler)
+
+class FileLogger(logging.Logger):
+    FORMAT = "[%(name)s] [%(levelname)s] [%(asctime)s] %(message)s"
+
+    def __init__(self, name, level, filename):
+        logging.Logger.__init__(self, name, level)
+
+        formatter = logging.Formatter(self.FORMAT)
+        handler = logging.FileHandler(filename=filename)
+        handler.setFormatter(fmt=formatter)
+
+        self.addHandler(hdlr=handler)