mirror of
https://github.com/fusion44/blitz_api.git
synced 2026-08-13 11:52:45 +02:00
25e4be8441
If the environment variable BAPI_ENV_PATH is set, the config system will try to read configs from the given path instead of .env in pwd If no file is found .env in pwd will be used as a fallback Env variables will always override settings in .env files.
101 lines
3 KiB
Python
101 lines
3 KiB
Python
import logging
|
|
import sys
|
|
|
|
from loguru import logger
|
|
|
|
from app.api.config import config as dconfig
|
|
|
|
# Sourced from LNbits project:
|
|
# https://github.com/lnbits/lnbits/blob/841e8e7bbd61fb942a776d82ca0b6d03668eb524/lnbits/app.py#L285
|
|
|
|
# More info: https://loguru.readthedocs.io/en/stable/api/logger.html
|
|
|
|
|
|
def configure_logger() -> None:
|
|
level = dconfig("BAPI_LOG_LEVEL", default="INFO", cast=str)
|
|
log_file = dconfig("BAPI_LOG_FILE", default="", cast=str)
|
|
|
|
logger.remove()
|
|
formatter = Formatter(level)
|
|
logger.add(sys.stdout, level=level, format=formatter.format, colorize=True)
|
|
|
|
if log_file is not None and log_file != "":
|
|
logger.add(
|
|
"blitz_api.log",
|
|
level=level,
|
|
format=formatter.format,
|
|
rotation="10 MB",
|
|
retention="30 days",
|
|
compression="zip",
|
|
)
|
|
|
|
logging.getLogger("uvicorn").handlers = [InterceptHandler()]
|
|
logging.getLogger("uvicorn.access").handlers = [InterceptHandler()]
|
|
|
|
|
|
class Formatter:
|
|
def __init__(self, level: str):
|
|
self.padding = 0
|
|
self.fmt_default: str = (
|
|
"<green>{time:YYYY-MM-DD HH:mm:ss}</green> | "
|
|
+ "<level>{level.icon}</level> | "
|
|
+ "<level>{file}:{line}</level> | "
|
|
+ "<level>{message}</level>\n"
|
|
)
|
|
|
|
self.fmt_default_exception: str = (
|
|
self.fmt_default + "<level>{exception}</level>\n"
|
|
)
|
|
|
|
self.fmt_warning: str = (
|
|
"<yellow>{time:YYYY-MM-DD HH:mm:ss}</yellow> | "
|
|
+ "<level>{level.icon}</level> | "
|
|
+ "<level>{file}:{line}</level> | "
|
|
+ "<level>{message}</level>\n"
|
|
)
|
|
self.fmt_warning_exception: str = (
|
|
self.fmt_warning + "<level>{exception}</level>\n"
|
|
)
|
|
|
|
self.fmt_error: str = (
|
|
"<red>{time:YYYY-MM-DD HH:mm:ss}</red> | "
|
|
+ "<level>{level.icon}</level> | "
|
|
+ "<level>{file}:{line}</level> | "
|
|
+ "<level>{message}</level>\n"
|
|
)
|
|
|
|
self.fmt_error_exception: str = self.fmt_error + "<level>{exception}</level>\n"
|
|
|
|
def format(self, record):
|
|
level = record["level"]
|
|
e = record["exception"] is not None
|
|
|
|
if level.no < 30:
|
|
return self.fmt_default_exception if e else self.fmt_default
|
|
|
|
if level.no == 30:
|
|
return self.fmt_warning_exception if e else self.fmt_warning
|
|
|
|
return self.fmt_error_exception if e else self.fmt_error
|
|
|
|
|
|
class InterceptHandler(logging.Handler):
|
|
def emit(self, record):
|
|
try:
|
|
level = logger.level(record.levelname).name
|
|
except ValueError:
|
|
level = record.levelno
|
|
|
|
try:
|
|
logger.log(level, record.getMessage())
|
|
except TypeError as e:
|
|
logger.error(
|
|
f"""Unable to process log message: {e}
|
|
Name of the record:
|
|
{record.name}
|
|
Message of the record:
|
|
{record.msg}
|
|
Args of the record:
|
|
{record.args}
|
|
"""
|
|
)
|