init commit

conduct_draw.py

@@ -0,0 +1,102 @@
+#!/usr/bin/env python3
+"""
+Скрипт для проведения розыгрыша
+"""
+import asyncio
+import json
+from database import async_session_maker
+from services import LotteryService, ParticipationService
+from models import Lottery, User
+
+async def conduct_lottery_draw(lottery_id: int):
+    """Проводим розыгрыш для указанного ID"""
+    print(f"🎲 Проведение розыгрыша #{lottery_id}")
+    
+    async with async_session_maker() as session:
+        # Получаем розыгрыш
+        lottery = await LotteryService.get_lottery(session, lottery_id)
+        if not lottery:
+            print(f"❌ Розыгрыш {lottery_id} не найден")
+            return
+            
+        print(f"📋 Розыгрыш: {lottery.title}")
+        
+        # Призы уже загружены как список
+        prizes = lottery.prizes if isinstance(lottery.prizes, list) else json.loads(lottery.prizes)
+        print(f"🏆 Призы: {len(prizes)} шт.")
+        
+        # Получаем участников
+        participants = await ParticipationService.get_participants(session, lottery_id)
+        print(f"👥 Участников: {len(participants)}")
+        
+        if len(participants) == 0:
+            print("⚠️ Нет участников для розыгрыша")
+            return
+            
+        # Выводим список участников
+        print("\n👥 Список участников:")
+        for i, user in enumerate(participants, 1):
+            accounts = user.account_number.split(',') if user.account_number else ['Нет счетов']
+            print(f"  {i}. {user.first_name} (@{user.username}) - {len(accounts)} счет(ов)")
+        
+        # Проводим розыгрыш
+        print(f"\n🎲 Проводим розыгрыш...")
+        
+        # Используем метод из LotteryService
+        try:
+            winners = await LotteryService.conduct_draw(session, lottery_id)
+            
+            if winners:
+                print(f"\n🎉 Победители определены:")
+                for i, winner_data in enumerate(winners, 1):
+                    user = winner_data['user']
+                    prize = winner_data['prize']
+                    print(f"  🏆 {i} место: {user.first_name} (@{user.username})")
+                    print(f"     💎 Приз: {prize}")
+                    
+                # Обновляем статус розыгрыша
+                await LotteryService.set_lottery_completed(session, lottery_id, True)
+                print(f"\n✅ Розыгрыш завершен и помечен как завершенный")
+                
+            else:
+                print("❌ Ошибка при проведении розыгрыша")
+                
+        except Exception as e:
+            print(f"💥 Ошибка: {e}")
+
+async def main():
+    """Главная функция"""
+    print("🎯 Скрипт проведения розыгрышей")
+    print("=" * 50)
+    
+    # Показываем доступные розыгрыши
+    async with async_session_maker() as session:
+        lotteries = await LotteryService.get_active_lotteries(session)
+        
+        if not lotteries:
+            print("❌ Нет активных розыгрышей")
+            return
+            
+        print("🎲 Активные розыгрыши:")
+        for lottery in lotteries:
+            participants = await ParticipationService.get_participants(session, lottery.id)
+            print(f"  {lottery.id}. {lottery.title} ({len(participants)} участников)")
+    
+    # Просим выбрать розыгрыш
+    print("\nВведите ID розыгрыша для проведения (или 'all' для всех):")
+    choice = input("> ").strip().lower()
+    
+    if choice == 'all':
+        # Проводим все розыгрыши
+        for lottery in lotteries:
+            await conduct_lottery_draw(lottery.id)
+            print("-" * 30)
+    else:
+        try:
+            lottery_id = int(choice)
+            await conduct_lottery_draw(lottery_id)
+        except ValueError:
+            print("❌ Неверный ID розыгрыша")
+
+if __name__ == "__main__":
+    asyncio.run(main())