refactor: repository create mixin
This commit is contained in:
@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user