# -*- coding: utf-8 -*-
"""
Copyright (C) 2020-2024 LiteyukiStudio. All Rights Reserved
@Time : 2024/7/27 上午9:12
@Author : snowykami
@Email : snowykami@outlook.com
@File : log.py
@Software: PyCharm
"""
import sys
import loguru
from typing import TYPE_CHECKING
logger = loguru.logger
if TYPE_CHECKING:
# avoid sphinx autodoc resolve annotation failed
# because loguru module do not have `Logger` class actually
from loguru import Record
def default_filter(record: "Record"):
"""默认的日志过滤器,根据 `config.log_level` 配置改变日志等级。"""
log_level = record["extra"].get("nonebot_log_level", "INFO")
levelno = logger.level(log_level).no if isinstance(log_level, str) else log_level
return record["level"].no >= levelno
# DEBUG日志格式
debug_format: str = (
"{time:YYYY-MM-DD HH:mm:ss} "
"[{level.icon}] "
"<{name}.{module}.{function}:{line}> "
"{message}"
)
# 默认日志格式
default_format: str = (
"{time:MM-DD HH:mm:ss} "
"[{level.icon}] "
"<{name}> "
"{message}"
)
def get_format(level: str) -> str:
if level == "DEBUG":
return debug_format
else:
return default_format
logger = loguru.logger.bind()
def init_log(config: dict):
"""
在语言加载完成后执行
Returns:
"""
global logger
logger.remove()
logger.add(
sys.stdout,
level=0,
diagnose=False,
filter=default_filter,
format=get_format(config.get("log_level", "INFO")),
)
show_icon = config.get("log_icon", True)
# debug = lang.get("log.debug", default="==DEBUG")
# info = lang.get("log.info", default="===INFO")
# success = lang.get("log.success", default="SUCCESS")
# warning = lang.get("log.warning", default="WARNING")
# error = lang.get("log.error", default="==ERROR")
#
# logger.level("DEBUG", color="", icon=f"{'🐛' if show_icon else ''}{debug}")
# logger.level("INFO", color="", icon=f"{'ℹ️' if show_icon else ''}{info}")
# logger.level("SUCCESS", color="", icon=f"{'✅' if show_icon else ''}{success}")
# logger.level("WARNING", color="", icon=f"{'⚠️' if show_icon else ''}{warning}")
# logger.level("ERROR", color="", icon=f"{'⭕' if show_icon else ''}{error}")