Добавлено поле ban_author в модель BlacklistUser и соответствующие изменения в базе данных для отслеживания автора блокировки пользователя. Обновлены методы работы с черным списком в AsyncBotDB и BlacklistRepository, а также обработка блокировок в AdminService и BanService. Обновлены тесты для проверки новых функциональностей.

helper_bot/handlers/admin/admin_handlers.py

@@ -359,7 +359,8 @@ async def confirm_ban(
             user_id=user_data['target_user_id'],
             username=user_data['target_username'],
             reason=user_data['ban_reason'],
-            ban_days=user_data['ban_days']
+            ban_days=user_data['ban_days'],
+            ban_author_id=message.from_user.id,
         )
         
         safe_username = escape_html(user_data['target_username'])

helper_bot/handlers/admin/services.py

@@ -117,7 +117,7 @@ class AdminService:
     
     @track_time("ban_user", "admin_service")
     @track_errors("admin_service", "ban_user")
-    async def ban_user(self, user_id: int, username: str, reason: str, ban_days: Optional[int]) -> None:
+    async def ban_user(self, user_id: int, username: str, reason: str, ban_days: Optional[int], ban_author_id: int) -> None:
         """Заблокировать пользователя"""
         try:
             # Проверяем, не заблокирован ли уже пользователь
@@ -130,7 +130,7 @@ class AdminService:
                 date_to_unban = add_days_to_date(ban_days)
             
             # Сохраняем в БД (username больше не передается, так как не используется в новой схеме)
-            await self.bot_db.set_user_blacklist(user_id, None, reason, date_to_unban)
+            await self.bot_db.set_user_blacklist(user_id, None, reason, date_to_unban, ban_author=ban_author_id)
             
             logger.info(f"Пользователь {user_id} ({username}) заблокирован. Причина: {reason}, срок: {ban_days} дней")
             

helper_bot/handlers/callback/services.py

@@ -439,11 +439,14 @@ class BanService:
         current_date = datetime.now()
         date_to_unban = int((current_date + timedelta(days=7)).timestamp())
         
+        ban_author_id = call.from_user.id
+
         await self.db.set_user_blacklist(
             user_id=author_id,
             user_name=None,
             message_for_user="Спам",
-            date_to_unban=date_to_unban
+            date_to_unban=date_to_unban,
+            ban_author=ban_author_id,
         )
         
         await self._get_bot(call.message).delete_message(chat_id=self.group_for_posts, message_id=call.message.message_id)