refactoring

app/stan.py

@@ -1,8 +1,43 @@
 """ Stan quote generator. """
 import random
+from .config import bot, types
+from .filters import is_white, is_nongrata
 
 
 def speak(chance_of):
     number = random.randint(0, chance_of)
     if number == 0:
         return random.choice([i.rstrip() for i in open('Stan.txt', 'r', encoding='utf8') if i])
+
+
+@bot.message_handler(func=is_white, commands=['add'])
+def add_stan_quote(message):
+    if message.reply_to_message and message.reply_to_message.text:
+        with open('Stan.txt', 'a', encoding='utf8') as stan_quotes:
+            if message.reply_to_message.text not in (i.rstrip() for i in open('Stan.txt', 'r', encoding='utf8')):
+                stan_quotes.write(message.reply_to_message.text.replace('\n', ' ') + '\n')
+                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):
+    if message.reply_to_message and message.reply_to_message.text:
+        if message.reply_to_message.text in (i.rstrip() for i in open('Stan.txt', 'r', encoding='utf8')):
+            quotes = list(open('Stan.txt', 'r', encoding='utf8'))
+            with open('Stan.txt', 'w', encoding='utf8') as stan_quotes:
+                quotes.remove(message.reply_to_message.text + '\n')
+                stan_quotes.writelines(quotes)
+            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'у нас тут таких не любят')