import sys import loguru from typing import TYPE_CHECKING from .config import load_from_yaml from .language import Language, get_default_lang, get_system_lang_code logger = loguru.logger if TYPE_CHECKING: # avoid sphinx autodoc resolve annotation failed # because loguru module do not have `Logger` class actually from loguru import Logger, 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} {time:HH:mm:ss} " "[{level.icon}] " "<{name}.{module}.{function}:{line}> " "{message}" ) # 默认日志格式 default_format: str = ( "{time:MM-DD} {time: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(): global logger config = load_from_yaml("config.yml") 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) lang = Language(config.get("default_language", get_system_lang_code())) 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}")