-
Notifications
You must be signed in to change notification settings - Fork 0
/
chatbot.py
91 lines (80 loc) · 2.92 KB
/
chatbot.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
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
#this is a sample chatbot that conducts very simple operation with messages
#this operation was meant to be replaced by text analysis
#the bot should recieve messages from operator-client chat and respond to operator-bot chat (giving advise)
#importing libraries to recieve, process and send messages in Telegram
import json
import requests
import time
import urllib
import sqlite3
#library for sample operation
from PyDictionary import PyDictionary
#api token
TOKEN = "api token of bot (removed)"
#url template
URL = "https://api.telegram.org/bot{}/".format(TOKEN)
#download information from url
def get_url(url):
response = requests.get(url)
content = response.content.decode("utf8")
return content
#obtain information in json-format
def get_json_from_url(url):
content = get_url(url)
js = json.loads(content)
return js
#list messages which are sent to the bot (get only recent messages according to offset value)
#save resources due to timeout parameter
def get_updates(offset=None):
url = URL + "getUpdates?timeout=100"
if offset:
url += "&offset={}".format(offset)
js = get_json_from_url(url)
return js
#get the latest update ID (to reply all meassages including duplicates)
def get_last_update_id(updates):
update_ids = []
for update in updates["result"]:
update_ids.append(int(update["update_id"]))
return max(update_ids)
#obtain chat id and latest messages in the chat
def get_last_chat_id_and_text(updates):
num_updates = len(updates["result"])
last_update = num_updates - 1
text = updates["result"][last_update]["message"]["text"]
chat_id = updates["result"][last_update]["message"]["chat"]["id"]
return (text, chat_id)
#sending message to the selected chat
def send_message(text, chat_id):
text = urllib.parse.quote_plus(text)
# url = URL + "sendMessage?text={}&chat_id={}".format(text, chat_id)
url = URL + "sendMessage?text={}&chat_id={}".format(text, #id of operator-bot chat)
# print(chat_id)
get_url(url)
#reply to all recieved message
def echo_all(updates):
for update in updates["result"]:
try:
text = update["message"]["text"]
chat = update["message"]["chat"]["id"]
#performing sample operation with text (in this example the bot sends antonyms to english words)
dictionary = PyDictionary()
text = str(dictionary.antonym(text)[1])
send_message(text, chat)
# print(text)
except Exception as e:
print(e)
#final replying function
def main():
db.setup()
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)
time.sleep(0.5)
#activation
if __name__ == '__main__':
main()