rSlashBot/app.py at main · GameGodS3/rSlashBot · GitHub

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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
from flask import Flask, request
import json
import requests
import telegram
from rslashbot.credentials import botToken, botUsername, herokuURL

# Bot API Token
global bot
global TOKEN
TOKEN = botToken
bot = telegram.Bot(token=TOKEN)

# Init Flask App
app = Flask(__name__)

#Response Function to respond to Telegram API calls


@app.route('/{}'.format(TOKEN), methods=['POST'])
def respond():
    """
    Responds to slash commands issued to bot via Telegram
    """

    # Retrieve message in JSON and transform it to Telegram object
    update = telegram.Update.de_json(request.get_json(force=True), bot)
    chat_id = update.effective_message.chat.id
    msg_id = update.effective_message.message_id

    # UTF-8 formatting
    text = update.message.text.encode('utf-8').decode()
    # for debugging
    print("Got text message: ", text)

    ### Slash Command
    # Welcome message / Start Message
    if text == "/start":
        bot_welcome = """
        Hello. This is rSlashBot. Give me a subreddit from which you wish to receive updates from.
        """
        bot.sendMessage(chat_id=chat_id, text=bot_welcome,
                        reply_to_message_id=msg_id)

    else:
        try:
            # Clear non-alphabets from message
            #text = re.sub(r"\W", "_", text)
            # create the api link for the avatar based on http://avatars.adorable.io/
            if text[0:2] == "r/":
                text = text[2:]
            bot.sendMessage(chat_id=chat_id,
                            text="Hold a sec. Fetching from r/{}".format(text))
            url = "https://www.reddit.com/r/{}.json".format(text)
            r = requests.get(
                url, headers={'User-agent': 'rSlashBot'}, allow_redirects=True)
            with open('reddit.json', 'wb') as f:
                f.write(r.content)
            with open('reddit.json', 'r') as f:
                l = json.load(f)

            i = 0
            while l["data"]["children"][i]["data"]["stickied"] == True:
                i += 1

            # Sends text content
            title = l["data"]["children"][i]["data"]["title"]
            content = l["data"]["children"][i]["data"]["selftext"]
            msg_text = "*"+title+"*"+"\n\n"+content
            if len(msg_text) > 4096:
                for x in range(0, len(msg_text), 4096):
                    if x == 0:
                        bot.sendMessage(
                            chat_id=chat_id, text=msg_text[x:x+4096], parse_mode='Markdown', reply_to_message_id=msg_id)
                    else:
                        bot.sendMessage(
                            chat_id=chat_id, text=msg_text[x:x+4096], parse_mode='Markdown')
            else:
                bot.sendMessage(chat_id=chat_id, text=msg_text,
                                parse_mode='Markdown', reply_to_message_id=msg_id)

            # For sending images or gifs
            latest = l["data"]["children"][i]["data"]
            format = latest["url"][-3:]
            formats = ['jpg', 'png']
            if format in formats:
                bot.sendPhoto(
                    chat_id=chat_id, photo=latest["url"])
            elif format == 'gif':
                bot.sendAnimation(chat_id=chat_id, animation=latest["url"])

            bot.sendMessage(chat_id, "http://www.reddit.com" +
                            latest["permalink"])

        except Exception as e:
            # if things went wrong
            bot.sendMessage(
                chat_id=chat_id, text="I am sorry. There was an error fetching from that subreddit.", reply_to_message_id=msg_id)
            print(e)

    return 'ok'


@app.route('/set_webhook', methods=['GET', 'POST'])
def set_webhook():
    s = bot.set_webhook('{URL}{HOOK}'.format(URL=herokuURL, HOOK=TOKEN))

    # debugging prints
    if s:
        return "Webhook setup OK"
    else:
        return "Webhook setup failed"


@app.route('/')
def index():
    return '.'


if __name__ == '__main__':
    app.run(threaded=True)