sharix-open-webservice-running/handlers/open_service_request_closed.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

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()


###################
botname = "open_service_request_closed"

operating_status = 239
JID = botname + "@" + config.JSERVER
PORT = config.PORT

PASSWORD = hashlib.md5((botname + config.PASSWORD).encode('utf-8')).hexdigest()

print(JID, PASSWORD)

token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL + "my/api/v1/auth/token/login/")
headers = {'Authorization': f'Token {token}'}

# таким образом хранится список jid, от которых можно получать сообщения этому боту
listen_to = ["799999999990@ej.sharix-app.org", "test@ej.sharix-app.org"]

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


def open_service_request_closed_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_closed_process(localtask):
    bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ в обработчике\n")
    response = requests.delete(config.API_URL + f"/dbsynce/api/orders/{localtask.id}/", headers=headers)
    text = (config.API_URL + f"dbsynce/api/orders/{localtask.id}/")
    if response:
        bot.proceed_status(proceed_to[2], 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")


def open_service_request_closed():
    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):
        #     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.status in ("239", "211", "212", "220", "221", "222", "231", "238", "229")):
            t1 = threading.Thread(target=open_service_request_closed_process, args=([localtask]))
            t1.start()

        # если никакие обработчики не подошли - отправляем обратно в очередь
        bot.proceed_status(proceed_to[0], localtask)
        print("Заказ возвращен в очередь\n")


# Авторизация и запуск бота

# пустой список заказов
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.start(open_service_request_closed)