mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-10-30 03:01:50 +08:00
Improve logging implementation (#2347)
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user