|
@@ -1,43 +1,46 @@
|
|
|
-import requests
|
|
|
-import xmpp
|
|
|
-from xmpp import cli
|
|
|
-import config
|
|
|
-from BotClass import JabberBot
|
|
|
-import json
|
|
|
-#раскомментировать, когда перейдем на разные пароли
|
|
|
+# раскомментировать, когда перейдем на разные пароли
|
|
|
import hashlib
|
|
|
+import threading
|
|
|
+import time
|
|
|
from datetime import datetime
|
|
|
|
|
|
-import time
|
|
|
-import threading
|
|
|
+import requests
|
|
|
|
|
|
-#test
|
|
|
+import config
|
|
|
+# test
|
|
|
import jsonAndRequest as jsreq
|
|
|
+from BotClass import JabberBot
|
|
|
|
|
|
global tasklist
|
|
|
+
|
|
|
+
|
|
|
def log_handler():
|
|
|
while True:
|
|
|
with open('log.txt', 'a') as log_file:
|
|
|
log_file.write(f"{time.strftime('%Y-%m-%d %H:%M:%S')} - Function is still running...\n")
|
|
|
time.sleep(3600)
|
|
|
-botname="open_service_request_booked"
|
|
|
|
|
|
-operating_status='211'
|
|
|
-JID=botname+"@"+config.JSERVER
|
|
|
-PORT=config.PORT
|
|
|
|
|
|
-PASSWORD=hashlib.md5((botname+config.PASSWORD).encode('utf-8')).hexdigest()
|
|
|
+botname = "open_service_request_booked"
|
|
|
+
|
|
|
+operating_status = '211'
|
|
|
+JID = botname + "@" + config.JSERVER
|
|
|
+PORT = config.PORT
|
|
|
+
|
|
|
+PASSWORD = hashlib.md5((botname + config.PASSWORD).encode('utf-8')).hexdigest()
|
|
|
|
|
|
print(JID, PASSWORD)
|
|
|
|
|
|
-#таким образом хранится список jid, от которых можно получать сообщения этому боту
|
|
|
-listen_to=["799999999990@ej.sharix-app.org"]
|
|
|
+# таким образом хранится список jid, от которых можно получать сообщения этому боту
|
|
|
+listen_to = ["799999999990@ej.sharix-app.org"]
|
|
|
+
|
|
|
+# тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
|
|
|
+proceed_to = ["open_service_request_assigned@ej.sharix-app.org", "open_service_request_pending@ej.sharix-app.org",
|
|
|
+ "open_service_request_forcemajeure@ej.sharix-app.org"]
|
|
|
|
|
|
-#тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
|
|
|
-proceed_to=["open_service_request_assigned@ej.sharix-app.org","open_service_request_pending@ej.sharix-app.org", "open_service_request_forcemajeure@ej.sharix-app.org"]
|
|
|
+# максимальное значение попыток обработать заказ
|
|
|
+idle_value = 10
|
|
|
|
|
|
-#максимальное значение попыток обработать заказ
|
|
|
-idle_value=10
|
|
|
|
|
|
def log_handler():
|
|
|
while True:
|
|
@@ -45,59 +48,61 @@ def log_handler():
|
|
|
log_file.write(f"{time.strftime('%Y-%m-%d %H:%M:%S')} - {botname} is still running...\n")
|
|
|
time.sleep(3600)
|
|
|
|
|
|
+
|
|
|
# обработчик входящих сообщений
|
|
|
def message_handler(conn, mess):
|
|
|
- text = mess.getBody()#текст сообщения боту
|
|
|
- user = mess.getFrom()#отправитель сообщения
|
|
|
+ text = mess.getBody() # текст сообщения боту
|
|
|
+ user = mess.getFrom() # отправитель сообщения
|
|
|
if (str(user).split("/")[0]) in listen_to:
|
|
|
# print (str(user).split("/")[0])
|
|
|
if text is not None:
|
|
|
- try :
|
|
|
+ try:
|
|
|
orderObj = jsreq.jsonToOrder(text)
|
|
|
tasklist.append(orderObj)
|
|
|
- bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n")
|
|
|
+ bot.bot_log(str(datetime.now()) + " Поступивший заказ успешно добавлен в очередь обработки\n")
|
|
|
except:
|
|
|
bot.proceed_status(proceed_to[-1], text)
|
|
|
- bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на assigned\n")
|
|
|
-
|
|
|
-
|
|
|
+ bot.bot_log(
|
|
|
+ botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на assigned\n")
|
|
|
+
|
|
|
|
|
|
-def open_service_request_booked_wait(period,localtask):
|
|
|
- bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ ожидает\n")
|
|
|
+def open_service_request_booked_wait(period, localtask):
|
|
|
+ bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ ожидает\n")
|
|
|
time.sleep(period)
|
|
|
- #ВАЖНО! Надо скорее всего через API редактировать заказ, иначе возможна потеря данных
|
|
|
- localtask.title=localtask.title+"1"
|
|
|
+ # ВАЖНО! Надо скорее всего через API редактировать заказ, иначе возможна потеря данных
|
|
|
+ localtask.title = localtask.title + "1"
|
|
|
+
|
|
|
+ # обязательно данный обработчик должен заканчиваться передачей заказа куда-то на обработку дальше - обратно или другому, иначе оно потеряется
|
|
|
+ bot.proceed_status(proceed_to[0], localtask)
|
|
|
+ bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ подождал и возвращен в очередь\n")
|
|
|
|
|
|
- #обязательно данный обработчик должен заканчиваться передачей заказа куда-то на обработку дальше - обратно или другому, иначе оно потеряется
|
|
|
- bot.proceed_status(proceed_to[0],localtask)
|
|
|
- bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ подождал и возвращен в очередь\n")
|
|
|
|
|
|
def open_service_request_booked_process(localtask):
|
|
|
- bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ в обработчике\n")
|
|
|
-
|
|
|
+ bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ в обработчике\n")
|
|
|
+
|
|
|
if not localtask.provider:
|
|
|
- answer,provider,service = set_provider(localtask.id)
|
|
|
+ answer, provider, service = set_provider(localtask.id)
|
|
|
# localtask.service =service.id
|
|
|
localtask.service = service['id']
|
|
|
if set_provider(localtask.id):
|
|
|
- bot.bot_log(botname+" "+str(datetime.now())+" "+"Водитель успешно установлен\n")
|
|
|
+ bot.bot_log(botname + " " + str(datetime.now()) + " " + "Водитель успешно установлен\n")
|
|
|
else:
|
|
|
- bot.bot_log(botname+" "+str(datetime.now())+" "+"'Ошибка водитель не был установлен'\n")
|
|
|
+ bot.bot_log(botname + " " + str(datetime.now()) + " " + "'Ошибка водитель не был установлен'\n")
|
|
|
localtask.provider = provider
|
|
|
current_time = datetime.now()
|
|
|
-
|
|
|
+
|
|
|
# if localtask.asap or (current_time - datetime.fromisoformat(localtask.time_start_predicted) < timedelta(hours=3)):
|
|
|
if localtask.asap:
|
|
|
localtask.status = '212'
|
|
|
- print(jsreq.change_to_task(localtask,token))
|
|
|
+ print(jsreq.change_to_task(localtask, token))
|
|
|
bot.proceed_status(proceed_to[0], localtask)
|
|
|
bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на assigned\n")
|
|
|
else:
|
|
|
localtask.status = '220'
|
|
|
- print(jsreq.change_to_task(localtask,token))
|
|
|
+ print(jsreq.change_to_task(localtask, token))
|
|
|
bot.proceed_status(proceed_to[1], localtask)
|
|
|
bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на pending\n")
|
|
|
-
|
|
|
+
|
|
|
# Проверка срочности заказа
|
|
|
# if order.asap:
|
|
|
# #Заглушка назначение исполнителя системой
|
|
@@ -111,14 +116,15 @@ def open_service_request_booked_process(localtask):
|
|
|
# print(localtask)
|
|
|
# bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на assigned\n")
|
|
|
# Проверка срочности заказа
|
|
|
-
|
|
|
+
|
|
|
+
|
|
|
def get_balance(client):
|
|
|
headers = {'Authorization': f'Token {token}'}
|
|
|
# Передаем id пользователя и получаем его баланс
|
|
|
- response = requests.get(config.API_URL+f"/api/example/{client}", headers=headers)
|
|
|
+ response = requests.get(config.API_URL + f"/api/example/{client}", headers=headers)
|
|
|
print(response.json())
|
|
|
- ordertext=response.json()
|
|
|
-
|
|
|
+ ordertext = response.json()
|
|
|
+
|
|
|
if not ordertext:
|
|
|
print("Empty array")
|
|
|
else:
|
|
@@ -128,27 +134,28 @@ def get_balance(client):
|
|
|
return balance
|
|
|
except Exception as ex:
|
|
|
print(ex)
|
|
|
-
|
|
|
-#set provider?
|
|
|
-def choose_provider(auth_token,geo):
|
|
|
- providers=requests.get(config.API_URL+"/dbsynce/api/provider/",headers=auth_token)
|
|
|
- provider=providers.json()[0]['id']
|
|
|
+
|
|
|
+
|
|
|
+# set provider?
|
|
|
+def choose_provider(auth_token, geo):
|
|
|
+ providers = requests.get(config.API_URL + "/dbsynce/api/provider/", headers=auth_token)
|
|
|
+ provider = providers.json()[0]['id']
|
|
|
return provider
|
|
|
-
|
|
|
+
|
|
|
+
|
|
|
def set_provider(localtask_id):
|
|
|
- token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL+"my/api/v1/auth/token/login/")
|
|
|
+ token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL + "my/api/v1/auth/token/login/")
|
|
|
headers = {'Authorization': f'Token {token}'}
|
|
|
- provider = choose_provider(headers,'geoposition')
|
|
|
+ provider = choose_provider(headers, 'geoposition')
|
|
|
# NOTE:BUG:ERROR:FIXME: модель сервис не должна была содержать поле id_provider
|
|
|
# service = requests.get(config.API_URL+f"/dbsynce/api/service/?id_provider={provider}", headers=headers)
|
|
|
- service=service.json()[0]
|
|
|
- data = {'provider':int(provider)}
|
|
|
+ service = service.json()[0]
|
|
|
+ data = {'provider': int(provider)}
|
|
|
# print(data)
|
|
|
# print(str(config.API_URL+f"/dbsynce/api/orders/{localtask_id}/"))
|
|
|
- response = requests.patch(config.API_URL+f"/dbsynce/api/orders/{localtask_id}/", headers=headers, json=data)
|
|
|
- return response ,provider , service
|
|
|
+ response = requests.patch(config.API_URL + f"/dbsynce/api/orders/{localtask_id}/", headers=headers, json=data)
|
|
|
+ return response, provider, service
|
|
|
|
|
|
-
|
|
|
# if not ordertext:
|
|
|
# print("Empty array")
|
|
|
# else:
|
|
@@ -159,26 +166,25 @@ def set_provider(localtask_id):
|
|
|
# except Exception as ex:
|
|
|
# print(ex)
|
|
|
|
|
|
+
|
|
|
def open_service_request_booked():
|
|
|
log_thread = threading.Thread(target=log_handler)
|
|
|
- log_thread.daemon = True
|
|
|
+ log_thread.daemon = True
|
|
|
log_thread.start()
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
+
|
|
|
while (len(tasklist)):
|
|
|
- localtask=tasklist.pop(0)
|
|
|
- if (str(localtask.status)!=str(operating_status)):
|
|
|
+ localtask = tasklist.pop(0)
|
|
|
+ if (str(localtask.status) != str(operating_status)):
|
|
|
# bot.proceed_status(proceed_to[-1],localtask) #временная проверка отправки
|
|
|
print(localtask)
|
|
|
- localtask.status='239'
|
|
|
- jsreq.change_to_task(localtask,token)
|
|
|
- bot.proceed_status(proceed_to[-1],localtask)
|
|
|
- bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ не по адресу, перенаправляем на форсмажор\n")
|
|
|
+ localtask.status = '239'
|
|
|
+ jsreq.change_to_task(localtask, token)
|
|
|
+ bot.proceed_status(proceed_to[-1], localtask)
|
|
|
+ bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ не по адресу, перенаправляем на форсмажор\n")
|
|
|
continue
|
|
|
else:
|
|
|
- bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " "+ "Заказ в очереди на обработке\n")
|
|
|
- t1=threading.Thread(target=open_service_request_booked_process, args=([localtask]))
|
|
|
+ bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " " + "Заказ в очереди на обработке\n")
|
|
|
+ t1 = threading.Thread(target=open_service_request_booked_process, args=([localtask]))
|
|
|
t1.start()
|
|
|
# elif (localtask.provider==T):
|
|
|
# t1=threading.Thread(target=open_service_request_booked_wait, args=(5,localtask))
|
|
@@ -187,29 +193,28 @@ def open_service_request_booked():
|
|
|
# if (localtask.title=="101"):
|
|
|
# t1=threading.Thread(target=open_service_request_booked_process, args=(10,localtask))
|
|
|
# t1.start()
|
|
|
-
|
|
|
-
|
|
|
- #если никакие обработчики не подошли - отправляем обратно в очередь
|
|
|
+
|
|
|
+ # если никакие обработчики не подошли - отправляем обратно в очередь
|
|
|
# bot.proceed_status(proceed_to[0],localtask)
|
|
|
# print("Заказ возвращен в очередь\n")
|
|
|
# print(tasklist)
|
|
|
|
|
|
-#Авторизация и запуск бота
|
|
|
|
|
|
-#пустой список заказов
|
|
|
+# Авторизация и запуск бота
|
|
|
+
|
|
|
+# пустой список заказов
|
|
|
tasklist = []
|
|
|
|
|
|
# #надо инициализировать tasklist при запуске из API
|
|
|
-token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL+"my/api/v1/auth/token/login/")
|
|
|
+token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL + "my/api/v1/auth/token/login/")
|
|
|
# #ВАЖНО! Надо еще фильтровать список по статусу обработки заявки
|
|
|
-tasklist_dbsynce=jsreq.requestGetList(token, config.API_URL+"/dbsynce/api/orders/") ###Должна быть проверка между двумя
|
|
|
+tasklist_dbsynce = jsreq.requestGetList(token,
|
|
|
+ config.API_URL + "/dbsynce/api/orders/") ###Должна быть проверка между двумя
|
|
|
|
|
|
# # print (len(tasklist))
|
|
|
|
|
|
bot = JabberBot(JID, PASSWORD, PORT)
|
|
|
|
|
|
-
|
|
|
-
|
|
|
bot.register_handler('message', message_handler)
|
|
|
bot.start(open_service_request_booked)
|
|
|
|