81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
""" Stan's commands and reactions. """
|
|
import html
|
|
import random
|
|
import threading
|
|
from time import sleep
|
|
from .config import bot, types
|
|
from .filters import is_white, is_nongrata
|
|
from .database import session
|
|
from .models import Quote
|
|
|
|
|
|
def speak(chance_of, group_id):
|
|
number = random.randint(0, chance_of)
|
|
if number == 0:
|
|
return random.choice([i[0] for i in session.query(Quote.text).filter(Quote.chat_id == group_id).all()])
|
|
|
|
|
|
def send_quote(after_sec, message, quote):
|
|
"""Pretend Reading, pretend Typing, send."""
|
|
if message.text:
|
|
sleep(
|
|
len(message.text) * 0.13 / 4
|
|
) # Reading time is quarter of the same text writing time
|
|
bot.send_chat_action(message.chat.id, action="typing")
|
|
sleep(after_sec) # Typing time
|
|
bot.send_message(message.chat.id, html.escape(quote))
|
|
|
|
|
|
def act(message: types.Message):
|
|
quote = speak(50, message.chat.id)
|
|
if quote:
|
|
threading.Thread(
|
|
target=send_quote, args=(len(quote) * 0.13, message, quote)
|
|
).start()
|
|
|
|
|
|
@bot.message_handler(func=is_white, commands=["add"])
|
|
def add_stan_quote(message: types.Message):
|
|
if message.reply_to_message and message.reply_to_message.text:
|
|
if message.reply_to_message.text not in [i[0] for i in session.query(Quote.text).filter(
|
|
Quote.chat_id == message.chat.id).all()]:
|
|
session.add(Quote(chat_id=message.chat.id, text=message.reply_to_message.text.replace("\n", " ")))
|
|
session.commit()
|
|
bot.send_message(
|
|
message.chat.id,
|
|
"✅ <b>Добавил</b>\n └ <i>"
|
|
+ message.reply_to_message.text.replace("\n", " ")
|
|
+ "</i>",
|
|
)
|
|
else:
|
|
bot.send_message(
|
|
message.chat.id,
|
|
f"⛔️ <b>Не добавил</b>, есть токое\n └ <i>{message.reply_to_message.text}</i>",
|
|
)
|
|
|
|
|
|
@bot.message_handler(func=is_white, commands=["remove"])
|
|
def remove_stan_quote(message: types.Message):
|
|
if message.reply_to_message and message.reply_to_message.text:
|
|
quote = session.query(Quote).filter_by(text=message.reply_to_message.text,
|
|
chat_id=message.chat.id).first()
|
|
if quote:
|
|
session.delete(quote)
|
|
session.commit()
|
|
|
|
bot.send_message(
|
|
message.chat.id,
|
|
f"✅ <b>Удалил</b>\n └ <i>{message.reply_to_message.text}</i>",
|
|
)
|
|
else:
|
|
bot.send_message(
|
|
message.chat.id,
|
|
f"⛔️ <b>Нет такого</b>\n └ <i>{message.reply_to_message.text}</i>",
|
|
)
|
|
|
|
|
|
@bot.message_handler(func=is_nongrata)
|
|
def tease_nongrata(message: types.Message):
|
|
"""Reply to non grata mentions."""
|
|
bot.reply_to(message, f"у нас тут таких не любят")
|