blob: e91dff76f9f88d4cc29a863a08db76f8b840ffca (
plain) (
blame)
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)
|
