fix quality code
This commit is contained in:
@@ -3,6 +3,7 @@
|
||||
Скрипт миграции для переноса записей из blacklist в blacklist_history.
|
||||
Переносит все существующие записи из таблицы blacklist в таблицу blacklist_history.
|
||||
"""
|
||||
|
||||
import argparse
|
||||
import asyncio
|
||||
import os
|
||||
@@ -14,6 +15,7 @@ project_root = Path(__file__).resolve().parent.parent
|
||||
sys.path.insert(0, str(project_root))
|
||||
|
||||
import aiosqlite
|
||||
|
||||
from logs.custom_logger import logger
|
||||
|
||||
DEFAULT_DB_PATH = "database/tg-bot-database.db"
|
||||
@@ -37,8 +39,12 @@ async def main(db_path: str) -> None:
|
||||
await cursor.close()
|
||||
|
||||
if not rows:
|
||||
logger.error("Таблица blacklist_history не найдена. Сначала запустите create_blacklist_history_table.py")
|
||||
print("Ошибка: таблица blacklist_history не найдена. Сначала запустите create_blacklist_history_table.py")
|
||||
logger.error(
|
||||
"Таблица blacklist_history не найдена. Сначала запустите create_blacklist_history_table.py"
|
||||
)
|
||||
print(
|
||||
"Ошибка: таблица blacklist_history не найдена. Сначала запустите create_blacklist_history_table.py"
|
||||
)
|
||||
return
|
||||
|
||||
# Получаем все записи из blacklist
|
||||
@@ -53,7 +59,9 @@ async def main(db_path: str) -> None:
|
||||
logger.info("В таблице blacklist нет записей для переноса")
|
||||
return
|
||||
|
||||
logger.info("Найдено записей в blacklist для переноса: %d", len(blacklist_records))
|
||||
logger.info(
|
||||
"Найдено записей в blacklist для переноса: %d", len(blacklist_records)
|
||||
)
|
||||
print(f"Найдено записей в blacklist для переноса: {len(blacklist_records)}")
|
||||
|
||||
# Получаем текущее время в Unix timestamp
|
||||
@@ -69,16 +77,20 @@ async def main(db_path: str) -> None:
|
||||
# Проверяем, нет ли уже записи для этого user_id с таким же date_ban
|
||||
# (чтобы избежать дубликатов при повторном запуске)
|
||||
date_ban = created_at if created_at is not None else current_time
|
||||
|
||||
|
||||
check_cursor = await conn.execute(
|
||||
"SELECT id FROM blacklist_history WHERE user_id = ? AND date_ban = ?",
|
||||
(user_id, date_ban)
|
||||
(user_id, date_ban),
|
||||
)
|
||||
existing = await check_cursor.fetchone()
|
||||
await check_cursor.close()
|
||||
|
||||
if existing:
|
||||
logger.debug("Запись для user_id=%d с date_ban=%d уже существует, пропускаем", user_id, date_ban)
|
||||
logger.debug(
|
||||
"Запись для user_id=%d с date_ban=%d уже существует, пропускаем",
|
||||
user_id,
|
||||
date_ban,
|
||||
)
|
||||
skipped_count += 1
|
||||
continue
|
||||
|
||||
@@ -96,8 +108,8 @@ async def main(db_path: str) -> None:
|
||||
date_to_unban,
|
||||
ban_author,
|
||||
created_at if created_at is not None else current_time,
|
||||
current_time
|
||||
)
|
||||
current_time,
|
||||
),
|
||||
)
|
||||
migrated_count += 1
|
||||
|
||||
@@ -106,9 +118,11 @@ async def main(db_path: str) -> None:
|
||||
logger.info(
|
||||
"Миграция завершена. Перенесено записей: %d, пропущено (дубликаты): %d",
|
||||
migrated_count,
|
||||
skipped_count
|
||||
skipped_count,
|
||||
)
|
||||
print(
|
||||
f"Миграция завершена. Перенесено записей: {migrated_count}, пропущено (дубликаты): {skipped_count}"
|
||||
)
|
||||
print(f"Миграция завершена. Перенесено записей: {migrated_count}, пропущено (дубликаты): {skipped_count}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user