open_template_bot.py 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. # раскомментировать, когда перейдем на разные пароли
  2. import hashlib
  3. import threading
  4. import time
  5. from datetime import datetime
  6. import config
  7. # test
  8. import jsonAndRequest as jsreq
  9. from BotClass import JabberBot
  10. global tasklist
  11. # В этом файле содержится только то, что запускает конкретный обработчик
  12. # botname="open_st_request_assigned"
  13. botname = "open_template_bot"
  14. operating_status = 210
  15. JID = botname + "@" + config.JSERVER
  16. PORT = config.PORT
  17. # раскомментировать, когда перейдем на разные пароли
  18. PASSWORD = hashlib.md5((botname + config.PASSWORD).encode('utf-8')).hexdigest()
  19. # таким образом хранится список jid, от которых можно получать сообщения этому боту
  20. listen_to = ["test@ej.sharix-app.org", "open_template_bot@ej.sharix-app.org"]
  21. # тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
  22. proceed_to = ["open_template_bot@ej.sharix-app.org", "test@ej.sharix-app.org", "test1@ej.sharix-app.org",
  23. "open_service_request_forcemajeure@ej.sharix-app.org"]
  24. # максимальное значение попыток обработать заказ
  25. idle_value = 10
  26. # обработчик входящих сообщений
  27. def message_handler(conn, mess):
  28. text = mess.getBody() # текст сообщения боту
  29. user = mess.getFrom() # отправитель сообщения
  30. # user = mess.getNode()#только jid
  31. # нужна проверка, данный обработчик вообще должен от данного отправителя получать сообщения или нет. Подсказка ниже
  32. print(str(user).split("/")[0])
  33. if (str(user).split("/")[0]) in listen_to:
  34. print(text)
  35. print(mess)
  36. if text is not None:
  37. orderObj = jsreq.jsonToOrderTicket(text)
  38. print(orderObj)
  39. tasklist.append(orderObj)
  40. bot.bot_log(str(datetime.now()) + " Поступивший заказ успешно добавлен в очередь обработки\n")
  41. def open_bot_template_wait(period, localtask):
  42. bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ ожидает\n")
  43. time.sleep(period)
  44. # ВАЖНО! Надо скорее всего через API редактировать заказ, иначе возможна потеря данных
  45. localtask.title = localtask.title + "1"
  46. # обязательно данный обработчик должен заканчиваться передачей заказа куда-то на обработку дальше - обратно или другому, иначе оно потеряется
  47. bot.proceed_status(proceed_to[0], localtask)
  48. bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ подождал и возвращен в очередь\n")
  49. def open_bot_template_process(localtask):
  50. bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ в обработчике\n")
  51. # ВАЖНО! Надо через API редактировать заказ, иначе возможна потеря данных
  52. # localtask.title +=1
  53. bot.proceed_status(proceed_to[1], localtask)
  54. bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан\n")
  55. def open_bot_template():
  56. # print("Тут описывается работа со списком задач в данном статусе")
  57. while (len(tasklist)):
  58. # for i in range(len(tasklist)):
  59. # проверяем элемент tasklist[i]
  60. # print (i, len(tasklist))
  61. localtask = tasklist.pop(0)
  62. bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " " + "Заказ в очереди на обработке\n")
  63. # bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " " + str(i) + " " + "Заказ в очереди на обработке\n")
  64. # print ("EACH TASKLIST", tasklist[i])
  65. # localtask=tasklist.pop(i)
  66. # если боту почему-то пришла не его заявка - это явно ошибка движения процессов и отдаем в форсмажор - этот обработчик указывать стоит всегда последним
  67. if (localtask.status != operating_status):
  68. bot.proceed_status(proceed_to[-1], localtask)
  69. bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ не по адресу, перенаправляем на форсмажор\n")
  70. continue
  71. if (localtask.title != "10"):
  72. t1 = threading.Thread(target=open_bot_template_wait, args=(5, localtask))
  73. t1.start()
  74. if (localtask.title == "101"):
  75. t1 = threading.Thread(target=open_bot_template_process, args=(10, localtask))
  76. t1.start()
  77. # если никакие обработчики не подошли - отправляем обратно в очередь
  78. bot.proceed_status(proceed_to[0], localtask)
  79. print("Заказ возвращен в очередь\n")
  80. print(tasklist)
  81. # Авторизация и запуск бота
  82. # пустой список заказов
  83. tasklist = []
  84. # надо инициализировать tasklist при запуске из API
  85. token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL + "/base/auth/token/login/")
  86. # ВАЖНО! Надо еще фильтровать список по статусу обработки заявки
  87. tasklist = jsreq.requestGetList(token, config.API_URL + "/webservice/api/webservice/")
  88. print(JID, PASSWORD, PORT)
  89. bot = JabberBot(JID, PASSWORD, PORT)
  90. bot.register_handler('message', message_handler)
  91. bot.start(open_bot_template)