crappy

schemas/client.py

@@ -3,6 +3,7 @@ from typing import List
 from schemas.base import CustomModelCamel, OkMessageSchema
 
 
+# region Entities
 class ClientDetailsSchema(CustomModelCamel):
     address: str | None = None
     phone_number: str | None = None
@@ -16,19 +17,37 @@ class ClientSchema(CustomModelCamel):
     details: ClientDetailsSchema | None = None
 
 
+# endregion
+
+# region Requests
 class ClientSearchRequest(CustomModelCamel):
     name: str
 
 
-class ClientSearchResponse(CustomModelCamel):
-    clients: List[ClientSchema]
-
-
 class ClientUpdateDetailsRequest(CustomModelCamel):
     client_id: int
     details: ClientDetailsSchema
 
 
+class ClientCreateRequest(CustomModelCamel):
+    data: ClientSchema
+
+
+class ClientUpdateRequest(CustomModelCamel):
+    data: ClientSchema
+
+
+class ClientDeleteRequest(CustomModelCamel):
+    client_id: int
+
+
+# endregion
+
+# region Responses
+class ClientSearchResponse(CustomModelCamel):
+    clients: List[ClientSchema]
+
+
 class ClientUpdateDetailsResponse(CustomModelCamel):
     ok: bool
 
@@ -41,21 +60,10 @@ class ClientCreateResponse(OkMessageSchema):
     pass
 
 
-class ClientCreateRequest(CustomModelCamel):
-    data: ClientSchema
-
-
-class ClientUpdateRequest(CustomModelCamel):
-    data: ClientSchema
-
-
 class ClientUpdateResponse(OkMessageSchema):
     pass
 
 
-class ClientDeleteRequest(CustomModelCamel):
-    client_id: int
-
-
 class ClientDeleteResponse(OkMessageSchema):
     pass
+# endregion

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