refactor: repository create mixin

This commit is contained in:
2025-09-05 00:04:09 +04:00
parent c632fb8037
commit e5be35be35
9 changed files with 54 additions and 43 deletions

View File

@ -5,11 +5,17 @@ from sqlalchemy.orm import selectinload
from models.project import Project
from repositories.base import BaseRepository
from repositories.mixins import RepDeleteMixin
from repositories.mixins import RepDeleteMixin, RepCreateMixin
from schemas.project import CreateProjectSchema, UpdateProjectSchema
class ProjectRepository(BaseRepository, RepDeleteMixin[Project]):
class ProjectRepository(
BaseRepository,
RepDeleteMixin[Project],
RepCreateMixin[Project, CreateProjectSchema],
):
entity_class = Project
async def get_all(self) -> list[Project]:
stmt = select(Project).where(Project.is_deleted.is_(False)).order_by(Project.id)
result = await self.session.execute(stmt)
@ -24,13 +30,6 @@ class ProjectRepository(BaseRepository, RepDeleteMixin[Project]):
result = await self.session.execute(stmt)
return result.scalar_one_or_none()
async def create(self, data: CreateProjectSchema) -> Project:
project = Project(**data.model_dump())
self.session.add(project)
await self.session.commit()
await self.session.refresh(project)
return project
async def update(self, project: Project, data: UpdateProjectSchema) -> Project:
project.name = data.name if data.name else project.name