如何在python日志记录模块中获取当前日志级别

新手上路,请多包涵

我正在尝试创建一个基本的记录器,它将在没有外部包的情况下着色,

 # these have to be the first functions so I can use it in the logger settings
def create_log_name(log_path="{}/log", filename="zeus-log-{}.log"):
    if not os.path.exists(log_path.format(os.getcwd())):
        os.mkdir(log_path.format(os.getcwd()))
    find_file_amount = len(os.listdir(log_path.format(os.getcwd())))
    full_log_path = "{}/{}".format(log_path.format(os.getcwd()), filename.format(find_file_amount + 1))
    return full_log_path

def set_color_value(levelname):
    log_set = {
        "INFO": "\033[92m{}\033[0m",
        "WARNING": "\033[93m{}\033[0m",
        "DEBUG": "\033[94m{}\033[0m",
        "ERROR": "\033[91m{}\033[0m",
        "CRITICAL": "\033[91m{}\033[0m"
    }
    return log_set[levelname].format(levelname)

logger = logging.getLogger("zeus-log")
logger.setLevel(logging.DEBUG)
file_handler = logging.FileHandler(
    filename=create_log_name(), mode="a+"
)
file_handler.setLevel(logging.DEBUG)
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.DEBUG)
file_format = logging.Formatter(
    '%(asctime)s;%(name)s;%(levelname)s;%(message)s'
)
console_format = logging.Formatter(
    "[%(asctime)s {}] %(message)s".format(set_color_value()), "%H:%M:%S"
)
file_handler.setFormatter(file_format)
console_handler.setFormatter(console_format)
logger.addHandler(console_handler)
logger.addHandler(file_handler)

所以现在,我需要做的就是获取将在 logging.Formatter 中设置的当前日志级别,并将其发送到我的小函数:

 console_format = logging.Formatter(
    "[%(asctime)s {}] %(message)s".format(set_color_value()), "%H:%M:%S"
)

是否可以从日志包中获取当前日志级别?


例如,假设我传递了 logger.INFO("test") 我需要一种方法来获取它 INFO 作为字符串的一部分,从那里, set_color_value("INFO") 应该返回:

在此处输入图像描述

原文由 wahwahwah 发布,翻译遵循 CC BY-SA 4.0 许可协议

阅读 763
2 个回答

我决定以不同的方式执行此操作,并通过带有级别编号的字符串本身添加颜色:

 def set_color(org_string, level=None):
    color_levels = {
        10: "\033[36m{}\033[0m",       # DEBUG
        20: "\033[32m{}\033[0m",       # INFO
        30: "\033[33m{}\033[0m",       # WARNING
        40: "\033[31m{}\033[0m",       # ERROR
        50: "\033[7;31;31m{}\033[0m"   # FATAL/CRITICAL/EXCEPTION
    }
    if level is None:
        return color_levels[20].format(org_string)
    else:
        return color_levels[int(level)].format(org_string)

例如:

 logger.info(set_color("test"))
logger.debug(set_color("test", level=10))
logger.warning(set_color("test", level=30))
logger.error(set_color("test", level=40))
logger.fatal(set_color("test", level=50))

将输出:

在此处输入图像描述

原文由 wahwahwah 发布,翻译遵循 CC BY-SA 3.0 许可协议

如果您使用的是根记录器,例如因为您调用了 logging.basicConfig() 那么您可以使用

import logging
logging.root.level

例如

if logging.DEBUG >= logging.root.level:
    # Do something

原文由 bcattle 发布,翻译遵循 CC BY-SA 4.0 许可协议

推荐问题