# -*- coding: utf-8 -*-

import logging
import sys

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)
