import requests import xmpp from xmpp import cli import config from BotClass import JabberBot import json #раскомментировать, когда перейдем на разные пароли import hashlib from datetime import datetime import time import threading #test import jsonAndRequest as jsreq global tasklist botname="open_servicerequest_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, кому бот может отправлять сообщения в результате обработки заявки proceed_to=["open_servicerequest_assigned@ej.sharix-app.org","open_servicerequest_pending@ej.sharix-app.org", "open_servicerequest_forcemajeure@ej.sharix-app.org"] #максимальное значение попыток обработать заказ idle_value=10 # обработчик входящих сообщений def message_handler(conn, mess): text = mess.getBody()#текст сообщения боту user = mess.getFrom()#отправитель сообщения if (str(user).split("/")[0]) in listen_to: # print (str(user).split("/")[0]) if text is not None: orderObj = jsreq.jsonToOrder(text) tasklist.append(orderObj) bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n") def open_servicerequest_booked_wait(period,localtask): bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ ожидает\n") time.sleep(period) #ВАЖНО! Надо скорее всего через API редактировать заказ, иначе возможна потеря данных localtask.title=localtask.title+"1" #обязательно данный обработчик должен заканчиваться передачей заказа куда-то на обработку дальше - обратно или другому, иначе оно потеряется bot.proceed_status(proceed_to[0],localtask) bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ подождал и возвращен в очередь\n") def open_servicerequest_booked_process(localtask): bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ в обработчике\n") # order = jsreq.jsonToOrder(localtask) #Пока заглушка # balance = get_balance(order.client_id) # print(balance) # price = order.real_price # if balance > price: # #Заказ на основного водителя? # if order.provider: # bot.proceed_status(proceed_to[1], localtask) # bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан\n") # else: # # Проверка срочности заказа # if order.asap: # #Заглушка назначение исполнителя системой # if set_provider(order.ticket, order.provider) == "success": # bot.proceed_status(proceed_to[0], localtask) # else: # bot.bot_log(botname+" "+str(datetime.now())+" "+"Ошибка назначения исполнителя\n") # else: # bot.proceed_status(proceed_to[1], localtask) # bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан\n") # else: # # Заглушка получения оплаты от клиента # pass # price = localtask.real_price #Заказ на основного водителя? if not localtask.provider: answer,provider = set_provider(localtask.id) print(answer,provider) if set_provider(localtask.id): print('Водитель успешно установлен') else: print('Ошибка водитель не был установлен') 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' 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' 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: # #Заглушка назначение исполнителя системой # if set_provider(order.ticket, order.provider) == "success": # bot.proceed_status(proceed_to[0], localtask) # else: # bot.bot_log(botname+" "+str(datetime.now())+" "+"Ошибка назначения исполнителя\n") # else: # localtask.status = 220 # bot.proceed_status(proceed_to[0], 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) print(response.json()) ordertext=response.json() if not ordertext: print("Empty array") else: print(ordertext[0], "PROCESSING") try: balance = ordertext[0].get('balance') 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'] return provider def set_provider(localtask_id): token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL+"my/api/v1/auth/token/login/") headers = {'Authorization': f'Token {token}'} provider = choose_provider(headers,'geoposition') 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 # if not ordertext: # print("Empty array") # else: # print(ordertext[0], "PROCESSING") # try: # message = ordertext[0].get('message') # return message # except Exception as ex: # print(ex) def open_servicerequest_booked(): while (len(tasklist)): localtask=tasklist.pop(0) if (str(localtask.status)!=str(operating_status)): # bot.proceed_status(proceed_to[-1],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") continue else: bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " "+ "Заказ в очереди на обработке\n") t1=threading.Thread(target=open_servicerequest_booked_process, args=([localtask])) t1.start() # elif (localtask.provider==T): # t1=threading.Thread(target=open_servicerequest_booked_wait, args=(5,localtask)) # t1.start() # if (localtask.title=="101"): # t1=threading.Thread(target=open_servicerequest_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/") # #ВАЖНО! Надо еще фильтровать список по статусу обработки заявки 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_servicerequest_booked) # print(set_provider('localtask'))