Skip to content

Commit c845589

Browse files
committed
修复日志问题,防止误删其他模块的日志
1 parent eb98f79 commit c845589

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

python/src/common/base_log.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,12 @@ def __init__(
3131
compression: Optional[str] = None, # 新增压缩功能
3232
is_backtrace: bool = True,
3333
):
34-
# 移除 loguru 默认的 stderr handler,避免日志重复打印
34+
# 只移除 loguru 默认的 stderr handler (ID 0),避免删除其他模块已添加的 handlers
3535
if not Log._default_handler_removed:
36-
logger.remove()
36+
try:
37+
logger.remove(0) # 只移除默认 handler
38+
except ValueError:
39+
pass # 默认 handler 可能已被移除
3740
Log._default_handler_removed = True
3841

3942
self.is_backtrace = is_backtrace
@@ -56,7 +59,7 @@ def __init__(
5659
colorize=colorful,
5760
backtrace=True,
5861
enqueue=True,
59-
filter=lambda record: record["extra"]["task"] == filename,
62+
filter=lambda record: record["extra"].get("task") == filename,
6063
)
6164

6265
# 文件输出配置
@@ -70,7 +73,7 @@ def __init__(
7073
retention=f"{backup_count} days",
7174
compression=compression,
7275
enqueue=True,
73-
filter=lambda record: record["extra"]["task"] == filename,
76+
filter=lambda record: record["extra"].get("task") == filename,
7477
)
7578

7679
def _formatter(self, record):

0 commit comments

Comments
 (0)