mirror of https://github.com/xtekky/gpt4free.git
33 lines
893 B
Python
33 lines
893 B
Python
|
import sys,logging
|
||
|
|
||
|
from loguru import logger
|
||
|
|
||
|
def __exception_handle(e_type, e_value, e_traceback):
|
||
|
if issubclass(e_type, KeyboardInterrupt):
|
||
|
print('\nBye...')
|
||
|
sys.exit(0)
|
||
|
|
||
|
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
|
||
|
|
||
|
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()
|
||
|
)
|
||
|
|
||
|
def hook_except_handle():
|
||
|
sys.excepthook = __exception_handle
|
||
|
|
||
|
def hook_logging(**kwargs):
|
||
|
logging.basicConfig(handlers=[__InterceptHandler()], **kwargs)
|