fix quality code

This commit is contained in:
2026-02-01 23:03:23 +03:00
parent 731e68a597
commit f8962225ee
106 changed files with 8456 additions and 5810 deletions

View File

@@ -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__":