tests/logger.py

   1 ##############################################################################
   2 # Copyright (c) 2016 ZTE Corporation and others.
   3 #
   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 # Usage:
  10 #  import doctor_logger
  11 #  logger = doctor_logger.Logger("script_name").getLogger()
  12 #  logger.info("message to be shown with - INFO - ")
  13 #  logger.debug("message to be shown with - DEBUG -")
  14
  15 import logging
  16 import os
  17
  18
  19 class Logger(object):
  20     def __init__(self, logger_name):
  21
  22         CI_DEBUG = os.getenv('CI_DEBUG')
  23
  24         filename = '%s.log' % logger_name
  25         logging.basicConfig(filemode='w', filename=filename)
  26         self.logger = logging.getLogger(logger_name)
  27         self.logger.propagate = 0
  28         self.logger.setLevel(logging.DEBUG)
  29
  30         formatter = logging.Formatter('%(asctime)s %(filename)s %(lineno)d '
  31                                       '%(levelname)-6s %(message)s')
  32
  33         ch = logging.StreamHandler()
  34         ch.setFormatter(formatter)
  35         if CI_DEBUG is not None and CI_DEBUG.lower() == "true":
  36             ch.setLevel(logging.DEBUG)
  37         else:
  38             ch.setLevel(logging.INFO)
  39         self.logger.addHandler(ch)
  40
  41         file_handler = logging.FileHandler(filename)
  42         file_handler.setFormatter(formatter)
  43         file_handler.setLevel(logging.DEBUG)
  44         self.logger.addHandler(file_handler)
  45
  46     def getLogger(self):
  47         return self.logger