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 import asyncio #test import jsonAndRequest as jsreq global tasklist botname="open_servicerequest_assigned" operating_status='212' JID=botname+"@"+config.JSERVER PORT=config.PORT PASSWORD=hashlib.md5((botname+config.PASSWORD).encode('utf-8')).hexdigest() print(JID, PASSWORD) drivers = ["799999999998@ej.sharix-app.org"] #таким образом хранится список jid, от которых можно получать сообщения этому боту listen_to=["open_servicerequest_booked@ej.sharix-app.org","open_servicerequest_pending@ej.sharix-app.org"] #тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки proceed_to=[botname, "open_servicerequest_accepted@ej.sharix-app.org", "open_servicerequest_declined@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('order_handler') if text is not None: orderObj = jsreq.jsonToOrder(text) tasklist.append(orderObj) bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n") def open_servicerequest_assigned_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 wait_for_answer(localtask): order = localtask driver = drivers[0] send_message(order,driver) # Здесь можно реализовать логику выбора водителя while True: if len(answers) > 0: answer = answers.pop(0) if answer.answer == 'Accept': localtask.status='221' jsreq.change_to_task(localtask,token) bot.proceed_status(proceed_to[1], localtask) bot.bot_log(botname + " " + str(datetime.now()) + f" Заказ отправлен на {proceed_to[1]}\n") else: localtask.status='229' localtask.provider=None jsreq.change_to_task(localtask,token) bot.proceed_status(proceed_to[2], localtask) bot.bot_log(botname + " " + str(datetime.now()) + f" Заказ отправлен на {proceed_to[2]}\n") break else: print("Ожидание ответа...") time.sleep(1) def open_servicerequest_assigned_process(localtask): bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ в обработчике\n") order = localtask if order.provider!=None: wait_thread = threading.Thread(target=wait_for_answer, args=(localtask,)) wait_thread.start() def message_answer(conn, mess): text = mess.getBody()#текст сообщения боту user = mess.getFrom()#отправитель сообщения if (str(user).split("/")[0]) in drivers: print ('answer_handler') if text is not None: answerObj = jsreq.to_answer(text) answers.append(answerObj) bot.bot_log(str(datetime.now())+" Поступивший ответ успешно добавлен в очередь обработки\n") def send_message(order,driver): order=json.dumps(order.__dict__) bot.send_notification(driver,order) def open_servicerequest_assigned(): while (len(tasklist)): localtask=tasklist.pop(0) if (localtask.status!=operating_status): 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_assigned_process, args=([localtask])) t1.start() # if (localtask.title!="10"): # 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() # t1=threading.Thread(target=open_servicerequest_assigned_process, args=(localtask)) # t1.start() #если никакие обработчики не подошли - отправляем обратно в очередь # bot.proceed_status(proceed_to[0],localtask) # print("Заказ возвращен в очередь\n") # print(tasklist) #Авторизация и запуск бота answers = [] #пустой список заказов 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/") bot = JabberBot(JID, PASSWORD, PORT) bot.register_handler('message', message_handler) bot.register_handler('message', message_answer) bot.start(open_servicerequest_assigned) #Добавить функцию поток для ожидания получения заказа.