sharix-open-backend/handlers/open_servicerequest_process.py

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_process"

operating_status='231'
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_accepted@ej.sharix-app.org"]

#тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
proceed_to=[botname,"open_servicerequest_done@ej.sharix-app.org", "open_servicerequest_preforcemajeure@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:
        if text is not None:
            orderObj = jsreq.jsonToOrder(text)
            print (orderObj)
            tasklist.append(orderObj)
            bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n")


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 open_servicerequest_process_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
    # Здесь можно реализовать логику выбора водителя
    while True:
        if len(answers) > 0:
            answer = answers.pop(0)
            if answer.answer == 'Complete':
                localtask.status='241'
                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")
            break
        else:
            print("Ожидание ответа...")
            time.sleep(30)

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 open_servicerequest_process_process(localtask):
    bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ в обработчике\n")
    order = localtask
    wait_thread = threading.Thread(target=wait_for_answer, args=(localtask,))#ожидание выполнения заказа
    wait_thread.start()

    # Заглушка проверка баланса
    # if check_balance() >= order.real_price:
    # if True:
    #     # Заглушка. Расчетное время прибытия на следующую точку превышаетзапланированное на Х
    #     if True:
    #         bot.proceed_status(proceed_to[1],localtask)
    #         bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ отправлен\n")
    #     else:
    #         bot.proceed_status(proceed_to[0],localtask)
    #         bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ подождал и отправлен в очередь\n")
    # else:
    #     # Оплата

    #     bot.proceed_status(proceed_to[1],localtask)
    #     bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ отправлен\n")
        
    #     pass


def check_balance():
    pass

def open_servicerequest_process():
    while (len(tasklist)):
        localtask=tasklist.pop(0)
        bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " "+ "Заказ в очереди на обработке\n")
        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

        # 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_process_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_process)