Add a log-file

1 files changed, 86 insertions, 0 deletions

diff --git a/portato/log.py b/portato/log.py
new file mode 100644
index 0000000..971dddf
--- /dev/null
+++ b/portato/log.py
@@ -0,0 +1,86 @@
+# -*- coding: utf-8 -*-
+#
+# File: portato/log.py
+# This file is part of the Portato-Project, a graphical portage-frontend.
+#
+# Copyright (C) 2008 René 'Necoro' Neumann
+# This is free software.  You may redistribute copies of it under the terms of
+# the GNU General Public License version 2.
+# There is NO WARRANTY, to the extent permitted by law.
+#
+# Written by René 'Necoro' Neumann <necoro@necoro.net>
+
+from __future__ import absolute_import
+
+import logging
+import sys
+import os
+
+from .constants import SESSION_DIR
+
+(S_NOT, S_STREAM_ONLY, S_BOTH) = range(3)
+
+started = S_NOT
+
+class OutputFormatter (logging.Formatter):
+
+    colors = {
+            "blue"    : 34,
+            "green"    : 32,
+            "red"    : 31,
+            "yellow": 33
+            }
+
+    def __init__(self, *args, **kwargs):
+        logging.Formatter.__init__(self, *args, **kwargs)
+
+        for key, value in self.colors.iteritems():
+            self.colors[key] = "\x1b[01;%02dm*\x1b[39;49;00m" % value
+
+    def format (self, record):
+        string = logging.Formatter.format(self, record)
+        color = None
+
+        if os.isatty(sys.stderr.fileno()):
+            if record.levelno <= logging.DEBUG:
+                color = self.colors["blue"]
+            elif record.levelno <= logging.INFO:
+                color = self.colors["green"]
+            elif record.levelno <= logging.WARNING:
+                color = self.colors["yellow"]
+            else:
+                color = self.colors["red"]
+        else:
+            color = "%s:" % record.levelname
+
+        return "%s %s" % (color, string)
+
+def start(file = True):
+    global started
+
+    if started == S_BOTH: return
+
+    # logging: root (file)
+    if file:
+        if not (os.path.exists(SESSION_DIR) and os.path.isdir(SESSION_DIR)):
+            os.mkdir(SESSION_DIR)
+
+        formatter = logging.Formatter("%(levelname)-8s: %(message)s (%(filename)s:%(lineno)s)")
+        handler = logging.FileHandler(os.path.join(SESSION_DIR, "portato.log"), "w")
+        handler.setFormatter(formatter)
+        logging.getLogger("portatoLogger").addHandler(handler)
+    
+    if started == S_NOT:
+        logging.getLogger("portatoLogger").setLevel(logging.DEBUG)
+        logging.getLogger("portatoLogger").propagate = False
+
+    # logging: stream
+    # this logger should be used
+    if started == S_NOT:
+        formatter = OutputFormatter("%(message)s (%(filename)s:%(lineno)s)")
+        handler = logging.StreamHandler()
+        handler.setFormatter(formatter)
+        logging.getLogger("portatoLogger.stream").addHandler(handler)
+        logging.getLogger("portatoLogger.stream").setLevel(logging.DEBUG)
+
+    started = S_BOTH if file else S_STREAM_ONLY