kotolex
d9ea26381c
- добавлено отслеживание входа и выхода юзеров - удаляем оба сообщения - логируем при входе основную инфу юзера включая его био, на этой основе можно тоже добавить фильтр - добавлены тесты на новую функцию
85 lines
3.6 KiB
Python
85 lines
3.6 KiB
Python
""" Stan's commands and reactions. """
|
||
import asyncio
|
||
import html
|
||
import logging
|
||
import random
|
||
|
||
from telebot import types, util
|
||
|
||
from .filters import is_white_id
|
||
from .helpers import is_nongrata, short_user_data
|
||
from .models import all_chat_quotes, add_quote, is_quote_in_chat, delete_quote_in_chat
|
||
from .robot import bot
|
||
|
||
TYPING_TIMEOUT = 0.13 / 4 # Reading time is quarter of the same text writing time
|
||
|
||
|
||
def speak(chance_of: int, chat_id: int) -> None | str:
|
||
number = random.randint(0, chance_of)
|
||
if number == 0:
|
||
return random.choice([i[0] for i in all_chat_quotes(chat_id)])
|
||
return None
|
||
|
||
|
||
async def send_quote(after_sec, message, quote):
|
||
"""Pretend Reading, pretend Typing, send."""
|
||
if message.text:
|
||
await asyncio.sleep(len(message.text) * TYPING_TIMEOUT)
|
||
await bot.send_chat_action(message.chat.id, action="typing")
|
||
await asyncio.sleep(after_sec) # Typing time
|
||
await bot.send_message(message.chat.id, html.escape(quote))
|
||
|
||
|
||
async def act(message: types.Message):
|
||
quote = speak(50, message.chat.id)
|
||
if quote:
|
||
await send_quote(len(quote) * 0.13, message, quote)
|
||
|
||
|
||
@bot.message_handler(func=is_white_id, commands=["add"])
|
||
async def add_stan_quote(message: types.Message):
|
||
if message.reply_to_message and message.reply_to_message.text:
|
||
quote = message.reply_to_message.text
|
||
if quote not in {i[0] for i in all_chat_quotes(message.chat.id)}:
|
||
add_quote(message.chat.id, quote.replace("\n", " "))
|
||
await bot.send_message(message.chat.id, "➕\n └ " + quote.replace("\n", " "), parse_mode='Markdown')
|
||
await bot.delete_message(message.chat.id, message.id)
|
||
else:
|
||
await bot.send_message(message.chat.id, f"⛔️ Не добавил, есть токое\n └ {quote}", parse_mode='Markdown')
|
||
|
||
|
||
@bot.message_handler(func=is_white_id, commands=["remove"])
|
||
async def remove_stan_quote(message: types.Message):
|
||
if message.reply_to_message and message.reply_to_message.text:
|
||
quote = message.reply_to_message.text
|
||
if is_quote_in_chat(quote, message.chat.id):
|
||
delete_quote_in_chat(quote, message.chat.id)
|
||
await bot.send_message(message.chat.id, f"➖ \n └ {quote}", parse_mode='Markdown')
|
||
await bot.delete_message(message.chat.id, message.id)
|
||
else:
|
||
await bot.send_message(message.chat.id, f"⛔️ Нет такого\n └ {quote}", parse_mode='Markdown')
|
||
|
||
|
||
@bot.message_handler(func=is_nongrata)
|
||
async def tease_nongrata(message: types.Message):
|
||
"""Reply to non grata mentions."""
|
||
await bot.reply_to(message, "у нас тут таких не любят")
|
||
|
||
|
||
@bot.message_handler(content_types=util.content_type_service)
|
||
async def check_new_members(message: types.Message):
|
||
if message.content_type == "new_chat_members":
|
||
from_user = message.from_user
|
||
info = short_user_data(from_user)
|
||
await bot.delete_message(message.chat.id, message.id)
|
||
# мы не знаем как быстро получим инфу о био, потому сначала удаляем сообщение, потом запрашиваем био/логируем
|
||
more_data = await bot.get_chat(from_user.id)
|
||
info['bio'] = more_data.bio
|
||
info['photo'] = more_data.photo.big_file_id
|
||
logging.info("[JOIN] User join chat: %s", info)
|
||
if message.content_type == "left_chat_member":
|
||
from_user = message.from_user
|
||
info = short_user_data(from_user)
|
||
logging.debug("[LEAVE] User left chat: %s", info)
|
||
await bot.delete_message(message.chat.id, message.id)
|