from pymongo import MongoClient
from pymongo.collection import Collection
from core.config import ConfSettings

# Conexión a MongoDB para Reportes
mongo_client = MongoClient(ConfSettings.mongodb_url)
mongodb = mongo_client[ConfSettings.mongodb_db]

# Conexión a MongoDB para Notificaciones
mongo_client_notifications = MongoClient(ConfSettings.mongodb_notifications_url)
mongodb_notifications = mongo_client_notifications[ConfSettings.mongodb_notifications_db]

def get_reports_collection() -> Collection:
    """Obtiene la colección de reportes desde MongoDB"""
    return mongodb["reportes"]

def get_notifications_collection() -> Collection:
    """Obtiene la colección de notificaciones desde MongoDB"""
    return mongodb_notifications["notificaciones"]