open_strequest_new.py 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. import requests
  2. import xmpp
  3. from xmpp import cli
  4. import config
  5. from BotClass import JabberBot
  6. import json
  7. #раскомментировать, когда перейдем на разные пароли
  8. import hashlib
  9. from datetime import datetime
  10. import time
  11. import threading
  12. #test
  13. import jsonAndRequest as jsreq
  14. global tasklist
  15. botname="open_strequest_new"
  16. operating_status=111
  17. JID=botname+"@"+config.JSERVER
  18. PORT=config.PORT
  19. PASSWORD=hashlib.md5((botname+config.PASSWORD).encode('utf-8')).hexdigest()
  20. print(JID, PASSWORD)
  21. #таким образом хранится список jid, от которых можно получать сообщения этому боту
  22. listen_to=["test@ej.sharix-app.org", "open_template_bot@ej.sharix-app.org"]
  23. #тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
  24. proceed_to=[botname, "open_strequest_assigned@ej.sharix-app.org", "open_servicerequest_forcemajeure@ej.sharix-app.org"]
  25. #максимальное значение попыток обработать заказ
  26. idle_value=10
  27. # обработчик входящих сообщений
  28. def message_handler(conn, mess):
  29. text = mess.getBody()#текст сообщения боту
  30. user = mess.getFrom()#отправитель сообщения
  31. print (str(user).split("/")[0])
  32. if (str(user).split("/")[0]) in listen_to:
  33. print(text)
  34. print(mess)
  35. if text is not None:
  36. orderObj = jsreq.jsonToOrderTicket(text)
  37. print (orderObj)
  38. tasklist.append(orderObj)
  39. bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n")
  40. def open_servicerequest_booked_wait(period,localtask):
  41. bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ ожидает\n")
  42. time.sleep(period)
  43. #ВАЖНО! Надо скорее всего через API редактировать заказ, иначе возможна потеря данных
  44. localtask.title=localtask.title+"1"
  45. #обязательно данный обработчик должен заканчиваться передачей заказа куда-то на обработку дальше - обратно или другому, иначе оно потеряется
  46. bot.proceed_status(proceed_to[0],localtask)
  47. bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ подождал и возвращен в очередь\n")
  48. def open_stequest_new_process(localtask):
  49. bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ в обработчике\n")
  50. order = jsreq.jsonToOrder(localtask)
  51. if order.provider:
  52. st_request_assigned(order)
  53. else:
  54. # Что такое маркер обработчика?
  55. if not order.marker:
  56. add_marker(order)
  57. assign_executor(localtask)
  58. if assign_executor(order):
  59. st_request_assigned(localtask)
  60. else:
  61. # Проверка таймера
  62. if True:
  63. if assign_executor(order):
  64. st_request_assigned(localtask)
  65. else:
  66. add_marker(order)
  67. assign_executor(order)
  68. def st_request_assigned(localtask):
  69. bot.proceed_status(proceed_to[1],localtask)
  70. # Пока не уверен
  71. def assign_executor(order):
  72. # Алгоритм назначения исполнителя
  73. # ...
  74. return True
  75. def add_marker(order):
  76. #
  77. pass
  78. def open_servicerequest_booked():
  79. while (len(tasklist)):
  80. localtask=tasklist.pop(0)
  81. bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " "+ "Заказ в очереди на обработке\n")
  82. print("EACH TASKLIST", tasklist)
  83. if (localtask.status!=operating_status):
  84. bot.proceed_status(proceed_to[-1],localtask)
  85. bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ не по адресу, перенаправляем на форсмажор\n")
  86. continue
  87. # if (localtask.title!="10"):
  88. # t1=threading.Thread(target=open_servicerequest_booked_wait, args=(5,localtask))
  89. # t1.start()
  90. # if (localtask.title=="101"):
  91. # t1=threading.Thread(target=open_servicerequest_booked_process, args=(10,localtask))
  92. # t1.start()
  93. t1=threading.Thread(target=open_stequest_new_process, args=(localtask))
  94. t1.start()
  95. #если никакие обработчики не подошли - отправляем обратно в очередь
  96. bot.proceed_status(proceed_to[0],localtask)
  97. print("Заказ возвращен в очередь\n")
  98. print(tasklist)
  99. #Авторизация и запуск бота
  100. #пустой список заказов
  101. tasklist = []
  102. #надо инициализировать tasklist при запуске из API
  103. token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL+"/auth/token/login/")
  104. #ВАЖНО! Надо еще фильтровать список по статусу обработки заявки
  105. tasklist=jsreq.requestGetList(token, config.API_URL+"/tickets/api/tickets/?list_id=10")
  106. print (len(tasklist))
  107. bot = JabberBot(JID, PASSWORD, PORT)
  108. bot.register_handler('message', message_handler)
  109. bot.start(open_servicerequest_booked)