feat: create and delete status endpoints

repositories/status.py

@@ -1,10 +1,11 @@
+from datetime import datetime
 from typing import Optional
 
-from sqlalchemy import select
+from sqlalchemy import select, func
 
-from models import Status
+from models import Status, Deal
 from repositories.base import BaseRepository
-from schemas.status import UpdateStatusSchema
+from schemas.status import UpdateStatusSchema, CreateStatusSchema
 
 
 class StatusRepository(BaseRepository):
@@ -22,6 +23,20 @@ class StatusRepository(BaseRepository):
         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 create(self, data: CreateStatusSchema) -> Status:
+        status_data = data.model_dump()
+        status_data["created_at"] = datetime.now()
+        status = Status(**status_data)
+        self.session.add(status)
+        await self.session.commit()
+        await self.session.refresh(status)
+        return status
+
     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
@@ -30,3 +45,13 @@ class StatusRepository(BaseRepository):
         await self.session.commit()
         await self.session.refresh(status)
         return status
+
+    async def delete(self, status: Status, is_soft: bool):
+        if not is_soft:
+            await self.session.delete(status)
+            await self.session.commit()
+            return
+
+        status.is_deleted = True
+        self.session.add(status)
+        await self.session.commit()