33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from typing import Optional
|
|
|
|
from sqlalchemy import select
|
|
|
|
from models import Status
|
|
from repositories.base import BaseRepository
|
|
from schemas.status import UpdateStatusSchema
|
|
|
|
|
|
class StatusRepository(BaseRepository):
|
|
async def get_all(self, board_id: int) -> list[Status]:
|
|
stmt = select(Status).where(
|
|
Status.is_deleted.is_(False), Status.board_id == board_id
|
|
)
|
|
result = await self.session.execute(stmt)
|
|
return list(result.scalars().all())
|
|
|
|
async def get_by_id(self, status_id: int) -> Optional[Status]:
|
|
stmt = select(Status).where(
|
|
Status.id == status_id, Status.is_deleted.is_(False)
|
|
)
|
|
result = await self.session.execute(stmt)
|
|
return result.scalar_one_or_none()
|
|
|
|
async def update(self, status: Status, data: UpdateStatusSchema) -> Status:
|
|
status.lexorank = data.lexorank if data.lexorank else status.lexorank
|
|
status.name = data.name if data.name else status.name
|
|
|
|
self.session.add(status)
|
|
await self.session.commit()
|
|
await self.session.refresh(status)
|
|
return status
|