sharix-open-webservice-running/handlers/open_service_request_process.py

# раскомментировать, когда перейдем на разные пароли
import hashlib
import threading
import time
from datetime import datetime

import config
# test
import jsonAndRequest as jsreq
from BotClass import JabberBot

global tasklist

botname = "open_service_request_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_service_request_accepted@ej.sharix-app.org"]

# тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
proceed_to = [botname, "open_service_request_done@ej.sharix-app.org",
              "open_service_request_preforcemajeure@ej.sharix-app.org",
              "open_service_request_declined@ej.sharix-app.org", "open_service_request_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:
        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_service_request_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_service_request_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_service_request_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_service_request_process():
    log_thread = threading.Thread(target=log_handler)
    log_thread.daemon = True
    log_thread.start()
    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_service_request_booked_wait, args=(5,localtask))
        #     t1.start()

        # if (localtask.title=="101"):
        #     t1=threading.Thread(target=open_service_request_booked_process, args=(10,localtask))
        #     t1.start()

        t1 = threading.Thread(target=open_service_request_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_service_request_process)