open_template_bot.py 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. import xmpp
  2. from xmpp import cli
  3. import config
  4. from BotClass import JabberBot
  5. import json
  6. #раскомментировать, когда перейдем на разные пароли
  7. import hashlib
  8. from datetime import datetime
  9. #test
  10. import jsonAndRequest as jsreq
  11. global tasklist
  12. #В этом файле содержится только то, что запускает конкретный обработчик
  13. # botname="open_strequest_assigned"
  14. botname="open_template_bot"
  15. #botname="test"
  16. JID=botname+"@"+config.JSERVER
  17. #JID=f"{botname}@msg.sharix-app.org"
  18. #PASSWORD=config.PASSWORD
  19. PORT=config.PORT
  20. #раскомментировать, когда перейдем на разные пароли
  21. PASSWORD=hashlib.md5((botname+config.PASSWORD).encode('utf-8')).hexdigest()
  22. #PASSWORD="test1234@"
  23. print(JID, PASSWORD)
  24. #таким образом хранится список jid, от которых можно получать сообщения этому боту
  25. listen_to=["test@ej.sharix-app.org", "test1@ej.sharix-app.org"]
  26. #тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
  27. process_to=["test@ej.sharix-app.org", "test1@ej.sharix-app.org"]
  28. # обработчик входящих сообщений
  29. def message_handler(conn, mess):
  30. text = mess.getBody()#текст сообщения боту
  31. user = mess.getFrom()#отправитель сообщения
  32. #user = mess.getNode()#только jid
  33. #example json to object user
  34. if text is not None:
  35. """ example json
  36. {
  37. "name":"Cristian",
  38. "address":
  39. {
  40. "street":"Sesame",
  41. "number":122
  42. }
  43. }
  44. """
  45. #нужна проверка, данный обработчик вообще должен от данного отправителя получать сообщения или нет. Подсказка ниже
  46. print (str(user).split("/")[0])
  47. if (str(user).split("/")[0]) in listen_to:
  48. print(text)
  49. print(mess)
  50. if text is not None:
  51. orderObj = jsreq.jsonToUser(text)
  52. tasklist.append(orderObj)
  53. bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n")
  54. def open_bot_template():
  55. #print("Тут описывается работа со списком задач в данном статусе")
  56. for i in range(len(tasklist)):
  57. #проверяем элемент tasklist[i]
  58. bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " " + str(i) + " " + "Заказ в очереди на обработке\n")
  59. print ("TASKLIST", tasklist[i])
  60. if (tasklist[i].address.number==122):
  61. bot.fake_api_call(tasklist[i].name)
  62. #он тут вроде не валится, но сообщения не приходят вроде
  63. bot.proceed_status("test@ej.sharix-app.org",tasklist[i])
  64. del tasklist[i]
  65. bot.bot_log(botname+" "+str(datetime.now())+" "+str(i)+" "+"Заказ успешно обработан и удален из очереди\n")
  66. print (tasklist)
  67. else:
  68. bot.bot_log(botname+" "+str(datetime.now())+" "+str(i)+" "+"Заказ оставлен в очереди\n")
  69. #Авторизация и запуск бота
  70. #пустой список заказов
  71. tasklist = []
  72. bot = JabberBot(JID, PASSWORD, PORT)
  73. #загружаем список заказов для обработки из базы (по идее, сюда попадаем в случае первого запуска или перезапуска)
  74. bot.fake_api_call("Получаем список заказов по названию бота-обработчика")
  75. bot.register_handler('message', message_handler)
  76. bot.start(open_bot_template)