34 lines
995 B
Python
34 lines
995 B
Python
from sqlalchemy import Select
|
|
from sqlalchemy.orm import selectinload
|
|
|
|
from models.project import Project
|
|
from repositories.base import BaseRepository
|
|
from repositories.mixins import (
|
|
RepDeleteMixin,
|
|
RepCreateMixin,
|
|
GetByIdMixin,
|
|
GetAllMixin,
|
|
RepUpdateMixin,
|
|
)
|
|
from schemas.project import CreateProjectSchema, UpdateProjectSchema
|
|
|
|
|
|
class ProjectRepository(
|
|
BaseRepository,
|
|
GetAllMixin[Project],
|
|
RepDeleteMixin[Project],
|
|
RepCreateMixin[Project, CreateProjectSchema],
|
|
RepUpdateMixin[Project, UpdateProjectSchema],
|
|
GetByIdMixin[Project],
|
|
):
|
|
entity_class = Project
|
|
|
|
def _process_get_all_stmt(self, stmt: Select) -> Select:
|
|
return stmt.order_by(Project.id)
|
|
|
|
def _process_get_by_id_stmt(self, stmt: Select) -> Select:
|
|
return stmt.options(selectinload(Project.boards))
|
|
|
|
async def update(self, project: Project, data: UpdateProjectSchema) -> Project:
|
|
return await self._apply_update_data_to_model(project, data, True)
|