🐛 fix log module name error

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

View File

@ -47,6 +47,10 @@ class Filter:
self.level = "DEBUG"
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]
levelno = logger.level(self.level).no
return record["level"].no >= levelno