crappy

schemas/deal.py

@@ -1,8 +1,9 @@
 import datetime
 from typing import List
 
-from schemas.base import CustomModelCamel
+from schemas.base import CustomModelCamel, OkMessageSchema
 from schemas.client import ClientSchema
+from schemas.service import ServiceSchema
 
 
 # region Entities
@@ -23,10 +24,20 @@ class DealSummary(CustomModelCamel):
 
 
 class DealServiceSchema(CustomModelCamel):
-    id: int
+    service: ServiceSchema
     quantity: int
 
 
+class DealSchema(CustomModelCamel):
+    id: int
+    name: str
+    client_id: int
+    created_at: datetime.datetime
+    current_status: int
+    services: List[DealServiceSchema]
+    # total_price: int
+
+
 # endregion Entities
 
 # region Requests
@@ -56,6 +67,36 @@ class DealAddServicesRequest(CustomModelCamel):
     services: list[DealServiceSchema]
 
 
+class DealGetAllResponse(CustomModelCamel):
+    deals: List[DealSchema]
+
+
+class DealUpdateServiceQuantityRequest(CustomModelCamel):
+    deal_id: int
+    service_id: int
+    quantity: int
+
+
+class DealAddServiceRequest(CustomModelCamel):
+    deal_id: int
+    service_id: int
+    quantity: int
+
+
+class DealDeleteServiceRequest(CustomModelCamel):
+    deal_id: int
+    service_id: int
+
+
+class DealDeleteServicesResponse(OkMessageSchema):
+    pass
+
+
+class DealDeleteServicesRequest(CustomModelCamel):
+    deal_id: int
+    service_ids: List[int]
+
+
 # endregion Requests
 
 # region Responses
@@ -79,4 +120,17 @@ class DealSummaryResponse(CustomModelCamel):
 class DealAddServicesResponse(CustomModelCamel):
     ok: bool
     message: str
+
+
+class DealUpdateServiceQuantityResponse(CustomModelCamel):
+    ok: bool
+    message: str
+
+
+class DealAddServiceResponse(OkMessageSchema):
+    pass
+
+
+class DealDeleteServiceResponse(OkMessageSchema):
+    pass
 # endregion Responses