Modernize bot for Python 3.12

scripts/db_setup.py

@@ -3,12 +3,10 @@
 Скрипт для очистки и инициализации БД с тестовыми данными
 """
 import asyncio
-import random
-from database import async_session_maker, Base, engine
-from models import User, Lottery, Participation, Winner
-from services import LotteryService, UserService, ParticipationService
-from datetime import datetime, timedelta
-from sqlalchemy import delete, update, select, func
+from src.core.database import async_session_maker
+from src.core.models import User, Lottery, Participation, Winner, Account
+from src.core.services import LotteryService, UserService, ParticipationService
+from sqlalchemy import delete, select, func
 
 async def clear_database():
     """Очистка всех таблиц"""
@@ -18,6 +16,7 @@ async def clear_database():
         # Удаляем все данные в правильном порядке (учитывая внешние ключи)
         await session.execute(delete(Winner))
         await session.execute(delete(Participation))
+        await session.execute(delete(Account))
         await session.execute(delete(Lottery))
         await session.execute(delete(User))
         await session.commit()
@@ -26,8 +25,10 @@ async def clear_database():
 
 def generate_account_numbers(user_id: int, count: int = 10) -> list:
     """Генерация номеров счетов для пользователя"""
-    base = user_id * 1000
-    return [f"{base + i:06d}" for i in range(1, count + 1)]
+    return [
+        "-".join(f"{part:02d}" for part in [user_id, i, 11, 22, 33, 44, 55])
+        for i in range(1, count + 1)
+    ]
 
 async def create_test_users():
     """Создание тестовых пользователей"""
@@ -62,12 +63,8 @@ async def create_test_users():
             # Генерируем и присваиваем номера счетов
             account_numbers = generate_account_numbers(i, 10)
             
-            # Обновляем счета
-            await session.execute(
-                update(User)
-                .where(User.id == user.id)
-                .values(account_number=",".join(account_numbers))
-            )
+            for account_number in account_numbers:
+                await UserService.set_account_number(session, user.telegram_id, account_number)
             
             await session.commit()
             created_users.append((user, account_numbers))
@@ -194,4 +191,4 @@ async def main():
     print("\n🎉 Инициализация завершена!")
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    asyncio.run(main())