summaryrefslogtreecommitdiff
path: root/ATRI/log.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/log.py')
-rw-r--r--ATRI/log.py13
1 files changed, 13 insertions, 0 deletions
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,
)