Improve logging implementation (#2347)

This commit is contained in:
foxfire52
2024-11-15 07:38:51 +01:00
committed by GitHub
parent 0d8e4ffa2c
commit f38428ab23
9 changed files with 46 additions and 31 deletions

View File

@@ -1,6 +1,6 @@
import sys,logging
from loguru import logger
#from loguru import logger
def __exception_handle(e_type, e_value, e_traceback):
if issubclass(e_type, KeyboardInterrupt):
@@ -9,24 +9,24 @@ def __exception_handle(e_type, e_value, e_traceback):
sys.__excepthook__(e_type, e_value, e_traceback)
class __InterceptHandler(logging.Handler):
def emit(self, record):
try:
level = logger.level(record.levelname).name
except ValueError:
level = record.levelno
#class __InterceptHandler(logging.Handler):
# def emit(self, record):
# try:
# level = logger.level(record.levelname).name
# except ValueError:
# level = record.levelno
#
# frame, depth = logging.currentframe(), 2
# while frame.f_code.co_filename == logging.__file__:
# frame = frame.f_back
# depth += 1
frame, depth = logging.currentframe(), 2
while frame.f_code.co_filename == logging.__file__:
frame = frame.f_back
depth += 1
logger.opt(depth=depth, exception=record.exc_info).log(
level, record.getMessage()
)
# logger.opt(depth=depth, exception=record.exc_info).log(
# level, record.getMessage()
# )
def hook_except_handle():
sys.excepthook = __exception_handle
def hook_logging(**kwargs):
logging.basicConfig(handlers=[__InterceptHandler()], **kwargs)
#def hook_logging(**kwargs):
# logging.basicConfig(handlers=[__InterceptHandler()], **kwargs)