Переглянути джерело

running schema according to previous task

mex3 1 рік тому
батько
коміт
b9fdab358f
4 змінених файлів з 66 додано та 68 видалено
  1. 14 2
      BotClass.py
  2. 0 61
      botrun.py
  3. 3 2
      config_template.py
  4. 49 3
      open-st_request-assigned.py

+ 14 - 2
BotClass.py

@@ -29,11 +29,23 @@ class JabberBot:
         except KeyboardInterrupt: return 0
         return 1
 
+    def fake_api_call(message):
+        cli.send_message(self.user, self.password, "test@msg.sharix-app.org", message)
+
+    def bot_log(message):
+        #далее надо будет переделать запись в файл вероятно
+        cli.send_message(self.user, self.password, "test@msg.sharix-app.org", message)
+
+# recipient - это конкретный jid следующего обработчика заявки, ticket - json с заявкой.
+    def proceed_status(recipient, ticket):
+        #тут могут быть различные проверки дополнительные, а так вообще эта функция нужна для передачи заявки на обработку следующему
+        cli.send_message(self.user, self.password, recipient, ticket)
+
 #сюда можно как параметры передавать переменную для значения Process в вызываемой функции, а также передавать информацию о том, какой из обработчиков ввызывать в бесконечном цикле
-    def start(self, handler):
+    def start(self, handler, *args):
         self.conn.sendInitPresence()#статус аккаунта бота (активен или нет)
         print("Bot started!")
         while self.step_on():
 #это вызов конкретной функции для обработки статуса конкретным ботом
-		handler()
+		handler(args)
 		pass

+ 0 - 61
botrun.py

@@ -1,61 +0,0 @@
-import xmpp
-from xmpp import cli
-import config
-from BotClass import JabberBot
-import json
-
-#В этом файле описаны функции, необходимые для запуска любого произвольного бота-обработчика заявок/статусов
-
-# загрузка json файла
-def read_inf(file_name):
-    with open(file_name, 'r', encoding="utf-8") as file:
-        return json.load(file)
-
-
-# обработчик входящих сообщений
-def message_handler(conn, mess):
-    text = mess.getBody()#текст сообщения боту
-    user = mess.getFrom()#отправитель сообщения
-#работа с файлом в таком виде для входящей информации не нужна, так как бот должен обрабатывать входящие сообщения, а не файлы на сервере, и не сохраняет входящие сообщения в файлы. Он должен входящее сообщение поместить в очередь (список) на обработку.
-# При запуске бота он инициализирует пустой список, далее делает запрос к API для выгрузки в него всех заявок с его статусом для обработки и все элементы (если они есть) добавляет с помощью append в список.
-#При получении сообщения от кого-нибудь он также делает в список append
-#Каждлый пустой (не существующий) вызов API описывается как fake_api_call
-#Элемент списка удаляется, если с момента timestamp внутри json прошло больше 1 часа (соответственно json должен быть такого вида, что эта информация там есть)
-    applications = read_inf("applications.json")
-    print(text)
-    print(mess)
-
-    if text is not None:
-        message = xmpp.Message()
-        applications = read_inf("applications.json")
-
-        #если бот получает сообщение, соответсвующее значению id("1","2","3")
-        #он отправит описание desc этого определенного объекта 
-        for app in applications["applications"]:
-            if app.get("id") == text:
-                message.setBody(app.get("desc"))
-
-        # если бот получает сообщение, он отправляет такое же сообщение отправителю
-        # message.setBody(text)
-        message.setFrom(config.JID)
-        message.setTo(user)
-        message.setType('chat')
-        
-        conn.send(message)
-
-# пример функции, которую можно вызвать из любой части кода для отправки сообщения от бота пользователю
-def simple_send_message():
-    cli.send_message(config.JID, config.PASSWORD, "test@msg.sharix-app.org", "request.data") 
-
-# recipient - это конкретный jid следующего обработчика заявки, ticket - json с заявкой.
-#С помощью sender мы отвязываемся от конфига на каждого обработчика и даем возможность для реализации новой схемы
-def proceed_status(recipient, ticket, sender=config.JID):
-    cli.send_message(sender, config.PASSWORD, recipient, ticket) 
-
-# в message помещается смысл и сам вызов API, который должен быть, но которого нет, а также хорошо туда добавить номер заказа (взять из json) и название бота
-# скорее всего эту функцию и подобные надо будет перенести в отдельный класс или библиотеку
-def fake_api_call(message):
-    cli.send_message(config.JID, config.PASSWORD, "test@msg.sharix-app.org", message) 
-
-
-

+ 3 - 2
config_template.py

@@ -1,8 +1,9 @@
 #rename this file to config.py
-JID="example@msg.sharix-app.org"
+#JID="example@msg.sharix-app.org"
+JSERVER="msg.sharix-app.org"
 PASSWORD="12345"
 #первое время пусть пароль будет у всех обработчиков одинаковый, далее надо будет генерировать уникальные связки на основе логина и ключа, указанного в файле.
 
 # бот в конверс
 # JID="bottickets@msg.sharix-app.org"
-# PASSWORD="12345!"
+# PASSWORD="12345!"

+ 49 - 3
open-st_request-assigned.py

@@ -4,14 +4,60 @@ import config
 from BotClass import JabberBot
 import json
 import botrun
+#раскомментировать, когда перейдем на разные пароли
+#import hashlib
+import datetime
 
 #В этом файле содержится только то, что запускает конкретный обработчик
 
-def open-st_request-assigned():
+botname="open-st_request-assigned"
+JID=botname+"@"+config.JSERVER
+PASSWORD=config.PASSWORD
+#раскомментировать, когда перейдем на разные пароли
+#PASSWORD=hashlib.md5((botname+config.PASSWORD).encode('utf-8')).hexdigest()
+#таким образом хранится список jid, от которых можно получать сообщения этому боту
+listen=["test@msg.sharix-app.org", "test1@msg.sharix-app.org"]
+
+
+# обработчик входящих сообщений
+def message_handler(conn, mess):
+    text = mess.getBody()#текст сообщения боту
+    user = mess.getFrom()#отправитель сообщения
+   
+    #нужна проверка, данный обработчик вообще должен от данного отправителя получать сообщения или нет. Подсказка ниже
+    #if user in listen:
+
+    print(text)
+    print(mess)
+
+    if text is not None:
+        message = xmpp.Message()
+        
+        tasklist.append(message)
+        bot_log(" ".join(datetime.now(), i, "Заказ успешно добавлен в очередь обработки"))
+
+
+def open-st_request-assigned(tasklist):
     print("Тут описывается работа со списком задач в данном статусе")
+    for i in range(len(tasklist()):
+        #проверяем элемент tasklist[i]
+        bot_log(" ".join(botname, datetime.now(), i, "Заказ в очереди на обработке"))
+        if tasklist[i]:
+            proceed_status("test@msg.sharix-app.org", tasklist[i])
+            del tasklist[i]
+            bot_log(" ".join(botname, datetime.now(), i, "Заказ успешно обработан и удален из очереди"))
+        else:
+            bot_log(" ".join(botname, datetime.now(), i, "Заказ оставлен в очереди"))
 
 #Авторизация и запуск бота
-bot = JabberBot(config.JID, config.PASSWORD)
+
+#пустой список заказов
+tasklist = []
+
+#загружаем список заказов для обработки из базы (по идее, сюда попадаем в случае первого запуска или перезапуска)
+fake_api_call("Получаем список заказов по названию бота-обработчика")
+
+bot = JabberBot(JID, PASSWORD)
 bot.register_handler('message', message_handler)
-bot.start(open-st_request-assigned)
+bot.start(open-st_request-assigned, tasklist)