From e5f382cdd51cd519445ebdc18b386ff98f6e9cfd Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Thu, 16 Dec 2021 23:27:07 +0800 Subject: =?UTF-8?q?=F0=9F=9A=B8=20=E4=BC=98=E5=8C=96log=E6=A0=B7=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/log.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/ATRI/log.py b/ATRI/log.py index 2c0301d..aaba2bd 100644 --- a/ATRI/log.py +++ b/ATRI/log.py @@ -21,11 +21,24 @@ log_format = ( ) +class LoguruNameDealer: + def __call__(self, record): + log_handle = record["name"] + if "nonebot.plugin.manager" in log_handle: + plugin_name = log_handle.split(".")[-1] + record["name"] = f"plugin.{plugin_name}" + else: + record["name"] = record["name"].split(".")[0] + + return record + + logger.remove() logger.add( sys.stdout, level="DEBUG" if BotSelfConfig.debug else "INFO", colorize=True, + filter=LoguruNameDealer(), format=log_format, ) -- cgit v1.2.3