open_servicerequest_booked.py 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  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_servicerequest_booked"
  16. operating_status=211
  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=["9090@ej.sharix-app.org","test@ej.sharix-app.org", "open_template_bot@ej.sharix-app.org","799999999990@ej.sharix-app.org"]
  23. #тут хранится список jid, кому бот может отправлять сообщения в результате обработки заявки
  24. proceed_to=["open_servicerequest_assigned@ej.sharix-app.org", "open_servicerequest_pending@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. if (str(user).split("/")[0]) in listen_to:
  32. # print (str(user).split("/")[0])
  33. if text is not None:
  34. orderObj = jsreq.jsonToOrderTicket(text)
  35. print (orderObj)
  36. tasklist.append(orderObj)
  37. bot.bot_log(str(datetime.now())+" Поступивший заказ успешно добавлен в очередь обработки\n")
  38. def open_servicerequest_booked_wait(period,localtask):
  39. bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ ожидает\n")
  40. time.sleep(period)
  41. #ВАЖНО! Надо скорее всего через API редактировать заказ, иначе возможна потеря данных
  42. localtask.title=localtask.title+"1"
  43. #обязательно данный обработчик должен заканчиваться передачей заказа куда-то на обработку дальше - обратно или другому, иначе оно потеряется
  44. bot.proceed_status(proceed_to[0],localtask)
  45. bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ подождал и возвращен в очередь\n")
  46. def open_servicerequest_booked_process(localtask):
  47. bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ в обработчике\n")
  48. order = jsreq.jsonToOrder(localtask)
  49. #Пока заглушка
  50. balance = get_balance(order.client_id)
  51. price = order.real_price
  52. if balance > price:
  53. #Заказ на основного водителя?
  54. if order.provider:
  55. bot.proceed_status(proceed_to[1], localtask)
  56. bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан\n")
  57. else:
  58. # Проверка срочности заказа
  59. if order.asap:
  60. #Заглушка назначение исполнителя системой
  61. if set_provider(order.ticket, order.provider) == "success":
  62. bot.proceed_status(proceed_to[0], localtask)
  63. else:
  64. bot.bot_log(botname+" "+str(datetime.now())+" "+"Ошибка назначения исполнителя\n")
  65. else:
  66. bot.proceed_status(proceed_to[1], localtask)
  67. bot.bot_log(botname + " " + str(datetime.now()) + " " + "Заказ успешно обработан\n")
  68. else:
  69. # Заглушка получения оплаты от клиента
  70. pass
  71. def get_balance(client):
  72. headers = {'Authorization': f'Token {token}'}
  73. # Передаем id пользователя и получаем его баланс
  74. response = requests.get(config.API_URL+f"/api/example/{client}", headers=headers)
  75. print(response.json())
  76. ordertext=response.json()
  77. if not ordertext:
  78. print("Empty array")
  79. else:
  80. print(ordertext[0], "PROCESSING")
  81. try:
  82. balance = ordertext[0].get('balance')
  83. return balance
  84. except Exception as ex:
  85. print(ex)
  86. def set_provider(ticket, provider):
  87. headers = {'Authorization': f'Token {token}'}
  88. data = {'ticket': ticket,
  89. 'provider': provider}
  90. response = requests.post(config.API_URL+f"/api/example/set-provider", headers=headers, data=data)
  91. print(response.json())
  92. ordertext=response.json()
  93. if not ordertext:
  94. print("Empty array")
  95. else:
  96. print(ordertext[0], "PROCESSING")
  97. try:
  98. message = ordertext[0].get('message')
  99. return message
  100. except Exception as ex:
  101. print(ex)
  102. def open_servicerequest_booked():
  103. while (len(tasklist)):
  104. localtask=tasklist.pop(0)
  105. bot.bot_log(botname + " " + datetime.now().strftime('%Y-%m-%d') + " "+ "Заказ в очереди на обработке\n")
  106. print("EACH TASKLIST", tasklist)
  107. # if (localtask.status!=operating_status):
  108. # bot.proceed_status(proceed_to[-1],localtask)
  109. # bot.bot_log(botname+" "+str(datetime.now())+" "+"Заказ не по адресу, перенаправляем на форсмажор\n")
  110. # continue
  111. # if (localtask.title!="10"):
  112. # t1=threading.Thread(target=open_servicerequest_booked_wait, args=(5,localtask))
  113. # t1.start()
  114. # if (localtask.title=="101"):
  115. # t1=threading.Thread(target=open_servicerequest_booked_process, args=(10,localtask))
  116. # t1.start()
  117. t1=threading.Thread(target=open_servicerequest_booked_process, args=(localtask))
  118. t1.start()
  119. #если никакие обработчики не подошли - отправляем обратно в очередь
  120. bot.proceed_status(proceed_to[0],localtask)
  121. print("Заказ возвращен в очередь\n")
  122. print(tasklist)
  123. #Авторизация и запуск бота
  124. #пустой список заказов
  125. tasklist = []
  126. #надо инициализировать tasklist при запуске из API
  127. token = jsreq.requestGetToken(botname, PASSWORD, config.API_URL+"/api/v1/auth/token/login/")
  128. #ВАЖНО! Надо еще фильтровать список по статусу обработки заявки
  129. tasklist=jsreq.requestGetList(token, config.API_URL+"/metaservicesynced/api/orders/")
  130. print (len(tasklist))
  131. bot = JabberBot(JID, PASSWORD, PORT)
  132. bot.register_handler('message', message_handler)
  133. bot.start(open_servicerequest_booked)