from sqlalchemy.orm import selectinload

from models import DealTag
from models.project import Project
from repositories.module import ModuleRepository
from repositories.mixins import *
from schemas.project import CreateProjectSchema, UpdateProjectSchema


class ProjectRepository(
    RepCrudMixin[Project, CreateProjectSchema, UpdateProjectSchema]
):
    entity_class = Project
    entity_not_found_msg = "Проект не найден"

    def _apply_options(self, stmt: Select) -> Select:
        return stmt.options(
            selectinload(Project.boards),
            selectinload(Project.tags).joinedload(DealTag.tag_color),
        )

    def _process_get_all_stmt(self, stmt: Select) -> Select:
        return self._apply_options(stmt).order_by(Project.id)

    def _process_get_by_id_stmt(self, stmt: Select) -> Select:
        return self._apply_options(stmt)

    async def update(self, project: Project, data: UpdateProjectSchema) -> Project:
        if data.modules is not None:
            modules = data.modules
            module_ids = [module.id for module in modules]
            data.modules = await ModuleRepository(
                self.session
            ).get_by_ids(module_ids)

        return await self._apply_update_data_to_model(project, data, True)