85 lines
3.4 KiB
Python
85 lines
3.4 KiB
Python
import os
|
|
import logging
|
|
import json
|
|
import sys
|
|
from datetime import datetime, timezone
|
|
from context import get_request_id, get_current_user # Importamos las funciones
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
|
|
# ============================================================================
|
|
class JsonFormatter(logging.Formatter):
|
|
"""Formateador personalizado para exportar logs en formato JSON puro."""
|
|
EXTRA_FIELDS = ("http_method", "endpoint", "status_code")
|
|
|
|
def format(self, record):
|
|
user = get_current_user()
|
|
log_record = {
|
|
"timestamp": datetime.fromtimestamp(record.created, tz=timezone.utc).isoformat(),
|
|
"level": record.levelname,
|
|
"request_id": get_request_id(), # ⚡️ Inyección automática del ID de rastreo
|
|
"webUser": user["webUser"],
|
|
"emp_id": user["emp_id"],
|
|
"full_name": user["full_name"],
|
|
"message": record.getMessage(),
|
|
"module": record.module,
|
|
"function": record.funcName,
|
|
"line": record.lineno
|
|
}
|
|
for field in self.EXTRA_FIELDS:
|
|
if hasattr(record, field):
|
|
log_record[field] = getattr(record, field)
|
|
# Si el log incluye excepciones, las añade al JSON
|
|
if record.exc_info:
|
|
log_record["exception"] = self.formatException(record.exc_info)
|
|
return json.dumps(log_record, ensure_ascii=False)
|
|
|
|
|
|
# ============================================================================
|
|
def setup_logging(environment: str = "production"):
|
|
"""Configura el logger global de la aplicación."""
|
|
root_logger = logging.getLogger()
|
|
|
|
# Evita duplicar logs si la función se ejecuta más de una vez
|
|
if root_logger.hasHandlers():
|
|
root_logger.handlers.clear()
|
|
|
|
# Definir el nivel mínimo de captura (INFO por defecto)
|
|
root_logger.setLevel(logging.INFO)
|
|
|
|
# Handler para la consola (Standard Output)
|
|
console_handler = logging.StreamHandler(sys.stdout)
|
|
|
|
if environment == "production":
|
|
# En producción usamos JSON estructurado
|
|
console_handler.setFormatter(JsonFormatter())
|
|
else:
|
|
# En desarrollo usamos texto limpio y con colores opcionales
|
|
readable_formatter = logging.Formatter(
|
|
fmt="%(asctime)s [%(levelname)s] %(module)s.%(funcName)s:%(lineno)d - %(message)s",
|
|
datefmt="%Y-%m-%d %H:%M:%S"
|
|
)
|
|
console_handler.setFormatter(readable_formatter)
|
|
|
|
root_logger.addHandler(console_handler)
|
|
|
|
# 2. CREAR CARPETA Y ARCHIVO FÍSICO EN DISCO
|
|
log_dir = "logs"
|
|
if not os.path.exists(log_dir):
|
|
os.makedirs(log_dir) # Crea la carpeta 'logs/' si no existe
|
|
|
|
# Usamos 'RotatingFileHandler' para que el archivo no crezca infinitamente.
|
|
# Cuando llegue a 5MB, lo cierra, lo renombra a backend.log.1 y crea uno nuevo.
|
|
file_handler = RotatingFileHandler(
|
|
filename=os.path.join(log_dir, "backend.log"),
|
|
maxBytes=5 * 1024 * 1024, # 5 Megabytes
|
|
backupCount=3, # Mantener máximo 3 archivos viejos de respaldo
|
|
encoding="utf-8"
|
|
)
|
|
file_handler.setFormatter(JsonFormatter())
|
|
root_logger.addHandler(file_handler)
|
|
|
|
|
|
# Silenciar logs ruidosos de librerías externas (opcional)
|
|
logging.getLogger("uvicorn.access").setLevel(logging.WARNING)
|
|
# ============================================================================ |