File size: 1,735 Bytes
d10e42a | 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 os
from collections import deque
from logging.handlers import RotatingFileHandler
if not os.path.exists("logs"):
os.makedirs("logs")
LOG_FORMAT = "%(asctime)s - %(name)s - %(levelname)s - %(filename)s:%(lineno)d - %(message)s"
LOG_FILE = "logs/app.log"
MAX_IN_MEMORY_LOGS = 2000
_recent_logs = deque(maxlen=MAX_IN_MEMORY_LOGS)
class InMemoryLogHandler(logging.Handler):
"""Keep recent log lines in memory for dashboard display."""
def emit(self, record):
try:
_recent_logs.append(self.format(record))
except Exception:
# Never let logging break business logic.
pass
def get_recent_logs(limit=400):
if limit <= 0:
return ""
return "\n".join(list(_recent_logs)[-limit:])
def setup_logger(name="app", level=logging.INFO):
logger = logging.getLogger(name)
logger.setLevel(level)
if logger.handlers:
return logger
formatter = logging.Formatter(LOG_FORMAT)
console_handler = logging.StreamHandler()
console_handler.setLevel(level)
console_handler.setFormatter(formatter)
logger.addHandler(console_handler)
memory_handler = InMemoryLogHandler()
memory_handler.setLevel(level)
memory_handler.setFormatter(formatter)
logger.addHandler(memory_handler)
try:
file_handler = RotatingFileHandler(
LOG_FILE,
maxBytes=5 * 1024 * 1024,
backupCount=3,
encoding="utf-8",
)
file_handler.setLevel(level)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
except Exception as exc:
logger.warning("Failed to initialize file logger: %s", exc)
return logger
|