python log封装

# _*_ coding:utf-8 _*_

import logging
import os
import sys
import time

log_path = os.path.dirname(sys.path[0]) + '/log_path'

class Log():
    def __init__(self):
        filename = 'test_' + time.strftime('%Y_%m_%d_%H%M%S') + '.log'   # 设置log名
        self.logname = os.path.join(log_path, filename)
        self.logger = logging.getLogger()
        self.logger.setLevel(logging.DEBUG)
        #设置日志输出格式
        self.formatter = logging.Formatter('%(asctime)s [%(levelname)s] [%(name)s:%(lineno)d] - %(message)s')

    def output(self, level, message):
        """
        :param level: 日志等级
        :param message: 日志需要打印的信息
        :return:
        """
        # send logging output to a disk file
        fh = logging.FileHandler(self.logname, 'a', encoding='utf-8')
        fh.setLevel(logging.DEBUG)
        fh.setFormatter(self.formatter)
        self.logger.addHandler(fh)

        # send logging output to streams
        ch = logging.StreamHandler()
        ch.setLevel(logging.DEBUG)
        ch.setFormatter(self.formatter)
        self.logger.addHandler(ch)

        if level == 'info':
            self.logger.info(message)
        elif level == 'debug':
            self.logger.debug(message)
        elif level == 'warn':
            self.logger.warn(message)
        elif level == 'error':
            self.logger.error(message)

        #防止重复打印
        self.logger.removeHandler(fh)
        self.logger.removeHandler(ch)

        fh.close()

    def info(self, message: object) -> object:
        self.output('info', message)

    def debug(self, message):
        self.output('debug', message)

    def warn(self, message):
        self.output('warn', message)

    def error(self, message):
        self.output('error', message)