133 lines
4.8 KiB
Python
133 lines
4.8 KiB
Python
import os
|
||
from random import randrange
|
||
|
||
import telebot
|
||
from cachetools import TTLCache
|
||
from telebot.types import Message
|
||
|
||
import settings
|
||
from mongo import mongo
|
||
|
||
bot = telebot.TeleBot(os.getenv("TELEGRAM_TOKEN"))
|
||
|
||
|
||
all_letters = "йцукенгшщзхъёфывапролджэячсмитьбюЙЦУКЕНГШЩЗХЪЁФЫВАПРОЛДЖЭЯЧСМИТЬБЮQWERTYUIOPASDFGHJKLZXCVBNMqwertyuiopasdfghjklzxcvbnm1234567890 "
|
||
answers_rus = {"да", "дa"}
|
||
answers_eng = {"da", "dа"}
|
||
answers_net_rus = {'нет', 'нeт', 'hет', 'heт'}
|
||
answers_net_eng = {"net", "nеt"}
|
||
answers_where_rus = {"где", "гдe"}
|
||
answers_where_eng = {"gde", "gdе"}
|
||
answers_300 = {"300", "триста"}
|
||
answers_a = {"a", "а"}
|
||
answers_che = {"че", "чё", "чe", "чо", "чo"}
|
||
answers_aga = {"ага"}
|
||
answers_kak = {"как"}
|
||
answers_kto = {"кто"}
|
||
|
||
|
||
cache = TTLCache(settings.CACHE_SIZE, settings.CACHE_TTL)
|
||
|
||
|
||
def get_chat_info(chat_id: int) -> dict:
|
||
cached_info = cache.get(chat_id)
|
||
if cached_info is not None:
|
||
return cached_info
|
||
mongo_info = mongo.chats_collection.find_one({"chat_id": chat_id})
|
||
if mongo_info is not None:
|
||
cache[chat_id] = mongo_info
|
||
return mongo_info
|
||
chat_info = {"chat_id": chat_id, "state": "default", "probability": 100}
|
||
mongo.chats_collection.insert_one(chat_info)
|
||
cache[chat_id] = chat_info
|
||
return chat_info
|
||
|
||
|
||
def set_values(chat_id: int, **values):
|
||
cached_info = cache.get(chat_id)
|
||
if cached_info is None:
|
||
mongo_info = mongo.chats_collection.find_one({"chat_id": chat_id})
|
||
if mongo_info is None:
|
||
chat_info = {"chat_id": chat_id, "state": "default", "probability": 100}
|
||
chat_info.update(values)
|
||
mongo.chats_collection.insert_one(chat_info)
|
||
cache[chat_id] = chat_info
|
||
else:
|
||
mongo.chats_collection.update_one({"chat_id": chat_id}, {"$set": values})
|
||
mongo_info = dict(mongo_info)
|
||
mongo_info.update(values)
|
||
cache[chat_id] = mongo_info
|
||
else:
|
||
cached_info.update(values)
|
||
mongo.chats_collection.update_one({"chat_id": chat_id}, {"$set": values})
|
||
|
||
|
||
@bot.message_handler(commands=['setprobability'])
|
||
def set_probability(message: Message):
|
||
bot.send_message(message.chat.id, "Отправь одно число - вероятность парирования")
|
||
set_values(message.chat.id, state="set_probability")
|
||
|
||
|
||
@bot.message_handler(commands=['rating'])
|
||
def set_probability(message: Message):
|
||
rating = list(mongo.counter_collection.find({"chat_id": message.chat.id}).sort("count", -1))
|
||
if not rating:
|
||
bot.send_message(message.chat.id, "В этом чате я пока никому не парировал")
|
||
return
|
||
text = "Вот кому я парировал:\n"
|
||
for index, value in enumerate(rating):
|
||
text += f"{index + 1}. @{value['username']} - {value['count']}\n"
|
||
bot.send_message(message.chat.id, text)
|
||
|
||
|
||
@bot.message_handler()
|
||
def do_action(message: Message):
|
||
info = get_chat_info(message.chat.id)
|
||
if info['state'] == "set_probability":
|
||
try:
|
||
value = int(message.text)
|
||
if value < 0 or value > 100:
|
||
bot.reply_to(message, "Число не попадает в диапозон от 0 до 100!")
|
||
else:
|
||
set_values(message.chat.id, probability=value, state="default")
|
||
bot.reply_to(message, "Ок! Установил")
|
||
except ValueError:
|
||
bot.reply_to(message, "Это не число!")
|
||
return
|
||
convert_text = ''.join([letter for letter in message.text if letter in all_letters]).lower().split()
|
||
if len(convert_text) > 0:
|
||
convert_text = convert_text[-1]
|
||
else:
|
||
return
|
||
ans = None
|
||
if convert_text in answers_rus:
|
||
ans = "Пизда!"
|
||
if convert_text in answers_eng:
|
||
ans = "Pizda!"
|
||
if convert_text in answers_net_rus:
|
||
ans = "Пидора ответ!"
|
||
if convert_text in answers_net_eng:
|
||
ans = "Pidora otvet!"
|
||
if convert_text in answers_where_rus:
|
||
ans = "В пизде!"
|
||
if convert_text in answers_where_eng:
|
||
ans = "В pizde!"
|
||
if convert_text in answers_300:
|
||
ans = "Отсоси у тракториста!"
|
||
if convert_text in answers_a:
|
||
ans = "Хуй на!"
|
||
if convert_text in answers_che:
|
||
ans = "Хуй через плечо!"
|
||
if convert_text in answers_aga:
|
||
ans = "В жопе нога!"
|
||
if convert_text in answers_kak:
|
||
ans = "Жопой об косяк!"
|
||
if convert_text in answers_kto:
|
||
ans = "Конь в пальто!"
|
||
if ans is not None and randrange(1, 101) <= info["probability"]:
|
||
bot.reply_to(message, ans)
|
||
mongo.inc(message.from_user.username, message.chat.id)
|
||
|
||
|
||
bot.polling()
|