add ban user action (#77)

* add ban user action

* fix tests

* send message through update.effective_message

bot_microservice/core/auth/models/users.py

@@ -28,6 +28,7 @@ class User(Base):
         backref="user",
         lazy="selectin",
         uselist=False,
+        cascade="delete",
     )
 
     @property

bot_microservice/core/auth/services.py

@@ -1,6 +1,13 @@
 import uuid
 from dataclasses import dataclass
+from functools import wraps
+from typing import Any
 
+from loguru import logger
+from telegram import Update
+from telegram.ext import ContextTypes
+
+from constants import BotCommands
 from core.auth.dto import UserIsBannedDTO
 from core.auth.models.users import User
 from core.auth.repository import UserRepository
@@ -54,3 +61,31 @@ class UserService:
 
     async def check_user_is_banned(self, user_id: int) -> UserIsBannedDTO:
         return await self.repository.check_user_is_banned(user_id)
+
+
+def check_user_is_banned(func: Any) -> Any:
+    @wraps(func)
+    async def wrapper(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+        if not update.effective_message:
+            logger.error('no effective message', update=update, context=context)
+            return
+
+        if not update.effective_user:
+            logger.error('no effective user', update=update, context=context)
+            await update.effective_message.reply_text(
+                "Бот не смог определить пользователя. :(\nОб ошибке уже сообщено."
+            )
+            return
+
+        user_service = UserService.build()  # noqa: NEW100
+        user_status = await user_service.check_user_is_banned(update.effective_user.id)
+        if user_status.is_banned:
+            await update.effective_message.reply_text(
+                text=f"You have banned for reason: *{user_status.ban_reason}*."
+                f"\nPlease contact the /{BotCommands.developer}",
+                parse_mode="Markdown",
+            )
+        else:
+            await func(update, context)
+
+    return wrapper

bot_microservice/core/bot/commands.py

@@ -7,6 +7,7 @@ from telegram import InlineKeyboardMarkup, Update
 from telegram.ext import ContextTypes
 
 from constants import BotCommands, BotEntryPoints
+from core.auth.services import check_user_is_banned
 from core.bot.app import get_bot
 from core.bot.keyboards import main_keyboard
 from core.bot.services import ChatGptService, SpeechToTextService
@@ -42,6 +43,7 @@ async def about_bot(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     )
 
 
+@check_user_is_banned
 async def website(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     if not update.effective_message:
         return
@@ -49,6 +51,7 @@ async def website(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     await update.effective_message.reply_text(f"Веб версия: {website}")
 
 
+@check_user_is_banned
 async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     """Send a message when the command /help is issued."""
 
@@ -63,6 +66,7 @@ async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No
     )
 
 
+@check_user_is_banned
 async def bug_report(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     """Send a message when the command /bug-report is issued."""
 
@@ -88,15 +92,11 @@ async def github(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     )
 
 
+@check_user_is_banned
 async def ask_question(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     if not update.message:
         return
 
-    if not update.effective_user:
-        logger.error('no effective user', update=update, context=context)
-        await update.message.reply_text("Бот не смог определить пользователя. :(\nОб ошибке уже сообщено.")
-        return
-
     await update.message.reply_text(
         f"Ответ в среднем занимает 10-15 секунд.\n"
         f"- Список команд: /{BotCommands.help}\n"
@@ -108,10 +108,10 @@ async def ask_question(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No
     answer, user = await asyncio.gather(
         chatgpt_service.request_to_chatgpt(question=update.message.text),
         chatgpt_service.get_or_create_bot_user(
-            user_id=update.effective_user.id,
-            username=update.effective_user.username,
-            first_name=update.effective_user.first_name,
-            last_name=update.effective_user.last_name,
+            user_id=update.effective_user.id,  # type: ignore[union-attr]
+            username=update.effective_user.username,  # type: ignore[union-attr]
+            first_name=update.effective_user.first_name,  # type: ignore[union-attr]
+            last_name=update.effective_user.last_name,  # type: ignore[union-attr]
         ),
     )
     await asyncio.gather(update.message.reply_text(answer), chatgpt_service.update_bot_user_message_count(user.id))

bot_microservice/core/bot/handlers.py

@@ -36,6 +36,7 @@ bot_event_handlers = BotEventHandlers()
 bot_event_handlers.add_handler(CommandHandler(BotCommands.help, help_command))
 bot_event_handlers.add_handler(CommandHandler(BotCommands.website, website))
 bot_event_handlers.add_handler(CommandHandler(BotCommands.bug_report, bug_report))
+bot_event_handlers.add_handler(CommandHandler(BotCommands.developer, about_me))
 
 bot_event_handlers.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, ask_question))
 bot_event_handlers.add_handler(MessageHandler(filters.VOICE | filters.AUDIO, voice_recognize))