testBot.py 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. #Библиотеки
  2. import xmpp
  3. from config import jid, jidpassword
  4. from BotClass import JabberBot
  5. from RequestClass import RequestDataBase
  6. requesDB = RequestDataBase()
  7. #Основной метод для обработки сообщинй от пользователя
  8. def message_handler(conn, mess):
  9. text = mess.getBody()
  10. user = mess.getFrom()
  11. if str(text).startswith("!"):
  12. commands_request(conn, mess)
  13. else:
  14. conn.send(xmpp.Message(user, "Я тебя не понимаю..."))
  15. def commands_request(conn, mess):
  16. try:
  17. text = mess.getBody()
  18. user = mess.getFrom()
  19. arguments = str(text).split()
  20. if text == "!addcl":
  21. conn.send(xmpp.Message(user, requesDB.add_car_log(arguments[0])))
  22. elif text == "!delcl":
  23. conn.send(xmpp.Message(user, requesDB.del_car_log(int(arguments[0]))))
  24. elif text == '!ссt':
  25. conn.send(xmpp.Message(user, requesDB.change_current_token(arguments[0], arguments[1])))
  26. elif text == '!addol':
  27. conn.send(xmpp.Message(user, requesDB.add_order_log()))
  28. elif text == '!delol':
  29. conn.send(xmpp.Message(user, requesDB.del_order_log(arguments[0])))
  30. elif text == '!addc':
  31. conn.send(xmpp.Message(user, requesDB.add_car(arguments[0])))
  32. elif text == '!cc':
  33. conn.send(xmpp.Message(user, requesDB.change_car(arguments[0])))
  34. elif text == '!gup':
  35. conn.send(xmpp.Message(user, requesDB.get_upcoming_orders(arguments[0])))
  36. elif text == '!ac':
  37. conn.send(xmpp.Message(user, requesDB.actuality_check(arguments[0])))
  38. elif text == '!ico':
  39. conn.send(xmpp.Message(user, requesDB.insert_current_order(arguments[0])))
  40. elif text == '!dco':
  41. conn.send(xmpp.Message(user, requesDB.del_current_order(arguments[0])))
  42. elif text == '!uco':
  43. conn.send(xmpp.Message(user, requesDB.update_current_order(arguments[0])))
  44. elif text == '!cs':
  45. conn.send(xmpp.Message(user, requesDB.change_status(arguments[0])))
  46. else:
  47. conn.send(xmpp.Message(user, "Такой команды нет"))
  48. except Exception as ex:
  49. conn.send(xmpp.Message(user, str(ex)))
  50. #Авторизация и запуск бота
  51. bot = JabberBot(jid, jidpassword)
  52. bot.register_handler('message', message_handler)
  53. bot.start()