Я пытаюсь создать базовый логгер, который будет окрашен без внешних пакетов,

# 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") должен вернуть:

enter image description here

13
wahwahwah 28 Авг 2017 в 19:14

3 ответа

Лучший ответ

Я решил сделать это по-другому и добавить цвет через саму строку с номером уровня:

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))

Выведет:

enter image description here

5
wahwahwah 28 Авг 2017 в 17:13

Да, вы можете проверить уровень регистратора по

level = logger.level
8
Milán Vásárhelyi 28 Авг 2017 в 16:18

Если вы используете корневой логгер, например, потому что вы назвали logging.basicConfig(), тогда вы можете использовать

import logging
logging.root.level

Например

if logging.DEBUG >= logging.root.level:
    # Do something
6
bcattle 27 Дек 2018 в 22:56