refactor: mixins for services

services/deal.py

@@ -1,18 +1,27 @@
 import math
 
-from fastapi import HTTPException
-from sqlalchemy.ext.asyncio import AsyncSession
-
+from models import Deal
 from repositories import DealRepository
 from schemas.base import PaginationSchema, SortingSchema
 from schemas.deal import *
+from services.mixins import *
 
 
-class DealService:
+class DealService(
+    ServiceCreateMixin[Deal, CreateDealRequest, DealSchema],
+    ServiceUpdateMixin[Deal, UpdateDealRequest],
+    ServiceDeleteMixin[Deal],
+):
+    schema_class = DealSchema
+    entity_not_found_msg = "Сделка не найдена"
+    entity_deleted_msg = "Сделка успешно удалена"
+    entity_updated_msg = "Сделка успешно обновлена"
+    entity_created_msg = "Сделка успешно создана"
+
     def __init__(self, session: AsyncSession):
         self.repository = DealRepository(session)
 
-    async def get_deals(
+    async def get_all(
         self,
         pagination: PaginationSchema,
         sorting: SortingSchema,
@@ -36,27 +45,3 @@ class DealService:
                 total_pages=total_pages, total_items=total_items
             ),
         )
-
-    async def create_deal(self, request: CreateDealRequest) -> CreateDealResponse:
-        deal_id = await self.repository.create(request.entity)
-        deal = await self.repository.get_by_id(deal_id)
-        return CreateDealResponse(
-            entity=DealSchema.model_validate(deal),
-            message="Сделка успешно создана",
-        )
-
-    async def update_deal(self, deal_id: int, request: UpdateDealRequest):
-        deal = await self.repository.get_by_id(deal_id)
-        if not deal:
-            raise HTTPException(status_code=404, detail="Сделка не найдена")
-
-        await self.repository.update(deal, request.entity)
-        return UpdateDealResponse(message="Сделка успешно обновлена")
-
-    async def delete_deal(self, deal_id: int) -> DeleteDealResponse:
-        deal = await self.repository.get_by_id(deal_id)
-        if not deal:
-            raise HTTPException(status_code=404, detail="Сделка не найдена")
-
-        await self.repository.delete(deal, True)
-        return DeleteDealResponse(message="Сделка успешно удалена")