forked from sunDalik/vk-bot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrndmsg_bot.py
executable file
·131 lines (113 loc) · 6.15 KB
/
rndmsg_bot.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
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
121
122
123
124
125
126
127
128
129
130
131
import vk_api
from vk_api.bot_longpoll import VkBotLongPoll, VkBotEventType
import random
import re
import sys
import os
import json
from config import token, group_id
import requests
import img2msg
import memgen
import traceback
import markovify
vk_session = vk_api.VkApi(token=token)
vk = vk_session.get_api()
with open("mkmodel.json") as f:
model_json = f.read()
model = markovify.Text.from_json(model_json)
def send_message(peer, msg="", random=random.randint(-2147483648, 2147483647), attachment=None):
if attachment is not None:
vk.messages.send(peer_id=peer, message=msg, random_id=random, attachment=attachment)
elif msg is not None and msg.strip() != "":
vk.messages.send(peer_id=peer, message=msg, random_id=random)
def upload_photo(photo_path):
img = {'photo': open(photo_path,"rb")}
upload_resp = json.loads(requests.post(vk.photos.getMessagesUploadServer()['upload_url'], files=img).text)
photo_id = upload_resp['photo']
if photo_id is not 'null':
photo_id = vk.photos.saveMessagesPhoto(photo=upload_resp['photo'], server=upload_resp['server'], hash=upload_resp['hash'])[0]['id']
attachment = str('photo-'+str(group_id)+'_'+str(photo_id))
return attachment
else:
return None
def rndmsg_mode(msg_list, mentions):
if not mentions:
mentions = ['rndmsg']
joker = ['джокер', 'joker', 'джокера', 'джокеру', 'джокером', 'джокере']
joker_images = ["joker/" + img for img in os.listdir("joker")]
mentions_re = re.compile(r"\b(?:" + "|".join(mentions) + r")\b")
endings_re = re.compile(r"(?:ах|а|ев|ей|е|ов|о|иях|ия|ие|ий|й|ь|ы|ии|и|ях|я|у)$")
opinions_re = re.compile(r"\b(?:" + "|".join(mentions) + r") (?:что (?:ты )?думаешь (?:об?|про)|как тебе|тво[её] мнение об?|ваше мнение об?|как (?:ты )?относишься к) (.+?)\b")
joker_re = re.compile(r"\b(?:" + "|".join(joker) + r") ?(\++|\b)")
shitpost_re = re.compile(r"\bприкол\w*(?: про (.+?))?\b")
while True:
longpoll = VkBotLongPoll(vk_session, group_id)
try:
for event in longpoll.listen():
if event.type == VkBotEventType.MESSAGE_NEW:
e = event.object
attachments = list(filter(lambda a: a.get("type") =="photo", e.attachments))
# generate messages with markov chain
if shitpost_re.search(e.text.lower()):
if shitpost_re.search(e.text.lower()).group(1) is not None:
message = model.make_sentence_with_start(shitpost_re.search(e.text.lower()).group(1), strict=False, max_words=15)
else:
message = model.make_short_sentence(115)
send_message(e.peer_id, message)
# if mention + what do you think of X then reply with random message about X
elif opinions_re.search(e.text.lower()):
thing = opinions_re.search(e.text.lower()).group(1)
thing = endings_re.sub("", thing)
list_with_thing = list(m for m in msg_list if thing in m)
message = random.choice(list_with_thing) if len(list_with_thing) != 0 else "Не знаю"
send_message(e.peer_id, message)
# on joker command send joker meme
elif joker_re.search(e.text.lower()):
word_count = len(joker_re.search(e.text.lower()).group(1))
memgen.make_meme(random.choice(joker_images), "temp.jpg", msg_list, word_count)
photo = upload_photo("temp.jpg")
send_message(e.peer_id, attachment=photo)
# on mention + photo reply to photo
elif (mentions_re.search(e.text.lower()) or (e.reply_message and e.reply_message.get("from_id") == -group_id)) and len(attachments) != 0:
img_resp = vk_session.http.get(attachments[0].get("photo").get("sizes")[-1].get("url"), allow_redirects=True)
open('temp', 'wb').write(img_resp.content)
send_message(e.peer_id, img2msg.get_msg(msg_list, "temp"))
# on mention or reply send random message
elif mentions_re.search(e.text.lower()) or (e.reply_message and e.reply_message.get("from_id") == -group_id):
send_message(e.peer_id, random.choice(msg_list))
# on photo with no mentions has a chance to reply to it
elif len(attachments) != 0 and random.randint(1,20) == 1:
img_resp = vk_session.http.get(attachments[0].get("photo").get("sizes")[-1].get("url"), allow_redirects=True)
open('temp', 'wb').write(img_resp.content)
send_message(e.peer_id, img2msg.get_msg(msg_list, "temp"))
# has a chance to reply to any message
elif random.randint(1, 50) == 1:
send_message(e.peer_id, random.choice(msg_list))
except requests.exceptions.ReadTimeout as timeout:
continue
except Exception as e:
print(traceback.format_exc())
def main():
if len(sys.argv) >= 2:
cf = open(sys.argv[1], "r")
mentions = []
for i, line in enumerate(cf):
line = line.rstrip("\n")
if i == 0:
msgs_file = line
elif i == 1:
delimiter = line
else:
mentions.append(line)
f = open(msgs_file, "r")
msgs = f.read()
msg_list = msgs.split(delimiter)
f.close()
print("You are all set! Bot is working...")
rndmsg_mode(msg_list, mentions)
else:
print("Specify config file as a first argument!")
print("(First line - filename; second line - delimiter; then mentions, each on a new line)")
exit(1)
main()