继承logging.Logger

    1. #!/usr/bin/env python
    2. #-*- encoding: utf8 -*-
    3. import sys
    4. import logging
    7. class MyLogger(logging.Logger):
    8.     def __init__(self,
    9.                  name=None,
    10.                  level=logging.DEBUG,
    11.                  fmt=None,
    12.                  datefmt=None,
    13.                  logfile=None,
    14.                  filemode='w',
    15.                  stream=sys.stderr,
    16.                  **kwargs):
    18.         super(MyLogger, self).__init__(name, level)
    20.         self.fmt = fmt or '[%(asctime)s %(name)s %(funcName)s %(levelname)s %(threadName)s] %(message)s'
    21.         self.datefmt = datefmt or '%Y-%m-%d %H:%M:%S'
    22.         self.formatter = logging.Formatter(self.fmt, self.datefmt)
    24.         if logfile:
    25.             self._addFilehandler(logfile, filemode)
    26.         else:
    27.             self._addStreamHandler(stream)
    29.     def _addFilehandler(self, filename, filemode):
    31.         file_hdlr = logging.FileHandler(filename, filemode)
    32.         file_hdlr.setFormatter(self.formatter)
    33.         self.addHandler(file_hdlr)
    35.     def _addStreamHandler(self, stream):
    37.         stream_hdlr = logging.StreamHandler(stream)
    38.         stream_hdlr.setFormatter(self.formatter)
    39.         self.addHandler(stream_hdlr)
    42. if __name__ == '__main__':
    44.     logger = MyLogger()
    45.     logger.info('info message')
    47.     logger2 = MyLogger(logfile='out.log')
    48.     logger2.warn('warn message')
    49.     logger2.warn('warn message')