47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
from typing import Optional
|
|
|
|
from sqlalchemy import select, func
|
|
|
|
from models import Status, Deal
|
|
from repositories.base import BaseRepository
|
|
from repositories.mixins import RepDeleteMixin, RepCreateMixin
|
|
from schemas.status import UpdateStatusSchema, CreateStatusSchema
|
|
|
|
|
|
class StatusRepository(
|
|
BaseRepository,
|
|
RepDeleteMixin[Status],
|
|
RepCreateMixin[Deal, CreateStatusSchema],
|
|
):
|
|
entity_class = Status
|
|
|
|
async def get_all(self, board_id: int) -> list[Status]:
|
|
stmt = (
|
|
select(Status)
|
|
.where(Status.is_deleted.is_(False), Status.board_id == board_id)
|
|
.order_by(Status.lexorank)
|
|
)
|
|
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 get_deals_count(self, status_id: int) -> int:
|
|
stmt = select(func.count(Deal.id)).where(Deal.status_id == status_id)
|
|
result = await self.session.execute(stmt)
|
|
return result.scalar()
|
|
|
|
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
|