g4f/api/_logging.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

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)