sharix-open-jabber-bot/open_template_bot.py

import config
import xmpp
import logging
import loggingConfig
import pprint


from xmpp import cli
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_strequest_assigned"
botname="open_template_bot"
operating_status=210
JID=botname+"@"+config.JSERVER
PORT=config.PORT

#раскомментировать, когда перейдем на разные пароли
PASSWORD=hashlib.md5((botname+config.PASSWORD).encode('utf-8')).hexdigest()

logging.debug(f"\n JID: {JID}\n PASSWORD: {PASSWORD}")

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

#тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
proceed_to=[
    "open_template_bot@ej.sharix-app.org", 
    "test@ej.sharix-app.org", 
    "test1@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()#отправитель сообщения
    #user = mess.getNode()#только jid 
   
   #нужна проверка, данный обработчик вообще должен от данного отправителя получать сообщения или нет. Подсказка ниже
    #print (str(user).split("/")[0])
    logging.debug(str(user).split("/")[0])
    # input()
    if (str(user).split("/")[0]) in listen_to:
        # print(text)
        # print(mess)
        logging.debug(pprint.pformat(text))
        logging.debug(pprint.pformat(mess))
        # input()

        if text is not None:
            logging.debug(f"text before jsonToOrderTicket: {pprint.pformat(text)}")
            orderObj = jsreq.jsonToOrderTicket(text)
            logging.debug(f"text after jsonToOrderTicket: {pprint.pformat(orderObj)}")
            # print (orderObj)
            tasklist.append(orderObj)
            logging.info("Поступивший заказ успешно добавлен в очередь обработки")
            logging.debug(pprint.pformat(tasklist))
            bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n")
            # input()

def open_bot_template_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_bot_template_process(localtask):
    bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ в обработчике\n")
    
    #ВАЖНО! Надо через API редактировать заказ, иначе возможна потеря данных
    #localtask.title +=1

    bot.proceed_status(proceed_to[1],localtask)
    bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ успешно обработан\n")


def open_bot_template():
    #print("Тут описывается работа со списком задач в данном статусе")
    while (len(tasklist)):
        logging.info(f"Всего задач: {len(tasklist)}")
        logging.debug(f"Tasklist: {pprint.pformat(tasklist)}")
    #for i in range(len(tasklist)):
        #проверяем элемент tasklist[i]
        #print (i, len(tasklist))
        localtask=tasklist.pop(0)
        bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " "+ "Заказ в очереди на обработке\n")
        logging.debug(f"Заказ в очереди на обработке: {localtask} ")
        # input("(open_template_bot) INPUT TO CONTINUE...")
        #bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " " + str(i) + " " + "Заказ в очереди на обработке\n")
        #print ("EACH TASKLIST", tasklist[i])
        #localtask=tasklist.pop(i)
        
        #если боту почему-то пришла не его заявка - это явно ошибка движения процессов и отдаем в форсмажор - этот обработчик указывать стоит всегда последним
        if (localtask.status!=operating_status):
            bot.proceed_status(proceed_to[-1],localtask)
            bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ не по адресу, перенаправляем на форсмажор\n")
            logging.info("Заказ не по адресу, перенаправляем на форсмажор.")
            continue

        if (localtask.title != "10"):
            t1=threading.Thread(target=open_bot_template_wait, args=(5,localtask))
            t1.start()

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

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

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

#пустой список заказов
tasklist = []

#надо инициализировать tasklist при запуске из API
logging.info("Инициализация токена...")
token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL+"/auth/token/login/")
logging.info("Токен инициализирован")
logging.debug(f"token: {token}")

#ВАЖНО! Надо еще фильтровать список по статусу обработки заявки
logging.info("Получение tasklist...")
tasklist=jsreq.requestGetList(token, config.API_URL+"/tickets/api/tickets/?list_id=10")
#tasklist=jsreq.requestGetList(token, config.API_URL+"/tickets/api/tickets")
logging.info("tasklist получен")
logging.debug(f"tasklist длина: {len(tasklist)}")
#print (len(tasklist))

bot = JabberBot(JID, PASSWORD, PORT)

bot.register_handler('message', message_handler)
bot.start(open_bot_template)