from controllers.base_controller import BaseController
from models.colaboradores import Colaborador
from models.capitanes import Capitan
from models.coordinadores import Coordinador
from sqlalchemy.orm import selectinload

class ColaboradoresController(BaseController):
    def __init__(self, session=None):
        super().__init__(model=Colaborador, session=session)

    def listar_colaboradores(self):
        db = self.session if self.session else self.get_db_session()
        try:
            query = self.query_activa(db).options(selectinload(Colaborador.capitan).selectinload(Capitan.coordinador).selectinload(Coordinador.area))
            return query.all()
        finally:
            if not self.session:
                db.close()

    def crear_colaborador(self, datos, user_context=None):
        def operacion(db):
            # Limpieza de datos jerárquicos redundantes
            datos.pop('id_area', None)
            nuevo = Colaborador(**datos)
            db.add(nuevo)
            return nuevo
        return self.ejecutar_transaccion(operacion, "Colaborador registrado exitosamente", user_context)

    def actualizar_colaborador(self, id, datos, user_context=None):
        def operacion(db):
            colaborador = self.query_activa(db).filter(Colaborador.id == id).first()
            if not colaborador: raise ValueError("Colaborador no encontrado.")
            datos.pop('id_area', None)
            for key, value in datos.items(): setattr(colaborador, key, value)
        return self.ejecutar_transaccion(operacion, "Colaborador actualizado.", user_context)

    def eliminar_colaborador(self, id, user_context=None):
        def operacion(db):
            colaborador = self.query_activa(db).filter(Colaborador.id == id).first()
            if not colaborador: raise ValueError("Colaborador no encontrado.")
            self.marcar_eliminado(colaborador, db)
        return self.ejecutar_transaccion(operacion, "Colaborador eliminado.", user_context)