working bot

Author: vistrcm

.gitignore

@@ -1,3 +1,4 @@
 # Created by .ignore support plugin (hsz.mobi)
 .idea/
 *.pyc
+katrinbot_settingsprod.py

katrinbot.py

@@ -1,5 +1,12 @@
+# -*- coding: utf-8 -*-
 import logging
+
 from flask import Flask
+from flask import jsonify
+from flask import request
+
+from siteworker import get_latest, get_random
+from telegram import Bot
 
 app = Flask(__name__)
 app.config.from_object('katrinbot_settings')
@@ -12,14 +19,40 @@ def index():
     return "index page"
 
 
-@app.route("/wh/%s" % app.config['TOKEN'])
+def processwh(msg):
+    repl_url = "http://lene.pois.org.ru/Katrin/img/{}"
+
+    user = msg["message"]["from"]["id"]
+    message_id = msg["message"]["message_id"]
+    text = msg["message"]["text"]
+
+    if text in ["/start", "/help"]:
+        response = "Используй команды /random или /latest."
+    elif text == "/latest":
+        item = get_latest()
+        response = repl_url.format(item)
+    elif text == "/random":
+        item = get_random()
+        response = repl_url.format(item)
+    else:
+        response = "Шо?"
+
+    reply_markup = {"keyboard": [["/latest"], ["/random"]]}
+    bot.send_message(user, response, reply_markup=reply_markup)
+    return response
+
+
+@app.route("/wh/%s" % app.config['TOKEN'], methods=['GET', 'POST'])
 def webhook():
     app.logger.debug("got message")
-    return "ok"
-
+    if request.method == 'POST':
+        msg = request.get_json()
+        app.logger.debug(msg)
+        answer = processwh(msg)
+        return jsonify(status="ok", answer=answer)
+    else:
+        return jsonify(status="ok")
 
-def register_webhook():
-    return "hh"
 
 if __name__ == "__main__":
     if not app.debug:
@@ -29,5 +62,8 @@ def register_webhook():
         formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
         ch.setFormatter(formatter)
         app.logger.addHandler(ch)
-        app.logger.setLevel(logging.INFO)
+        app.logger.setLevel(logging.DEBUG)
+    bot = Bot(app.config['BOT_TOKEN'])
+    wh_result = bot.register_webhook(WH_URL)
+    app.logger.debug(wh_result)
     app.run()

katrinbot_settings.py

@@ -0,0 +1,4 @@
+DEBUG = False
+URL = "https://vitko.info/katrinphotobot/wh/"
+TOKEN = "yourtokenhere"
+BOT_TOKEN = "yourbottokenhere"

siteworker.py

@@ -0,0 +1,19 @@
+import requests
+import random
+
+
+def get_items():
+    r = requests.get("http://lene.pois.org.ru/Katrin/items")
+    return [_["img"] for _ in r.json()]
+
+
+def get_sorted_items():
+    return sorted(get_items())
+
+
+def get_latest():
+    return get_sorted_items()[-1]
+
+
+def get_random():
+    return random.choice(get_items())

telegram.py

@@ -0,0 +1,33 @@
+import logging
+import requests
+
+
+logger = logging.getLogger(__name__)
+
+
+class Bot(object):
+    def __init__(self, token):
+        self.__token = token
+        self.__bot_url = "https://api.telegram.org/bot{bottoken}".format(bottoken=self.__token)
+
+    def __send_request(self, method_name, payload):
+        uri = "{bot_url}/{method_name}".format(bot_url=self.__bot_url, method_name=method_name)
+        r = requests.post(uri, json=payload)
+        return r.json()
+
+    def send_message(self, chat_id, text, reply_to_message_id=None, reply_markup=None):
+        logger.warn("sending message to %s", chat_id)
+        print("reply_markup: %s" % reply_markup)
+        payload = {
+            "chat_id": chat_id,
+            "text": text,
+            "reply_to_message_id": reply_to_message_id,
+            "reply_markup": reply_markup
+        }
+        return self.__send_request("sendMessage", payload)
+
+    def register_webhook(self, url):
+        logger.info("Trying to unregister webhook")
+        logger.info(self.__send_request("setWebhook", {"url": ""}))
+        logger.info("Setting webhook to: %s" % url)
+        return self.__send_request("setWebhook", {"url": url})