Merging 18 commit(s) from Chromium's toolchain-utils am: 0ae38c8498 am: 2a19d36a82 am: e617e3393dr_aml_301500702android-mainline-12.0.0_r55android-mainline-11.0.0_r9android-mainline-11.0.0_r8android-mainline-11.0.0_r7android-mainline-11.0.0_r6android-mainline-11.0.0_r5android-mainline-11.0.0_r45android-mainline-11.0.0_r44android-mainline-11.0.0_r43android-mainline-11.0.0_r42android-mainline-11.0.0_r41android-mainline-11.0.0_r40android-mainline-11.0.0_r4android-mainline-11.0.0_r39android-mainline-11.0.0_r38android-mainline-11.0.0_r37android-mainline-11.0.0_r36android-mainline-11.0.0_r35android-mainline-11.0.0_r34android-mainline-11.0.0_r33android-mainline-11.0.0_r32android-mainline-11.0.0_r31android-mainline-11.0.0_r30android-mainline-11.0.0_r3android-mainline-11.0.0_r29android-mainline-11.0.0_r28android-mainline-11.0.0_r27android-mainline-11.0.0_r26android-mainline-11.0.0_r25android-mainline-11.0.0_r24android-mainline-11.0.0_r23android-mainline-11.0.0_r22android-mainline-11.0.0_r21android-mainline-11.0.0_r20android-mainline-11.0.0_r2android-mainline-11.0.0_r19android-mainline-11.0.0_r18android-mainline-11.0.0_r17android-mainline-11.0.0_r16android-mainline-11.0.0_r15android-mainline-11.0.0_r14android-mainline-11.0.0_r13android-mainline-11.0.0_r12android-mainline-11.0.0_r10android-mainline-11.0.0_r1android-11.0.0_r48android-11.0.0_r47android-11.0.0_r46android-11.0.0_r45android-11.0.0_r44android-11.0.0_r43android-11.0.0_r42android-11.0.0_r41android-11.0.0_r40android-11.0.0_r39android-11.0.0_r38android-11.0.0_r37android-11.0.0_r36android-11.0.0_r35android-11.0.0_r34android-11.0.0_r33android-11.0.0_r32android-11.0.0_r31android-11.0.0_r30android-11.0.0_r29android-11.0.0_r28android-11.0.0_r27android-11.0.0_r26android-11.0.0_r24android-11.0.0_r23android-11.0.0_r22android-11.0.0_r21android-11.0.0_r20android-11.0.0_r19android-11.0.0_r18android-11.0.0_r16android11-qpr3-s1-releaseandroid11-qpr3-releaseandroid11-qpr2-releaseandroid11-qpr1-s2-releaseandroid11-qpr1-s1-releaseandroid11-qpr1-releaseandroid11-qpr1-d-s1-releaseandroid11-qpr1-d-releaseandroid11-qpr1-c-releaseandroid11-mainline-tethering-releaseandroid11-mainline-sparse-2021-jan-releaseandroid11-mainline-sparse-2020-dec-releaseandroid11-mainline-releaseandroid11-mainline-permission-releaseandroid11-mainline-os-statsd-releaseandroid11-mainline-networkstack-releaseandroid11-mainline-media-swcodec-releaseandroid11-mainline-media-releaseandroid11-mainline-extservices-releaseandroid11-mainline-documentsui-releaseandroid11-mainline-conscrypt-releaseandroid11-mainline-cellbroadcast-releaseandroid11-mainline-captiveportallogin-releaseandroid11-devandroid11-d2-releaseandroid11-d1-b-release

Change-Id: I3f25c7ee034b2e20e37ed941b8eae24eec7043eb

1 files changed, 144 insertions, 0 deletions

diff --git a/deprecated/automation/common/logger.py b/deprecated/automation/common/logger.py
new file mode 100644
index 00000000..4aeee052
--- /dev/null
+++ b/deprecated/automation/common/logger.py
@@ -0,0 +1,144 @@
+# Copyright 2010 Google Inc. All Rights Reserved.
+
+from itertools import chain
+import gzip
+import logging
+import logging.handlers
+import time
+import traceback
+
+
+def SetUpRootLogger(filename=None, level=None, display_flags={}):
+  console_handler = logging.StreamHandler()
+  console_handler.setFormatter(CustomFormatter(AnsiColorCoder(), display_flags))
+  logging.root.addHandler(console_handler)
+
+  if filename:
+    file_handler = logging.handlers.RotatingFileHandler(
+        filename,
+        maxBytes=10 * 1024 * 1024,
+        backupCount=9,
+        delay=True)
+    file_handler.setFormatter(CustomFormatter(NullColorCoder(), display_flags))
+    logging.root.addHandler(file_handler)
+
+  if level:
+    logging.root.setLevel(level)
+
+
+class NullColorCoder(object):
+
+  def __call__(self, *args):
+    return ''
+
+
+class AnsiColorCoder(object):
+  CODES = {'reset': (0,),
+           'bold': (1, 22),
+           'italics': (3, 23),
+           'underline': (4, 24),
+           'inverse': (7, 27),
+           'strikethrough': (9, 29),
+           'black': (30, 40),
+           'red': (31, 41),
+           'green': (32, 42),
+           'yellow': (33, 43),
+           'blue': (34, 44),
+           'magenta': (35, 45),
+           'cyan': (36, 46),
+           'white': (37, 47)}
+
+  def __call__(self, *args):
+    codes = []
+
+    for arg in args:
+      if arg.startswith('bg-') or arg.startswith('no-'):
+        codes.append(self.CODES[arg[3:]][1])
+      else:
+        codes.append(self.CODES[arg][0])
+
+    return '\033[%sm' % ';'.join(map(str, codes))
+
+
+class CustomFormatter(logging.Formatter):
+  COLORS = {'DEBUG': ('white',),
+            'INFO': ('green',),
+            'WARN': ('yellow', 'bold'),
+            'ERROR': ('red', 'bold'),
+            'CRIT': ('red', 'inverse', 'bold')}
+
+  def __init__(self, coder, display_flags={}):
+    items = []
+
+    if display_flags.get('datetime', True):
+      items.append('%(asctime)s')
+    if display_flags.get('level', True):
+      items.append('%(levelname)s')
+    if display_flags.get('name', True):
+      items.append(coder('cyan') + '[%(threadName)s:%(name)s]' + coder('reset'))
+    items.append('%(prefix)s%(message)s')
+
+    logging.Formatter.__init__(self, fmt=' '.join(items))
+
+    self._coder = coder
+
+  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)
+
+  def formatLevelName(self, record):
+    if record.levelname in ['WARNING', 'CRITICAL']:
+      levelname = record.levelname[:4]
+    else:
+      levelname = record.levelname
+
+    return ''.join([self._coder(*self.COLORS[levelname]), levelname,
+                    self._coder('reset')])
+
+  def formatMessagePrefix(self, record):
+    try:
+      return ' %s%s:%s ' % (self._coder('black', 'bold'), record.prefix,
+                            self._coder('reset'))
+    except AttributeError:
+      return ''
+
+  def format(self, record):
+    if record.exc_info:
+      if not record.exc_text:
+        record.exc_text = self.formatException(record.exc_info)
+    else:
+      record.exc_text = ''
+
+    fmt = record.__dict__.copy()
+    fmt.update({'levelname': self.formatLevelName(record),
+                'asctime': self.formatTime(record),
+                'prefix': self.formatMessagePrefix(record)})
+
+    s = []
+
+    for line in chain(record.getMessage().splitlines(),
+                      record.exc_text.splitlines()):
+      fmt['message'] = line
+
+      s.append(self._fmt % fmt)
+
+    return '\n'.join(s)
+
+
+class CompressedFileHandler(logging.FileHandler):
+
+  def _open(self):
+    return gzip.open(self.baseFilename + '.gz', self.mode, 9)
+
+
+def HandleUncaughtExceptions(fun):
+  """Catches all exceptions that would go outside decorated fun scope."""
+
+  def _Interceptor(*args, **kwargs):
+    try:
+      return fun(*args, **kwargs)
+    except StandardError:
+      logging.exception('Uncaught exception:')
+
+  return _Interceptor