python log日志（二）

ゝ一纸荒年。 2022-06-01 12:29 179阅读 0赞

上一篇文章已经介绍log简单使用方法，本文介绍下另一种用法，可以打印日志，同时记录日志文件，可以控制日志颜色

import inspect
    import logging
    import logging.handlers
    
    
    import os
    from colorama import Fore, Style
    
    
    
    
    logger_dict = {}
    
    
    
    
    class LyrebirdLogger:
        def __init__(self, name):
            file_fmt = '[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s'
            date_fmt = '%Y-%m-%d %H:%M:%S'
            level = logging.DEBUG
            log_file = os.path.join(os.path.join(os.path.expanduser('~'), '.yourtoolpath'), 'log.log')
            mode = 'a'
            encoding = 'UTF-8'
            max_bytes = 1024*1024*2
            back_count = 1
            file_handler = create_file_handler(level, file_fmt, date_fmt, log_file, mode, max_bytes, back_count, encoding)
            self.name = name
            self.logger = logging.getLogger(name)
            self.logger_prefix = get_logger_prefix()
            self.logger.setLevel(level)
            self.logger.addHandler(file_handler)
    
    
        def debug(self, *msg):
            print(tag_blue(self.logger_prefix), *msg)
            self.logger.debug(get_msg(tag_blue(self.logger_prefix), *msg))
    
    
        def info(self, *msg):
            print(tag_green(self.logger_prefix), *msg)
            self.logger.info(get_msg(tag_green(self.logger_prefix), *msg))
    
    
        def warning(self, *msg):
            print(tag_yellow(self.logger_prefix), *msg)
            self.logger.warning(get_msg(tag_yellow(self.logger_prefix), *msg))
    
    
        def error(self, *msg):
            print(tag_red(self.logger_prefix), *msg)
            self.logger.error(get_msg(tag_red(self.logger_prefix), *msg))
    
    
        def critical(self, *msg):
            print(tag_red(self.logger_prefix), *msg)
            self.logger.critical(get_msg(tag_red(self.logger_prefix), *msg))
    
    
    
    
    def tag_green(msg):
        if not msg:
            return ''
        return f'{Fore.GREEN}{msg}{Style.RESET_ALL}'
    
    
    
    
    def tag_yellow(msg):
        if not msg:
            return ''
        return f'{Fore.YELLOW}{msg}{Style.RESET_ALL}'
    
    
    
    
    def tag_blue(msg):
        if not msg:
            return ''
        return f'{Fore.BLUE}{msg}{Style.RESET_ALL}'
    
    
    
    
    def tag_red(msg):
        if not msg:
            return ''
        return f'{Fore.RED}{msg}{Style.RESET_ALL}'
    
    
    
    
    def tag_cyan(msg):
        if not msg:
            return ''
        return f'{Fore.CYAN}{msg}{Style.RESET_ALL}'
    
    
    
    
    def get_msg(*msg):
        func_name = get_current_func_name()
        line_num = get_current_line_num()
        message = '[%s] [%d] ' % (func_name, line_num)
        for i in range(len(msg)):
            message = message + str(msg[i]) + ' '
        return message
    
    
    
    
    def create_file_handler(level, fmt, date_fmt, log_file, mode, max_bytes, back_count, encoding):
        file_handler = logging.handlers.RotatingFileHandler(log_file, mode, max_bytes, back_count, encoding)
        file_handler.setLevel(level)
        formatter = logging.Formatter(fmt, date_fmt)
        file_handler.setFormatter(formatter)
        return file_handler
    
    
    
    
    def get_logger_prefix():
        """
        获取top_module_name,拼接日志使用
        :return: 最外层包名
        """
        stack = inspect.stack()
        caller_module = inspect.getmodule(stack[1].frame)
        caller_module_name = caller_module.__name__
        if caller_module_name.find('.') > 0:
            caller_top_module_name = caller_module_name.split('.')[0]
        else:
            caller_top_module_name = caller_module_name
        return caller_top_module_name
    
    
    
    
    def get_current_line_num():
        return inspect.stack()[3][2]
    
    
    
    
    def get_current_func_name():
        return inspect.stack()[3][3]
    
    
    
    
    def get_logger(name):
        logger = logger_dict.get(name)
        if not logger:
            logger = LyrebirdLogger(name)
            logger_dict[name] = logger
        return logger

需要日志时，调用即可 _logger = get_logger('mock_server') _logger.info('...')