Feature: 优化调用栈识别 (#2644)

This commit is contained in:
Ju4tCode
2024-04-17 17:24:38 +08:00
committed by GitHub
parent 5e72461391
commit 30d3c1bbce
2 changed files with 13 additions and 4 deletions

View File

@ -13,6 +13,7 @@ FrontMatter:
"""
import sys
import inspect
import logging
from typing import TYPE_CHECKING
@ -45,6 +46,7 @@ logger: "Logger" = loguru.logger
# logger.addHandler(default_handler)
# https://loguru.readthedocs.io/en/stable/overview.html#entirely-compatible-with-standard-logging
class LoguruHandler(logging.Handler): # pragma: no cover
"""logging 与 loguru 之间的桥梁,将 logging 的日志转发到 loguru。"""
@ -54,8 +56,8 @@ class LoguruHandler(logging.Handler): # pragma: no cover
except ValueError:
level = record.levelno
frame, depth = sys._getframe(6), 6
while frame and frame.f_code.co_filename == logging.__file__:
frame, depth = inspect.currentframe(), 0
while frame and (depth == 0 or frame.f_code.co_filename == logging.__file__):
frame = frame.f_back
depth += 1