feat: deal's status and board update
This commit is contained in:
@ -1,6 +1,7 @@
|
||||
from typing import Optional
|
||||
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import joinedload
|
||||
|
||||
from models import Deal, CardStatusHistory, Board
|
||||
from repositories.base import BaseRepository
|
||||
@ -22,7 +23,11 @@ class DealRepository(BaseRepository):
|
||||
id: Optional[int],
|
||||
name: Optional[str],
|
||||
) -> tuple[list[Deal], int]:
|
||||
stmt = select(Deal).where(Deal.is_deleted.is_(False))
|
||||
stmt = (
|
||||
select(Deal)
|
||||
.options(joinedload(Deal.status), joinedload(Deal.board))
|
||||
.where(Deal.is_deleted.is_(False))
|
||||
)
|
||||
|
||||
if id:
|
||||
stmt = stmt.where(Deal.id == id)
|
||||
@ -49,7 +54,11 @@ class DealRepository(BaseRepository):
|
||||
return list(result.scalars().all()), total_items
|
||||
|
||||
async def get_by_id(self, deal_id: int) -> Optional[Deal]:
|
||||
stmt = select(Deal).where(Deal.id == deal_id, Deal.is_deleted.is_(False))
|
||||
stmt = (
|
||||
select(Deal)
|
||||
.options(joinedload(Deal.status), joinedload(Deal.board))
|
||||
.where(Deal.id == deal_id, Deal.is_deleted.is_(False))
|
||||
)
|
||||
result = await self.session.execute(stmt)
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
@ -58,11 +67,12 @@ class DealRepository(BaseRepository):
|
||||
self.session.add(deal)
|
||||
await self.session.commit()
|
||||
await self.session.refresh(deal)
|
||||
return deal
|
||||
return await self.get_by_id(deal.id)
|
||||
|
||||
async def update(self, deal: Deal, data: UpdateDealSchema) -> Deal:
|
||||
deal.lexorank = data.lexorank if data.lexorank else deal.lexorank
|
||||
deal.name = data.name if data.name else deal.name
|
||||
deal.board_id = data.board_id if data.board_id else deal.board_id
|
||||
|
||||
if data.status_id and deal.status_id != data.status_id:
|
||||
deal.status_history.append(
|
||||
|
||||
Reference in New Issue
Block a user