-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmain.py
43 lines (35 loc) · 1.33 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import telegram
from flask import Flask
from flask import request
from queue import Queue
from telegram.ext import Filters, MessageHandler, CallbackQueryHandler
from telegram.ext import Dispatcher
from tools import upload_handler, callback_handler, error_handler, upload_sticker_handler
from credentials import TOKEN, APP_URL
from requests_toolbelt.adapters import appengine
appengine.monkeypatch()
app = Flask(__name__)
global bot
bot = telegram.Bot(token=TOKEN)
dp = Dispatcher(bot, Queue(), workers=10)
dp.add_handler(MessageHandler(Filters.document, upload_handler))
dp.add_handler(MessageHandler(Filters.photo, upload_handler))
dp.add_handler(MessageHandler(Filters.sticker, upload_sticker_handler))
dp.add_handler(CallbackQueryHandler(callback_handler))
dp.add_error_handler(error_handler)
@app.route('/HOOK', methods=['POST', 'GET'])
def webhook_handler():
if request.method == "POST":
# retrieve the message in JSON and then transform it to Telegram object
update = telegram.Update.de_json(request.get_json(force=True), bot)
dp.process_update(update)
return 'ok'
@app.route('/set_webhook', methods=['GET', 'POST'])
def set_webhook():
s = bot.setWebhook(APP_URL + '/HOOK')
if s:
return "webhook setup ok"
else:
return "webhook setup failed"
if __name__ == '__main__':
app.run()