Spaces:
Runtime error
Runtime error
File size: 1,835 Bytes
0a1b571 |
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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import logging
import re
import sys
from datetime import timedelta
from pathlib import Path
import sentry_sdk.integrations.logging as sentry
from loguru import logger as _logger
from hibiapi.utils.config import Config
LOG_FILE = Config["log"]["file"].get_optional(Path)
LOG_LEVEL = Config["log"]["level"].as_str().strip().upper()
LOG_FORMAT = Config["log"]["format"].as_str().strip()
class LoguruHandler(logging.Handler):
_tag_escape_re = re.compile(r"</?((?:[fb]g\s)?[^<>\s]*)>")
@classmethod
def escape_tag(cls, string: str) -> str:
return cls._tag_escape_re.sub(r"\\\g<0>", string)
def emit(self, record: logging.LogRecord):
try:
level = logger.level(record.levelname).name
except ValueError:
level = record.levelno
frame, depth, message = logging.currentframe(), 2, record.getMessage()
while frame.f_code.co_filename == logging.__file__: # type: ignore
frame = frame.f_back # type: ignore
depth += 1
logger.opt(depth=depth, exception=record.exc_info, colors=True).log(
level, f"<e>{self.escape_tag(message)}</e>"
)
logger = _logger.opt(colors=True)
logger.remove()
logger.add(
sys.stdout,
level=LOG_LEVEL,
format=LOG_FORMAT,
filter=lambda record: record["level"].no < logging.WARNING,
)
logger.add(
sys.stderr,
level=LOG_LEVEL,
filter=lambda record: record["level"].no >= logging.WARNING,
format=LOG_FORMAT,
)
logger.add(sentry.BreadcrumbHandler(), level=LOG_LEVEL)
logger.add(sentry.EventHandler(), level="ERROR")
if LOG_FILE is not None:
LOG_FILE.parent.mkdir(parents=True, exist_ok=True)
logger.add(
str(LOG_FILE),
level=LOG_LEVEL,
encoding="utf-8",
rotation=timedelta(days=1),
)
logger.level(LOG_LEVEL)
|