141 lines
6.2 KiB
Python
141 lines
6.2 KiB
Python
# bot/handlers.py
|
||
import re
|
||
from telegram import Update, InlineKeyboardMarkup, InlineKeyboardButton
|
||
from telegram.ext import ContextTypes
|
||
from asgiref.sync import sync_to_async
|
||
from webapp.models import BindingRequest, Client, Invoice
|
||
from draw.models import Lottery, DrawResult, LotteryParticipant
|
||
from bot.state import BINDING_PENDING
|
||
|
||
@sync_to_async
|
||
def get_client_by_telegram_id(telegram_id: int):
|
||
try:
|
||
return Client.objects.get(telegram_id=str(telegram_id))
|
||
except Client.DoesNotExist:
|
||
return None
|
||
|
||
@sync_to_async
|
||
def get_invoices_for_client(client):
|
||
return list(Invoice.objects.filter(client_club_card_number=client.club_card_number))
|
||
|
||
@sync_to_async
|
||
def get_draws_for_client(client):
|
||
return list(
|
||
DrawResult.objects.filter(participant__invoice__client_club_card_number=client.club_card_number)
|
||
.select_related('lottery', 'prize', 'participant__invoice')
|
||
)
|
||
|
||
@sync_to_async
|
||
def create_binding_request(telegram_chat_id, client_card):
|
||
return BindingRequest.objects.create(
|
||
telegram_chat_id=str(telegram_chat_id),
|
||
client_card=client_card
|
||
)
|
||
|
||
async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||
if not update.message:
|
||
return
|
||
chat_id = update.message.chat_id
|
||
client = await get_client_by_telegram_id(chat_id)
|
||
if not client:
|
||
await update.message.reply_text(
|
||
"❌ Пользователь не найден. Возможно, Вы ещё не привязали свою клубную карту."
|
||
)
|
||
return
|
||
|
||
invoices = await get_invoices_for_client(client)
|
||
draws = await get_draws_for_client(client)
|
||
|
||
message_text = f"👤 *Профиль клиента:*\n" \
|
||
f"• *Имя:* {client.name}\n" \
|
||
f"• *Клубная карта:* {client.club_card_number}\n\n"
|
||
|
||
message_text += "💳 *Счета:*\n"
|
||
if invoices:
|
||
show_all = len(invoices) > 5
|
||
displayed_invoices = invoices[:5] if show_all else invoices
|
||
for inv in displayed_invoices:
|
||
invoice_str = str(inv).split('/')[0].strip()
|
||
message_text += f" • {invoice_str} (*{inv.sum}*)\n"
|
||
else:
|
||
message_text += " _Нет счетов_\n"
|
||
|
||
message_text += "\n🎉 *Розыгрыши и выигранные призы:*\n"
|
||
if draws:
|
||
for draw in draws:
|
||
lottery_name = draw.lottery.name if draw.lottery else "неизвестно"
|
||
invoice_info = (draw.participant.invoice
|
||
if draw.participant and hasattr(draw.participant, 'invoice')
|
||
else "неизвестно")
|
||
prize_info = (draw.prize.reward
|
||
if draw.prize and hasattr(draw.prize, 'reward')
|
||
else "неизвестно")
|
||
message_text += f" • Лотерея: *{lottery_name}*\n Счет: _{invoice_info}_\n Выигрыш: *{prize_info}*\n\n"
|
||
else:
|
||
message_text += " _Нет результатов розыгрышей_\n"
|
||
|
||
reply_markup = None
|
||
if invoices and len(invoices) > 5:
|
||
reply_markup = InlineKeyboardMarkup([
|
||
[InlineKeyboardButton("Показать все счета", callback_data="show_all_invoices")]
|
||
])
|
||
|
||
await update.message.reply_text(message_text, parse_mode="Markdown", reply_markup=reply_markup)
|
||
|
||
async def handle_client_card(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||
"""
|
||
Обработчик команды /bind.
|
||
После ввода команды бот запрашивает ввод номера КК и ожидает сообщение с кодом.
|
||
Здесь мы просто добавляем ID пользователя в pending.
|
||
"""
|
||
if not update.message:
|
||
return
|
||
telegram_chat_id = update.message.chat_id
|
||
# Если пользователь уже привязан, сообщаем об этом.
|
||
client = await get_client_by_telegram_id(telegram_chat_id)
|
||
if client:
|
||
await update.message.reply_text("✅ Вы уже привязаны!")
|
||
return
|
||
|
||
# Добавляем пользователя в список pending и просим ввести номер КК.
|
||
BINDING_PENDING.add(telegram_chat_id)
|
||
await update.message.reply_text("Введите номер вашей клубной карты (КК):")
|
||
|
||
async def process_binding_input(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||
"""
|
||
Обрабатывает ввод номера КК от пользователя, который находится в состоянии ожидания.
|
||
"""
|
||
if not update.message or not update.message.text:
|
||
return
|
||
client_card = update.message.text.strip()
|
||
telegram_chat_id = update.message.chat_id
|
||
|
||
if not re.fullmatch(r'\d{3,}', client_card):
|
||
await update.message.reply_text("❌ Неверный формат клиентской карты. Пожалуйста, введите корректный код (минимум 3 цифры).")
|
||
return
|
||
|
||
await create_binding_request(telegram_chat_id, client_card)
|
||
if telegram_chat_id in BINDING_PENDING:
|
||
BINDING_PENDING.remove(telegram_chat_id)
|
||
await update.message.reply_text("✅ Заявка отправлена. После подтверждения вы сможете участвовать в чате.")
|
||
|
||
async def show_all_invoices_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||
query = update.callback_query
|
||
await query.answer()
|
||
chat_id = query.message.chat_id
|
||
client = await get_client_by_telegram_id(chat_id)
|
||
if not client:
|
||
await query.edit_message_text("❌ Пользователь не найден. Возможно, Вы ещё не привязали свою клубную карту.")
|
||
return
|
||
|
||
invoices = await get_invoices_for_client(client)
|
||
message_text = "💳 *Все счета:*\n"
|
||
if invoices:
|
||
for inv in invoices:
|
||
invoice_str = str(inv).split('/')[0].strip()
|
||
message_text += f" • {invoice_str} (*{inv.sum}*)\n"
|
||
else:
|
||
message_text += " _Нет счетов_\n"
|
||
|
||
await query.edit_message_text(message_text, parse_mode="Markdown")
|