diff options
| author | Kyomotoi <[email protected]> | 2021-12-16 23:27:07 +0800 |
|---|---|---|
| committer | Kyomotoi <[email protected]> | 2021-12-16 23:27:07 +0800 |
| commit | e5f382cdd51cd519445ebdc18b386ff98f6e9cfd (patch) | |
| tree | 21fc60d78e72b0b8af9eb88c6b401ed0670757a1 /ATRI/log.py | |
| parent | e19782340dc7c19968acda55f9767b9a5dbd062c (diff) | |
| download | ATRI-e5f382cdd51cd519445ebdc18b386ff98f6e9cfd.tar.gz ATRI-e5f382cdd51cd519445ebdc18b386ff98f6e9cfd.tar.bz2 ATRI-e5f382cdd51cd519445ebdc18b386ff98f6e9cfd.zip | |
🚸 优化log样式
Diffstat (limited to 'ATRI/log.py')
| -rw-r--r-- | ATRI/log.py | 13 |
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, ) |