#74 Jefferson/Rafa: Centralized all logs (python logging and twisted logs) to /tmp/pixelated.log file

1 files changed, 80 insertions, 0 deletions

diff --git a/service/pixelated/config/logging_setup.py b/service/pixelated/config/logging_setup.py
new file mode 100644
index 00000000..f42f91c9
--- /dev/null
+++ b/service/pixelated/config/logging_setup.py
@@ -0,0 +1,80 @@
+#
+# Copyright (c) 2014 ThoughtWorks, Inc.
+#
+# Pixelated is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pixelated is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
+
+import logging
+import socket
+import sys
+import os
+from twisted.python import log
+from twisted.python import util
+
+
+LOG_FILE = '/tmp/pixelated.log'
+
+
+def init_logging(args):
+    pixelated_log_format = '[%(asctime)s] ' + socket.gethostname() + ' %(name)-12s %(levelname)-8s %(message)s'
+    pixelated_log = logging.FileHandler(LOG_FILE)
+    pixelated_log.setLevel(logging.DEBUG)
+    pixelated_log.setFormatter(logging.Formatter(pixelated_log_format))
+
+    logging.getLogger('').addHandler(pixelated_log)
+    init_twisted_logging()
+    debug_enabled = args.debug or os.environ.get('DEBUG', False)
+
+    if debug_enabled:
+        init_debugger()
+
+
+def init_debugger():
+    logging.basicConfig(level=logging.DEBUG,
+                        format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s',
+                        datefmt='%m-%d %H:%M:%S',
+                        filename='/tmp/leap.log',
+                        filemode='w')  # define a Handler which writes INFO messages or higher to the sys.stderr
+
+    formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')
+    console = logging.StreamHandler()
+    console.setLevel(logging.DEBUG)
+    console.setFormatter(formatter)
+    logging.getLogger('').addHandler(console)
+
+
+def init_twisted_logging():
+    log.startLogging(sys.stdout)
+    file_observer = PixelatedLogObserver(file(LOG_FILE, 'a'))
+    log.addObserver(file_observer.emit)
+
+
+class PixelatedLogObserver(log.FileLogObserver):
+
+    """ FileLogObserver with a customized format """
+    def emit(self, event):
+        text = log.textFromEventDict(event)
+
+        if text is None:
+            return
+
+        self.timeFormat = '[%Y-%m-%d %H:%M:%S]'
+        time_str = self.formatTime(event['time'])
+
+        fmt_dict = {'text': text.replace('\n', '\n\t')}
+        msg_str = log._safeFormat('%(text)s\n', fmt_dict)
+
+        logging.debug(str(event))
+
+        util.untilConcludes(self.write, time_str + ' ' + socket.gethostname() + ' ' + msg_str)
+        util.untilConcludes(self.flush)