utils/functest_logger.py

   1 #!/usr/bin/env python
   2 #
   3 # jose.lausuch@ericsson.com
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Logging levels:
  10 #  Level     Numeric value
  11 #  CRITICAL  50
  12 #  ERROR     40
  13 #  WARNING   30
  14 #  INFO      20
  15 #  DEBUG     10
  16 #  NOTSET    0
  17 #
  18 # Usage:
  19 #  import functest_logger as fl
  20 #  logger = fl.Logger("script_name").getLogger()
  21 #  logger.info("message to be shown with - INFO - ")
  22 #  logger.debug("message to be shown with - DEBUG -")
  23
  24 import logging
  25 import os
  26
  27
  28 class Logger:
  29     def __init__(self, logger_name):
  30
  31         self.logger = logging.getLogger(logger_name)
  32         self.logger.setLevel(logging.DEBUG)
  33
  34         ch = logging.StreamHandler()
  35         formatter = logging.Formatter('%(asctime)s - %(name)s - '
  36                                       '%(levelname)s - %(message)s')
  37         ch.setFormatter(formatter)
  38
  39         CI_DEBUG = os.getenv('CI_DEBUG')
  40
  41         if CI_DEBUG.lower() == "true":
  42             ch.setLevel(logging.DEBUG)
  43         else:
  44             ch.setLevel(logging.INFO)
  45
  46         self.logger.addHandler(ch)
  47
  48     def getLogger(self):
  49         return self.logger