Refactoring: step 1
- пакет app переименован в src - файл БД теперь должен лежать в data - удалена папка коммандс, все команды вынесены в модуль коммандс или админ_коммандс - нет импортов в ините - убраны все импорты * - теперь в .env можно указать use_reminder="FALSE", тогда не будет стартовать ремайндер - можно указать урл для Правил - в конфиге теперь только конфиг -то, что из енва получаем - константы теперь лежат отдельно - добавлены явные return там, где не было - бот и вебхук разнесены отдельно - при старте проверяем, что Правила доступны - добавлено еще логирование
This commit is contained in:
+68
@@ -0,0 +1,68 @@
|
||||
""" Stan's commands and reactions. """
|
||||
import asyncio
|
||||
import html
|
||||
import random
|
||||
|
||||
from sqlalchemy import delete
|
||||
from telebot import types
|
||||
|
||||
from .filters import is_white, is_nongrata
|
||||
from .models import Quote
|
||||
from .models import session
|
||||
from .robot import bot
|
||||
|
||||
TYPING_TIMEOUT = 0.13 / 4 # Reading time is quarter of the same text writing time
|
||||
|
||||
|
||||
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()])
|
||||
|
||||
|
||||
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, 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 session.query(Quote.text).filter(Quote.chat_id == message.chat.id).all()]:
|
||||
session.add(Quote(chat_id=message.chat.id, text=quote.replace("\n", " ")))
|
||||
session.commit()
|
||||
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, 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
|
||||
already_exist = session.query(Quote.text).filter_by(text=quote, chat_id=message.chat.id).first()
|
||||
if already_exist:
|
||||
session.execute(delete(Quote).filter_by(text=quote, chat_id=message.chat.id))
|
||||
session.commit()
|
||||
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, "у нас тут таких не любят")
|
||||
Reference in New Issue
Block a user