aboutsummaryrefslogtreecommitdiff
path: root/automation/common/logger.py
diff options
context:
space:
mode:
Diffstat (limited to 'automation/common/logger.py')
-rw-r--r--automation/common/logger.py21
1 files changed, 12 insertions, 9 deletions
diff --git a/automation/common/logger.py b/automation/common/logger.py
index efa1a904..4aeee052 100644
--- a/automation/common/logger.py
+++ b/automation/common/logger.py
@@ -1,5 +1,3 @@
-#!/usr/bin/python
-#
# Copyright 2010 Google Inc. All Rights Reserved.
from itertools import chain
@@ -17,7 +15,10 @@ def SetUpRootLogger(filename=None, level=None, display_flags={}):
if filename:
file_handler = logging.handlers.RotatingFileHandler(
- filename, maxBytes=10*1024*1024, backupCount=9, delay=True)
+ filename,
+ maxBytes=10 * 1024 * 1024,
+ backupCount=9,
+ delay=True)
file_handler.setFormatter(CustomFormatter(NullColorCoder(), display_flags))
logging.root.addHandler(file_handler)
@@ -26,12 +27,13 @@ def SetUpRootLogger(filename=None, level=None, display_flags={}):
class NullColorCoder(object):
+
def __call__(self, *args):
return ''
class AnsiColorCoder(object):
- CODES = {'reset': (0, ),
+ CODES = {'reset': (0,),
'bold': (1, 22),
'italics': (3, 23),
'underline': (4, 24),
@@ -82,8 +84,8 @@ class CustomFormatter(logging.Formatter):
def formatTime(self, record):
ct = self.converter(record.created)
- t = time.strftime("%Y-%m-%d %H:%M:%S", ct)
- return "%s.%02d" % (t, record.msecs / 10)
+ t = time.strftime('%Y-%m-%d %H:%M:%S', ct)
+ return '%s.%02d' % (t, record.msecs / 10)
def formatLevelName(self, record):
if record.levelname in ['WARNING', 'CRITICAL']:
@@ -96,8 +98,8 @@ class CustomFormatter(logging.Formatter):
def formatMessagePrefix(self, record):
try:
- return ' %s%s:%s ' % (
- self._coder('black', 'bold'), record.prefix, self._coder('reset'))
+ return ' %s%s:%s ' % (self._coder('black', 'bold'), record.prefix,
+ self._coder('reset'))
except AttributeError:
return ''
@@ -125,6 +127,7 @@ class CustomFormatter(logging.Formatter):
class CompressedFileHandler(logging.FileHandler):
+
def _open(self):
return gzip.open(self.baseFilename + '.gz', self.mode, 9)
@@ -136,6 +139,6 @@ def HandleUncaughtExceptions(fun):
try:
return fun(*args, **kwargs)
except StandardError:
- logging.exception("Uncaught exception:")
+ logging.exception('Uncaught exception:')
return _Interceptor