ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-10-31 06:56:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 fix log module name error

Browse Source
This commit is contained in:
yanyongyu
2021-04-19 21:40:27 +08:00
parent c24d0def09
commit 64d51448b8
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
nonebot/log.py
View File

@@ -47,6 +47,10 @@ class Filter:
self.level = "DEBUG" self.level = "DEBUG"
def __call__(self, record): def __call__(self, record):
module = sys.modules.get(record["name"])
if module:
plugin_name = getattr(module, "__plugin_name__", record["name"])
record["name"] = plugin_name
record["name"] = record["name"].split(".")[0] record["name"] = record["name"].split(".")[0]
levelno = logger.level(self.level).no levelno = logger.level(self.level).no
return record["level"].no >= levelno return record["level"].no >= levelno

3
nonebot/plugin/manager.py
View File

@@ -215,6 +215,9 @@ class PluginLoader(SourceFileLoader):
return return
# really need? # really need?
# setattr(module, "__manager__", self.manager) # setattr(module, "__manager__", self.manager)
if self._plugin_token:
setattr(module, "__plugin_name__",
_current_plugin.get(self._plugin_token))
if self._export_token: if self._export_token:
setattr(module, "__export__", _export.get()) setattr(module, "__export__", _export.get())