mirror of
https://github.com/grillazz/fastapi-sqlalchemy-asyncpg.git
synced 2025-08-26 16:40:40 +03:00
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
import logging
|
|
import os
|
|
from logging.handlers import RotatingFileHandler
|
|
from pathlib import Path
|
|
|
|
import orjson
|
|
import structlog
|
|
from rich.console import Console
|
|
from rich.logging import RichHandler
|
|
from whenever._whenever import Instant
|
|
|
|
from app.utils.singleton import SingletonMeta
|
|
|
|
|
|
class AppLogger(metaclass=SingletonMeta):
|
|
_logger = None
|
|
|
|
def __init__(self):
|
|
self._logger = logging.getLogger(__name__)
|
|
|
|
def get_logger(self):
|
|
return self._logger
|
|
|
|
|
|
class RichConsoleHandler(RichHandler):
|
|
def __init__(self, width=200, style=None, **kwargs):
|
|
super().__init__(
|
|
console=Console(color_system="256", width=width, style=style, stderr=True),
|
|
**kwargs,
|
|
)
|
|
|
|
|
|
|
|
# TODO: merge this wrapper with the one in structlog under one hood of AppLogger
|
|
class BytesToTextIOWrapper:
|
|
def __init__(self, handler, encoding="utf-8"):
|
|
self.handler = handler
|
|
self.encoding = encoding
|
|
|
|
def write(self, b):
|
|
if isinstance(b, bytes):
|
|
self.handler.stream.write(b.decode(self.encoding))
|
|
else:
|
|
self.handler.stream.write(b)
|
|
self.handler.flush()
|
|
|
|
def flush(self):
|
|
self.handler.flush()
|
|
|
|
def close(self):
|
|
self.handler.close()
|
|
|
|
|
|
def setup_structlog() -> structlog.BoundLogger:
|
|
log_date = Instant.now().py_datetime().strftime("%Y%m%d")
|
|
log_path = Path(f"{log_date}_{os.getpid()}.log")
|
|
handler = RotatingFileHandler(
|
|
filename=log_path,
|
|
mode="a", # text mode
|
|
maxBytes=10 * 1024 * 1024,
|
|
backupCount=5,
|
|
encoding="utf-8"
|
|
)
|
|
file_like = BytesToTextIOWrapper(handler)
|
|
structlog.configure(
|
|
cache_logger_on_first_use=True,
|
|
wrapper_class=structlog.make_filtering_bound_logger(logging.INFO),
|
|
processors=[
|
|
structlog.contextvars.merge_contextvars,
|
|
structlog.processors.add_log_level,
|
|
structlog.processors.format_exc_info,
|
|
structlog.processors.TimeStamper(fmt="iso", utc=True),
|
|
structlog.processors.JSONRenderer(serializer=orjson.dumps),
|
|
],
|
|
logger_factory=structlog.BytesLoggerFactory(
|
|
file=file_like
|
|
)
|
|
)
|
|
return structlog.get_logger()
|