feat: Система автоматического подтверждения выигрышей с поддержкой множественных счетов
Some checks reported errors
continuous-integration/drone/push Build encountered an error
Some checks reported errors
continuous-integration/drone/push Build encountered an error
Основные изменения: ✨ Новые функции: - Система регистрации пользователей с множественными счетами - Автоматическое подтверждение выигрышей через inline-кнопки - Механизм переигровки для неподтвержденных выигрышей (24 часа) - Подтверждение на уровне счетов (каждый счет подтверждается отдельно) - Скрипт полной очистки базы данных 🔧 Технические улучшения: - Исправлена ошибка MissingGreenlet при lazy loading (добавлен joinedload/selectinload) - Добавлено поле claimed_at для отслеживания времени подтверждения - Пакетное добавление счетов с выбором розыгрыша - Проверка владения конкретным счетом при подтверждении 📚 Документация: - docs/AUTO_CONFIRM_SYSTEM.md - Полная документация системы подтверждения - docs/ACCOUNT_BASED_CONFIRMATION.md - Подтверждение на уровне счетов - docs/REGISTRATION_SYSTEM.md - Система регистрации - docs/ADMIN_COMMANDS.md - Команды администратора - docs/CLEAR_DATABASE.md - Очистка БД - docs/QUICK_GUIDE.md - Быстрое начало - docs/UPDATE_LOG.md - Журнал обновлений 🗄️ База данных: - Миграция 003: Таблицы accounts, winner_verifications - Миграция 004: Поле claimed_at в таблице winners - Скрипт scripts/clear_database.py для полной очистки 🎮 Новые команды: Админские: - /check_unclaimed <lottery_id> - Проверка неподтвержденных выигрышей - /redraw <lottery_id> - Повторный розыгрыш - /add_accounts - Пакетное добавление счетов - /list_accounts <telegram_id> - Список счетов пользователя Пользовательские: - /register - Регистрация с вводом данных - /my_account - Просмотр своих счетов - Callback confirm_win_{id} - Подтверждение выигрыша 🛠️ Makefile: - make clear-db - Очистка всех данных из БД (с подтверждением) 🔒 Безопасность: - Проверка владения счетом при подтверждении - Защита от подтверждения чужих счетов - Независимое подтверждение каждого выигрышного счета 📊 Логика работы: 1. Пользователь регистрируется и добавляет счета 2. Счета участвуют в розыгрыше 3. Победители получают уведомление с кнопкой подтверждения 4. Каждый счет подтверждается отдельно (24 часа на подтверждение) 5. Неподтвержденные выигрыши переигрываются через /redraw
This commit is contained in:
314
src/handlers/redraw_handlers.py
Normal file
314
src/handlers/redraw_handlers.py
Normal file
@@ -0,0 +1,314 @@
|
||||
"""Команды для повторного розыгрыша неподтвержденных выигрышей"""
|
||||
from aiogram import Router, F
|
||||
from aiogram.types import Message, InlineKeyboardButton, InlineKeyboardMarkup
|
||||
from aiogram.filters import Command
|
||||
from sqlalchemy import select, and_
|
||||
from datetime import datetime, timezone, timedelta
|
||||
import random
|
||||
|
||||
from src.core.database import async_session_maker
|
||||
from src.core.registration_services import AccountService, WinnerNotificationService
|
||||
from src.core.services import LotteryService
|
||||
from src.core.models import User, Winner
|
||||
from src.core.config import ADMIN_IDS
|
||||
|
||||
|
||||
router = Router()
|
||||
|
||||
|
||||
def is_admin(user_id: int) -> bool:
|
||||
"""Проверка прав администратора"""
|
||||
return user_id in ADMIN_IDS
|
||||
|
||||
|
||||
@router.message(Command("check_unclaimed"))
|
||||
async def check_unclaimed_winners(message: Message):
|
||||
"""
|
||||
Проверить неподтвержденные выигрыши (более 24 часов)
|
||||
Формат: /check_unclaimed <lottery_id>
|
||||
"""
|
||||
if not is_admin(message.from_user.id):
|
||||
await message.answer("❌ Недостаточно прав")
|
||||
return
|
||||
|
||||
parts = message.text.split()
|
||||
if len(parts) != 2:
|
||||
await message.answer(
|
||||
"❌ Неверный формат команды\n\n"
|
||||
"Используйте: /check_unclaimed <lottery_id>"
|
||||
)
|
||||
return
|
||||
|
||||
try:
|
||||
lottery_id = int(parts[1])
|
||||
except ValueError:
|
||||
await message.answer("❌ lottery_id должен быть числом")
|
||||
return
|
||||
|
||||
try:
|
||||
async with async_session_maker() as session:
|
||||
from sqlalchemy.orm import selectinload
|
||||
from src.core.models import Lottery
|
||||
|
||||
# Загружаем розыгрыш с участниками
|
||||
lottery_result = await session.execute(
|
||||
select(Lottery)
|
||||
.options(selectinload(Lottery.participations))
|
||||
.where(Lottery.id == lottery_id)
|
||||
)
|
||||
lottery = lottery_result.scalar_one_or_none()
|
||||
|
||||
if not lottery:
|
||||
await message.answer(f"❌ Розыгрыш #{lottery_id} не найден")
|
||||
return
|
||||
|
||||
winners = await LotteryService.get_winners(session, lottery_id)
|
||||
|
||||
if not winners:
|
||||
await message.answer(f"В розыгрыше '{lottery.title}' нет победителей")
|
||||
return
|
||||
|
||||
# Находим неподтвержденные выигрыши старше 24 часов
|
||||
now = datetime.now(timezone.utc)
|
||||
unclaimed = []
|
||||
|
||||
for winner in winners:
|
||||
if not winner.is_claimed and winner.is_notified:
|
||||
# Проверяем, прошло ли 24 часа
|
||||
time_passed = now - winner.created_at
|
||||
if time_passed.total_seconds() > 24 * 3600: # 24 часа
|
||||
unclaimed.append({
|
||||
'winner': winner,
|
||||
'hours_passed': int(time_passed.total_seconds() / 3600)
|
||||
})
|
||||
|
||||
if not unclaimed:
|
||||
await message.answer(
|
||||
f"✅ Все победители розыгрыша '{lottery.title}' подтвердили выигрыш\n"
|
||||
f"или срок подтверждения еще не истек."
|
||||
)
|
||||
return
|
||||
|
||||
text = f"⚠️ **Неподтвержденные выигрыши в розыгрыше '{lottery.title}':**\n\n"
|
||||
|
||||
for item in unclaimed:
|
||||
winner = item['winner']
|
||||
hours = item['hours_passed']
|
||||
|
||||
text += f"🏆 {winner.place} место - {winner.prize}\n"
|
||||
|
||||
# Получаем информацию о победителе
|
||||
async with async_session_maker() as session:
|
||||
if winner.user_id:
|
||||
user_result = await session.execute(
|
||||
select(User).where(User.id == winner.user_id)
|
||||
)
|
||||
user = user_result.scalar_one_or_none()
|
||||
if user:
|
||||
text += f" 👤 {user.first_name}"
|
||||
if user.club_card_number:
|
||||
text += f" (КК: {user.club_card_number})"
|
||||
text += "\n"
|
||||
|
||||
if winner.account_number:
|
||||
text += f" 💳 {winner.account_number}\n"
|
||||
|
||||
text += f" ⏰ Прошло: {hours} часов\n\n"
|
||||
|
||||
text += f"\n📊 Всего неподтвержденных: {len(unclaimed)}\n\n"
|
||||
text += f"Используйте /redraw {lottery_id} для повторного розыгрыша"
|
||||
|
||||
await message.answer(text, parse_mode="Markdown")
|
||||
|
||||
except Exception as e:
|
||||
await message.answer(f"❌ Ошибка: {str(e)}")
|
||||
|
||||
|
||||
@router.message(Command("redraw"))
|
||||
async def redraw_lottery(message: Message):
|
||||
"""
|
||||
Переиграть розыгрыш для неподтвержденных выигрышей
|
||||
Формат: /redraw <lottery_id>
|
||||
"""
|
||||
if not is_admin(message.from_user.id):
|
||||
await message.answer("❌ Недостаточно прав")
|
||||
return
|
||||
|
||||
parts = message.text.split()
|
||||
if len(parts) != 2:
|
||||
await message.answer(
|
||||
"❌ Неверный формат команды\n\n"
|
||||
"Используйте: /redraw <lottery_id>"
|
||||
)
|
||||
return
|
||||
|
||||
try:
|
||||
lottery_id = int(parts[1])
|
||||
except ValueError:
|
||||
await message.answer("❌ lottery_id должен быть числом")
|
||||
return
|
||||
|
||||
try:
|
||||
async with async_session_maker() as session:
|
||||
from sqlalchemy.orm import selectinload
|
||||
from src.core.models import Lottery
|
||||
|
||||
# Загружаем розыгрыш с участниками
|
||||
lottery_result = await session.execute(
|
||||
select(Lottery)
|
||||
.options(selectinload(Lottery.participations))
|
||||
.where(Lottery.id == lottery_id)
|
||||
)
|
||||
lottery = lottery_result.scalar_one_or_none()
|
||||
|
||||
if not lottery:
|
||||
await message.answer(f"❌ Розыгрыш #{lottery_id} не найден")
|
||||
return
|
||||
|
||||
winners = await LotteryService.get_winners(session, lottery_id)
|
||||
|
||||
# Находим неподтвержденные выигрыши старше 24 часов
|
||||
now = datetime.now(timezone.utc)
|
||||
unclaimed_winners = []
|
||||
|
||||
for winner in winners:
|
||||
if not winner.is_claimed and winner.is_notified:
|
||||
time_passed = now - winner.created_at
|
||||
if time_passed.total_seconds() > 24 * 3600:
|
||||
unclaimed_winners.append(winner)
|
||||
|
||||
if not unclaimed_winners:
|
||||
await message.answer(
|
||||
"✅ Нет неподтвержденных выигрышей старше 24 часов.\n"
|
||||
"Повторный розыгрыш не требуется."
|
||||
)
|
||||
return
|
||||
|
||||
# Получаем всех участников, исключая текущих победителей
|
||||
all_participants = []
|
||||
current_winner_accounts = set()
|
||||
|
||||
for winner in winners:
|
||||
if winner.account_number:
|
||||
current_winner_accounts.add(winner.account_number)
|
||||
|
||||
for p in lottery.participations:
|
||||
if p.account_number and p.account_number not in current_winner_accounts:
|
||||
all_participants.append(p)
|
||||
|
||||
if not all_participants:
|
||||
await message.answer(
|
||||
"❌ Нет доступных участников для повторного розыгрыша.\n"
|
||||
"Все участники уже являются победителями."
|
||||
)
|
||||
return
|
||||
|
||||
# Переигрываем каждое неподтвержденное место
|
||||
redraw_results = []
|
||||
|
||||
for old_winner in unclaimed_winners:
|
||||
if not all_participants:
|
||||
break
|
||||
|
||||
# Выбираем нового победителя
|
||||
new_participant = random.choice(all_participants)
|
||||
all_participants.remove(new_participant)
|
||||
|
||||
# Удаляем старого победителя
|
||||
await session.delete(old_winner)
|
||||
|
||||
# Создаем нового победителя
|
||||
new_winner = Winner(
|
||||
lottery_id=lottery_id,
|
||||
user_id=None,
|
||||
account_number=new_participant.account_number,
|
||||
account_id=new_participant.account_id,
|
||||
place=old_winner.place,
|
||||
prize=old_winner.prize,
|
||||
is_manual=False,
|
||||
is_notified=False,
|
||||
is_claimed=False
|
||||
)
|
||||
session.add(new_winner)
|
||||
|
||||
redraw_results.append({
|
||||
'place': old_winner.place,
|
||||
'prize': old_winner.prize,
|
||||
'old_account': old_winner.account_number,
|
||||
'new_account': new_participant.account_number
|
||||
})
|
||||
|
||||
await session.commit()
|
||||
|
||||
# Отправляем уведомления новым победителям
|
||||
for result in redraw_results:
|
||||
# Находим нового победителя
|
||||
new_winner_result = await session.execute(
|
||||
select(Winner).where(
|
||||
and_(
|
||||
Winner.lottery_id == lottery_id,
|
||||
Winner.place == result['place'],
|
||||
Winner.account_number == result['new_account']
|
||||
)
|
||||
)
|
||||
)
|
||||
new_winner = new_winner_result.scalar_one_or_none()
|
||||
|
||||
if new_winner:
|
||||
# Отправляем уведомление новому победителю
|
||||
owner = await AccountService.get_account_owner(session, new_winner.account_number)
|
||||
|
||||
if owner and owner.telegram_id:
|
||||
# Создаем токен верификации
|
||||
await WinnerNotificationService.create_verification_token(
|
||||
session,
|
||||
new_winner.id
|
||||
)
|
||||
|
||||
# Формируем сообщение
|
||||
notification_message = (
|
||||
f"🎉 Поздравляем! Ваш счет выиграл!\n\n"
|
||||
f"🎯 Розыгрыш: {lottery.title}\n"
|
||||
f"🏆 Место: {new_winner.place}\n"
|
||||
f"🎁 Приз: {new_winner.prize}\n"
|
||||
f"💳 Счет: {new_winner.account_number}\n\n"
|
||||
f"⏰ **У вас есть 24 часа для подтверждения!**\n\n"
|
||||
f"Нажмите кнопку ниже, чтобы подтвердить получение приза."
|
||||
)
|
||||
|
||||
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
||||
[InlineKeyboardButton(
|
||||
text="✅ Подтвердить получение приза",
|
||||
callback_data=f"confirm_win_{new_winner.id}"
|
||||
)]
|
||||
])
|
||||
|
||||
try:
|
||||
await message.bot.send_message(
|
||||
owner.telegram_id,
|
||||
notification_message,
|
||||
reply_markup=keyboard,
|
||||
parse_mode="Markdown"
|
||||
)
|
||||
|
||||
new_winner.is_notified = True
|
||||
await session.commit()
|
||||
except:
|
||||
pass
|
||||
|
||||
# Формируем отчет для админа
|
||||
text = f"🔄 **Повторный розыгрыш завершен!**\n\n"
|
||||
text += f"🎯 Розыгрыш: {lottery.title}\n"
|
||||
text += f"📊 Переиграно мест: {len(redraw_results)}\n\n"
|
||||
|
||||
for result in redraw_results:
|
||||
text += f"🏆 {result['place']} место - {result['prize']}\n"
|
||||
text += f" ❌ Было: {result['old_account']}\n"
|
||||
text += f" ✅ Стало: {result['new_account']}\n\n"
|
||||
|
||||
text += "📨 Новым победителям отправлены уведомления"
|
||||
|
||||
await message.answer(text, parse_mode="Markdown")
|
||||
|
||||
except Exception as e:
|
||||
await message.answer(f"❌ Ошибка: {str(e)}")
|
||||
Reference in New Issue
Block a user