add check balance func

open_servicerequest_booked.py

@@ -65,20 +65,20 @@ def open_servicerequest_booked_wait(period,localtask):
 def open_servicerequest_booked_process(localtask):
     bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ в обработчике\n")
 
+    order = jsreq.jsonToOrder(localtask)
     #Пока заглушка
-    balance = 1000
-    price = 999
+    balance = get_balance(order.client_id)
 
-    if balance > price:
-        # Пока не может преобразовать в Order
-        order = jsreq.jsonToOrder(localtask)
+    # Какой тип цены лучше использовать?
+    price = order.real_price
 
+    if balance > price:
         #Заказ на основного водителя?
-        if check_provider(order):
+        if order.provider:
             bot.proceed_status(proceed_to[1], localtask)
             bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан\n")
         else:
-            # Затычка срочности заказа
+            # Проверка срочности заказа
             if True:
 
                 #Назначение исполнителя системой
@@ -90,15 +90,13 @@ def open_servicerequest_booked_process(localtask):
             
             bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан\n")
     else:
-        # Процесс получения оплаты от клиента
+        # Заглушка получения оплаты от клиента
         pass
 
-# Является ли водитель основным
-def check_provider(order):
+def get_balance(client):
     headers = {'Authorization': f'Token {token}'}
-
-    # Нужно искать среди всех взаимоотношений или это все сделать в АПИ?
-    response = requests.get(config.API_URL+"/metaservicesynced/api/relationship/", headers=headers)
+    # Передаем id пользователя и получаем его баланс
+    response = requests.get(config.API_URL+f"/api/exaple/{client}", headers=headers)
     print(response.json())
     ordertext=response.json()
     
@@ -107,12 +105,8 @@ def check_provider(order):
     else:
         print(ordertext[0], "PROCESSING")
         try:
-            neg_type = ordertext[0].get('neg_type')
-            print(ordertext[0].get('neg_type'))
-            if neg_type == 1:
-                return True
-            else:
-                return False
+            balance = ordertext[0].get('balance')
+            return balance
         except Exception as ex:
             print(ex)