-
Notifications
You must be signed in to change notification settings - Fork 0
/
echobot.py
60 lines (51 loc) · 1.87 KB
/
echobot.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import urllib
import time
import json
import requests
TOKEN = "<your-bot-token>"
URL = "https://api.telegram.org/bot{}/".format(TOKEN)
def get_url(url):
"""Downloads the content from a URL and gives us a string"""
response = requests.get(url)
content = response.content.decode("utf8")
return content
def get_json_from_url(url):
"""Gets the string response as above and parses this into a Python dictionary using json.loads()"""
content = get_url(url)
js = json.loads(content)
return js
def get_updates(offset=None):
"""Call this command https://api.telegram.org/bot<your-bot-token>/getUpdates and retrieves a list of updates"""
url = URL + "getUpdates?timeout=100"
if offset:
#If this is specified, we'll pass it along to the Telegram API to indicate that we don't want to receive any messages with smaller IDs this.
url += "&offset={}".format(offset)
js = get_json_from_url(url)
return js
def get_last_update_id(updates):
"""Calculates the highest ID of all the updates we receive from getUpdates"""
update_ids = []
for update in updates["result"]:
update_ids.append(int(update["update_id"]))
return max(update_ids)
def send_message(text,chat_id):
"""Takes the text of the message we want to send (text) and the chat ID of the chat where we want to send the message (chat_id)"""
text = urllib.quote_plus(text)
url = URL + "sendMessage?text={}&chat_id={}".format(text, chat_id)
get_url(url)
def echo_all(updates):
"""Send an echo reply for each message that we receive"""
for update in updates["result"]:
text = update["message"]["text"].encode('utf8')
chat = update["message"]["chat"]["id"]
send_message(text, chat)
def main():
last_update_id = None
while True:
print("getting updates")
updates = get_updates(last_update_id)
if len(updates["result"]) > 0:
last_update_id = get_last_update_id(updates) + 1
echo_all(updates)
if __name__ == '__main__':
main()