._.

app/core/usecases/new_total_order_service.py

@@ -26,7 +26,7 @@ class NewTotalOrderService:
         self.accessory_repository = AccessoriesRepository(db)
         self.truck_repository = TrucksRepository(db)
 
-    def total_calculate(self):
+    def total_calculate(self, current_user):
 
         accessories = self.accessory_repository.get_with_cities_all()
         cities = self.city_repository.get_all_with_federal_district()
@@ -42,6 +42,8 @@ class NewTotalOrderService:
                 count_robots=2000,
                 deadline=datetime.now(),
                 price=0,
+                user_id=current_user.id,
+                status_id=1,
             )
             session.add(new_total_order)
             session.flush()

app/infrastructure/fastapi/total_order_routes.py

@@ -58,7 +58,7 @@ def delete_total_order(total_order_id: int, db: Session = Depends(get_db),
 
 
 @router.post("/total-orders/calculate")
-def calculate_total_order(db: Session = Depends(get_db)):
+def calculate_total_order(db: Session = Depends(get_db), current_user: User = Depends(verify_token)):
     service = NewTotalOrderService(db)
-    service.total_calculate()
+    service.total_calculate(current_user)
     return "OK"