From cae22d12031aed0b3786ecc6f868f72dced009c3 Mon Sep 17 00:00:00 2001
From: andrei <andrei@duvakin.ru>
Date: Sat, 5 Oct 2024 22:32:26 +0500
Subject: [PATCH] ._.

---
 app/core/usecases/new_total_order_service.py     | 4 +++-
 app/infrastructure/fastapi/total_order_routes.py | 4 ++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/app/core/usecases/new_total_order_service.py b/app/core/usecases/new_total_order_service.py
index 5945e24..44d487b 100644
--- a/app/core/usecases/new_total_order_service.py
+++ b/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()
diff --git a/app/infrastructure/fastapi/total_order_routes.py b/app/infrastructure/fastapi/total_order_routes.py
index d580253..5caed5f 100644
--- a/app/infrastructure/fastapi/total_order_routes.py
+++ b/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"