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

scripts/add_ban_author_column_to_blacklist.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+"""
+Скрипт миграции для добавления колонки ban_author в таблицу blacklist.
+Колонка хранит user_id администратора, инициировавшего бан.
+"""
+import argparse
+import asyncio
+import os
+import sys
+from pathlib import Path
+
+import aiosqlite
+
+project_root = Path(__file__).resolve().parent.parent
+sys.path.insert(0, str(project_root))
+
+from logs.custom_logger import logger  # noqa: E402
+
+DEFAULT_DB_PATH = "database/tg-bot-database.db"
+
+
+def _column_exists(rows: list, name: str) -> bool:
+    """PRAGMA table_info returns (cid, name, type, notnull, dflt_value, pk)."""
+    for row in rows:
+        if row[1] == name:
+            return True
+    return False
+
+
+async def main(db_path: str) -> None:
+    db_path = os.path.abspath(db_path)
+    if not os.path.exists(db_path):
+        logger.error("База данных не найдена: %s", db_path)
+        print(f"Ошибка: база данных не найдена: {db_path}")
+        return
+
+    async with aiosqlite.connect(db_path) as conn:
+        await conn.execute("PRAGMA foreign_keys = ON")
+
+        # Проверяем наличие колонки ban_author
+        cursor = await conn.execute("PRAGMA table_info(blacklist)")
+        rows = await cursor.fetchall()
+        await cursor.close()
+
+        if not _column_exists(rows, "ban_author"):
+            logger.info("Добавление колонки ban_author в blacklist")
+            await conn.execute(
+                "ALTER TABLE blacklist "
+                "ADD COLUMN ban_author INTEGER REFERENCES our_users (user_id) ON DELETE SET NULL"
+            )
+            await conn.commit()
+            print("Колонка ban_author добавлена в таблицу blacklist.")
+        else:
+            print("Колонка ban_author уже существует в таблице blacklist.")
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(
+        description="Добавление колонки ban_author в blacklist"
+    )
+    parser.add_argument(
+        "--db",
+        default=os.environ.get("DB_PATH", DEFAULT_DB_PATH),
+        help="Путь к БД (или DB_PATH)",
+    )
+    args = parser.parse_args()
+    asyncio.run(main(args.db))
+