193 lines
10 KiB
Python
193 lines
10 KiB
Python
import time
|
|
import vk_api
|
|
import random
|
|
import config as c
|
|
from threading import Thread
|
|
from modules import sqlite_methods as m
|
|
from vk_api.bot_longpoll import VkBotLongPoll, VkBotEventType
|
|
|
|
vk = vk_api.VkApi(token=c.g_token)
|
|
vk._auth_token()
|
|
vk.get_api()
|
|
longpoll = VkBotLongPoll(vk, c.group_id)
|
|
|
|
|
|
def get_variants(path):
|
|
with open(path, "r", encoding="utf-8") as f:
|
|
return f.readlines()
|
|
|
|
|
|
def send_msg(peer_id: int, message: str, attachment: str = ""):
|
|
return vk.method("messages.send", {**locals(), "random_id": 0})
|
|
|
|
|
|
message_user_ids = []
|
|
message_users = m.get_all_message_users()
|
|
for user in message_users:
|
|
message_user_ids.append(user[0])
|
|
|
|
photo_user_ids = []
|
|
photo_users = m.get_all_photo_users()
|
|
for user in photo_users:
|
|
photo_user_ids.append(user[0])
|
|
|
|
settings_result = []
|
|
settings = m.get_settings(2)
|
|
for value in settings:
|
|
settings_result.append(value)
|
|
|
|
MESSAGE_VARIANTS = get_variants("variants/message_variants.txt")
|
|
random.shuffle(MESSAGE_VARIANTS)
|
|
PHOTO_VARIANTS = get_variants("variants/photo_variants.txt")
|
|
random.shuffle(PHOTO_VARIANTS)
|
|
|
|
|
|
def hate():
|
|
message_index = 0
|
|
photo_index = 0
|
|
while True:
|
|
try:
|
|
for event in longpoll.listen():
|
|
if event.type == VkBotEventType.MESSAGE_NEW:
|
|
|
|
if event.obj.from_id in message_user_ids:
|
|
if message_index < len(MESSAGE_VARIANTS):
|
|
time.sleep(settings_result[1])
|
|
send_msg(event.obj.peer_id, f'{MESSAGE_VARIANTS[message_index]}')
|
|
message_index += 1
|
|
else:
|
|
message_index = 0
|
|
random.shuffle(MESSAGE_VARIANTS)
|
|
time.sleep(settings_result[1])
|
|
send_msg(event.obj.peer_id, f'{MESSAGE_VARIANTS[message_index]}')
|
|
message_index += 1
|
|
|
|
if event.obj.from_id in photo_user_ids:
|
|
if photo_index < len(PHOTO_VARIANTS):
|
|
time.sleep(settings_result[2])
|
|
send_msg(event.obj.peer_id, '', f'{PHOTO_VARIANTS[photo_index]}')
|
|
photo_index += 1
|
|
else:
|
|
photo_index = 0
|
|
random.shuffle(PHOTO_VARIANTS)
|
|
time.sleep(settings_result[2])
|
|
send_msg(event.obj.peer_id, '', f'{PHOTO_VARIANTS[photo_index]}')
|
|
photo_index += 1
|
|
except Exception as e:
|
|
print(repr(e))
|
|
|
|
|
|
def commands():
|
|
while True:
|
|
try:
|
|
for event in longpoll.listen():
|
|
if event.type == VkBotEventType.MESSAGE_NEW:
|
|
|
|
split_text = event.object.text.split(' ')
|
|
|
|
if split_text[0] in c.hate:
|
|
if len(split_text) == 2 and split_text[1] in c.hate_message_prefix:
|
|
if event.obj.from_id in c.g_admins:
|
|
if 'reply_message' in event.object:
|
|
user = event.obj.reply_message['from_id']
|
|
if user != -c.group_id:
|
|
if not m.is_message_user_hatelisted(user):
|
|
message_user_ids.append(user)
|
|
m.insert_message_hatelist(user)
|
|
send_msg(event.obj.peer_id, '✅ Пользователь добавлен в текстовый хейт лист')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Пользователь уже находится в текстовом хейт листе')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Невозможно добавить бота в хейт лист')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Пользователь не указан')
|
|
else:
|
|
choice = random.choice(c.ERRORS)
|
|
send_msg(event.obj.peer_id, f'{choice}')
|
|
|
|
elif len(split_text) == 2 and split_text[1] in c.hate_photo_prefix:
|
|
if event.obj.from_id in c.g_admins:
|
|
if 'reply_message' in event.object:
|
|
user = event.obj.reply_message['from_id']
|
|
if user != -c.group_id:
|
|
if not m.is_photo_user_hatelisted(user):
|
|
photo_user_ids.append(user)
|
|
m.insert_photo_hatelist(user)
|
|
send_msg(event.obj.peer_id, '✅ Пользователь добавлен в фото хейт лист')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Пользователь уже находится в фото хейт листе')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Невозможно добавить бота в хейт лист')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Пользователь не указан')
|
|
else:
|
|
choice = random.choice(c.ERRORS)
|
|
send_msg(event.obj.peer_id, f'{choice}')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Неверный формат команды')
|
|
|
|
if split_text[0] in c.unhate:
|
|
if len(split_text) == 2 and split_text[1] in c.hate_message_prefix:
|
|
if event.obj.from_id in c.g_admins:
|
|
if 'reply_message' in event.object:
|
|
user = event.obj.reply_message['from_id']
|
|
if m.is_message_user_hatelisted(user):
|
|
message_user_ids.remove(user)
|
|
m.delete_message_hatelist(user)
|
|
send_msg(event.obj.peer_id, '✅ Пользователь удален из текстового хейт листа')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Пользователь отсутствует в текстовом хейт листе')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Пользователь не указан')
|
|
else:
|
|
choice = random.choice(c.ERRORS)
|
|
send_msg(event.obj.peer_id, f'{choice}')
|
|
|
|
elif len(split_text) == 2 and split_text[1] in c.hate_photo_prefix:
|
|
if event.obj.from_id in c.g_admins:
|
|
if 'reply_message' in event.object:
|
|
user = event.obj.reply_message['from_id']
|
|
if m.is_photo_user_hatelisted(user):
|
|
photo_user_ids.remove(user)
|
|
m.delete_photo_hatelist(user)
|
|
send_msg(event.obj.peer_id, '✅ Пользователь удален из фото хейт листа')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Пользователь отсутствует в фото хейт листе')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Пользователь не указан')
|
|
else:
|
|
choice = random.choice(c.ERRORS)
|
|
send_msg(event.obj.peer_id, f'{choice}')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Неверный формат команды')
|
|
|
|
if split_text[0] in c.cooldown:
|
|
if len(split_text) == 3 and split_text[1] in c.hate_message_prefix and split_text[2].isnumeric():
|
|
if event.obj.from_id in c.g_admins:
|
|
settings_result.pop(1)
|
|
settings_result.insert(1, int(split_text[2]))
|
|
m.set_message_cooldown(int(split_text[2]), 2)
|
|
send_msg(event.obj.peer_id, f'✅ Задержка текста изменена на {split_text[2]} секунд')
|
|
else:
|
|
choice = random.choice(c.ERRORS)
|
|
send_msg(event.obj.peer_id, f'{choice}')
|
|
|
|
elif len(split_text) == 3 and split_text[1] in c.hate_photo_prefix and split_text[2].isnumeric():
|
|
if event.obj.from_id in c.g_admins:
|
|
settings_result.pop(2)
|
|
settings_result.insert(2, int(split_text[2]))
|
|
m.set_photo_cooldown(int(split_text[2]), 2)
|
|
send_msg(event.obj.peer_id, f'✅ Задержка фото изменена на {split_text[2]} секунд')
|
|
else:
|
|
choice = random.choice(c.ERRORS)
|
|
send_msg(event.obj.peer_id, f'{choice}')
|
|
else:
|
|
send_msg(event.obj.peer_id, '❎ Неверный формат команды')
|
|
except Exception as e:
|
|
print(repr(e))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
Thread(target=hate, args=[]).start()
|
|
Thread(target=commands, args=[]).start()
|