1 files changed, 65 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..a15413a0
--- /dev/null
+++ b/service/pixelated/config/logging_setup.py
@@ -0,0 +1,65 @@
+#
+# 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
+
+
+def init_logging(args):
+    debug_enabled = args.debug or os.environ.get('DEBUG', False)
+
+    logging.basicConfig(level=logging.DEBUG if debug_enabled else logging.WARNING,
+                        format='[%(asctime)s] ' + socket.gethostname() + ' %(name)-12s %(levelname)-8s %(message)s',
+                        datefmt='%m-%d %H:%M:%S',
+                        filemode='a')
+
+    if debug_enabled:
+        init_debugger()
+
+    log.startLogging(sys.stdout)
+
+
+def init_debugger():
+    formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')
+    console = logging.StreamHandler()
+    console.setLevel(logging.DEBUG)
+    console.setFormatter(formatter)
+    logging.getLogger('').addHandler(console)
+
+
+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)