🚸 优化log样式

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