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

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

import requests

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

global tasklist


def log_handler():
    while True:
        with open('log.txt', 'a') as log_file:
            log_file.write(f"{time.strftime('%Y-%m-%d %H:%M:%S')} - Function is still running...\n")
        time.sleep(3600)


botname = "open_service_request_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_service_request_assigned@ej.sharix-app.org", "open_service_request_pending@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:
        # print (str(user).split("/")[0])
        if text is not None:
            try:
                orderObj = jsreq.jsonToOrder(text)
                tasklist.append(orderObj)
                bot.bot_log(str(datetime.now()) + " Поступивший заказ успешно добавлен в очередь обработки\n")
            except:
                bot.proceed_status(proceed_to[-1], text)
                bot.bot_log(
                    botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан и отправлен на assigned\n")


def open_service_request_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_service_request_booked_process(localtask):
    bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ в обработчике\n")

    if not localtask.provider:
        answer, provider, service = set_provider(localtask.id)
        # localtask.service =service.id
        localtask.service = service['id']
        if set_provider(localtask.id):
            bot.bot_log(botname + " " + str(datetime.now()) + " " + "Водитель успешно установлен\n")
        else:
            bot.bot_log(botname + " " + str(datetime.now()) + " " + "'Ошибка водитель не был установлен'\n")
        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'
        print(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'
        print(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')
    # NOTE:BUG:ERROR:FIXME: модель сервис не должна была содержать поле id_provider
    # service = requests.get(config.API_URL+f"/dbsynce/api/service/?id_provider={provider}", headers=headers)
    service = service.json()[0]
    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, service

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

    while (len(tasklist)):
        localtask = tasklist.pop(0)
        if (str(localtask.status) != str(operating_status)):
            # bot.proceed_status(proceed_to[-1],localtask) #временная проверка отправки
            print(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_service_request_booked_process, args=([localtask]))
            t1.start()
        # elif (localtask.provider==T):
        #     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()

        # если никакие обработчики не подошли - отправляем обратно в очередь
        # 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_service_request_booked)

# print(set_provider('localtask'))