[virt-who/systemd: 2/2] Use journal for logging when available

[Radek Novacek]

commit 865efb828eb68d88b8bf16aa6b267298abc587a5
Author: Radek Novacek 
Date:   Tue Nov 27 12:35:33 2012 +0100

Use journal for logging when available

log.py |   51 ++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 48 insertions(+), 3 deletions(-)
---
diff --git a/log.py b/log.py
index e415001..bc85544 100644
--- a/log.py
+++ b/log.py
@@ -24,6 +24,40 @@ import logging.handlers
 import os
 import sys

+journalEnabled = False
+try:
+    from systemd import journal
+    journalEnabled = True
+except ImportError:
+    pass
+
+class NoExceptionFormatter(logging.Formatter):
+    def format(self, record):
+        if record.exc_info is not None:
+            record.exc_text = "\t" + str(record.exc_info[1])
+        return logging.Formatter.format(self, record)
+
+class JournalHandler(logging.Handler):
+    def __init__(self, level=logging.NOTSET):
+        logging.Handler.__init__(self, level)
+
+    def emit(self, record):
+        if journalEnabled:
+            try:
+                msg = self.format(record)
+                args = ['MESSAGE=' + record.message,
+                        'LOGGER=' + record.name,
+                        'THREAD_NAME=' + record.threadName,
+                        'CODE_FILE=' + record.pathname,
+                        'CODE_LINE=%d' % record.lineno,
+                        'CODE_FUNC=' + record.funcName]
+
+                journal.sendv(*args)
+            except (KeyboardInterrupt, SystemExit):
+                raise
+            except:
+                self.handleError(record)
+
 def getLogger(debug, background):
     logger = logging.getLogger("rhsm-app")
     logger.setLevel(logging.DEBUG)
@@ -56,10 +90,21 @@ def getLogger(debug, background):
             streamHandler.setLevel(logging.WARNING)

# Don't print exceptions to stdout in non-debug mode
-            f = logging.Filter()
-            f.filter = lambda record: record.exc_info is None
-            streamHandler.addFilter(f)
+            streamHandler.setFormatter(NoExceptionFormatter())
+

logger.addHandler(streamHandler)

+    if os.getppid() == 1:
+        # Also log to journal if available
+        journalHandler = JournalHandler()
+        if debug:
+            journalHandler.setLevel(logging.DEBUG)
+        else:
+            journalHandler.setLevel(logging.WARNING)
+
+            # Don't print exceptions to journal in non-debug mode
+            journalHandler.setFormatter(NoExceptionFormatter())
+
+        logger.addHandler(journalHandler)
     return logger