style: isort + black

scripts/drop_vector_hash_column.py

@@ -8,6 +8,7 @@
 SQLite не поддерживает DROP COLUMN напрямую (до версии 3.35.0),
 поэтому используем пересоздание таблицы.
 """
+
 import argparse
 import asyncio
 import os
@@ -25,7 +26,10 @@ try:
     from logs.custom_logger import logger
 except ImportError:
     import logging
-    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+    logging.basicConfig(
+        level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+    )
     logger = logging.getLogger(__name__)
 
 DEFAULT_DB_PATH = "database/tg-bot-database.db"
@@ -42,7 +46,7 @@ async def get_sqlite_version(conn: aiosqlite.Connection) -> tuple:
     """Возвращает версию SQLite."""
     cursor = await conn.execute("SELECT sqlite_version()")
     version_str = (await cursor.fetchone())[0]
-    return tuple(map(int, version_str.split('.')))
+    return tuple(map(int, version_str.split(".")))
 
 
 async def main(db_path: str) -> None:
@@ -50,21 +54,21 @@ async def main(db_path: str) -> None:
     Удаляет колонку vector_hash из таблицы post_from_telegram_suggest.
     """
     db_path = os.path.abspath(db_path)
-    
+
     if not os.path.exists(db_path):
         logger.error(f"База данных не найдена: {db_path}")
         return
-    
+
     async with aiosqlite.connect(db_path) as conn:
         # Проверяем существует ли колонка
         if not await column_exists(conn, "post_from_telegram_suggest", "vector_hash"):
             logger.info("Колонка vector_hash не существует, миграция не требуется")
             return
-        
+
         # Проверяем версию SQLite
         version = await get_sqlite_version(conn)
         logger.info(f"Версия SQLite: {'.'.join(map(str, version))}")
-        
+
         # SQLite 3.35.0+ поддерживает DROP COLUMN
         if version >= (3, 35, 0):
             logger.info("Используем ALTER TABLE DROP COLUMN")
@@ -74,15 +78,15 @@ async def main(db_path: str) -> None:
         else:
             # Для старых версий пересоздаём таблицу
             logger.info("Используем пересоздание таблицы (SQLite < 3.35.0)")
-            
+
             # Получаем список колонок без vector_hash
             cursor = await conn.execute("PRAGMA table_info(post_from_telegram_suggest)")
             columns = await cursor.fetchall()
             column_names = [col[1] for col in columns if col[1] != "vector_hash"]
             columns_str = ", ".join(column_names)
-            
+
             logger.info(f"Колонки для сохранения: {columns_str}")
-            
+
             # Пересоздаём таблицу
             await conn.execute("BEGIN TRANSACTION")
             try:
@@ -91,21 +95,21 @@ async def main(db_path: str) -> None:
                     f"CREATE TABLE post_from_telegram_suggest_backup AS "
                     f"SELECT {columns_str} FROM post_from_telegram_suggest"
                 )
-                
+
                 # Удаляем старую таблицу
                 await conn.execute("DROP TABLE post_from_telegram_suggest")
-                
+
                 # Переименовываем временную
                 await conn.execute(
                     "ALTER TABLE post_from_telegram_suggest_backup "
                     "RENAME TO post_from_telegram_suggest"
                 )
-                
+
                 await conn.execute("COMMIT")
             except Exception as e:
                 await conn.execute("ROLLBACK")
                 raise e
-        
+
         await conn.commit()
         logger.info("Колонка vector_hash успешно удалена")