sharix-open-webservice-running/handlers/open_servicerequest_assigned.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
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 log_handler():
    while True:
        with open(f'{botname}.txt', 'a') as log_file:
            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()#отправитель сообщения
    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:
        start_time = time.time()
        if time.time()+400 - start_time > 300: ##Логика временная
            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")
                elif answer.answer == 'DECLINE':
                    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")
                else:
                    bot.bot_log(botname + " " + str(datetime.now()) + f"Ошибка ответа\n")
                break
            else:
                print("Ожидание ответа...")
                time.sleep(3)
        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
            

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():
    log_thread = threading.Thread(target=log_handler)
    log_thread.daemon = True 
    log_thread.start()
    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)


#Добавить функцию поток для ожидания получения заказа.