logging 模块和 colorlog 模块

最新推荐文章于 2024-08-24 09:14:53 发布

江南野栀子

最新推荐文章于 2024-08-24 09:14:53 发布

阅读量955

点赞数

分类专栏： # Python 第三方模块文章标签： javascript 开发语言 ecmascript

本文链接：https://blog.csdn.net/u010701274/article/details/127068859

版权

Python 第三方模块专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一个结合logging和colorlog模块实现的日志管理系统。该系统不仅能在控制台输出带有颜色的日志，还能将日志信息动态地保存到指定文件中。通过自定义Logger，实现了灵活的日志级别控制和格式配置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近写的自动化工具需要日志，使用了 logging 模块和 colorlog 模块，塑造了颜色丰富的控制台输出，非常喜欢。

我在代码中，通过对 logging 模块的 logger 进行自己的定制，得到我想要的 logger，既能在控制台彩色输出，也能在我指定的文件中动态存储日志信息。

import os
import sys
import platform
import sys,ctypes
import logging
import colorlog
from smb.SMBConnection import SMBConnection

Log_File = 'athena_log.log'
exit_word = r' Athena run exit! Athena run exit! Athena run exit!'.center(115,"*")
Athena_Version = "0.9_20220916"
GUI_CMD = "CMD"

Debug = False


def isadmin():
    if(ctypes.windll.shell32.IsUserAnAdmin()==1):
        return(True)
    else:
        return(False)

class LogHandler(object):
    def __init__(self, filename, level=logging.INFO):
        self.logger = logging.getLogger(filename)
        self.log_colors_config = {
            'DEBUG': 'cyan',
            'INFO': 'green',
            'WARNING': 'yellow',
            'ERROR': 'red',
            'CRITICAL': 'red',
        }
        formatter = colorlog.ColoredFormatter(
            '%(log_color)s%(asctime)s  %(filename)s [line:%(lineno)d] %(levelname)s: %(message)s',
            log_colors=self.log_colors_config)

        # 设置日志级别
        self.logger.setLevel(level)
        # 往屏幕上输出
        console_handler = logging.StreamHandler()

        # 输出到日志文件
        if(isadmin()):
            file_handler = logging.FileHandler(filename=filename, mode='a', encoding='utf8')
            file_formatter = logging.Formatter('%(asctime)s  %(filename)s [line:%(lineno)d] %(levelname)s: %(message)s')
            # 设置写入文件的格式
            file_handler.setFormatter(file_formatter)

        # 设置屏幕上显示的格式
        console_handler.setFormatter(formatter)


        # 把对象加到logger里
        self.logger.addHandler(console_handler)
        if(isadmin()):
            self.logger.addHandler(file_handler)

my_loghandler = LogHandler(Log_File, level=logging.DEBUG)


def get_logger():
    return (my_loghandler)

下面是使用的情况