Bot container restart from admin-panel

lottery/bot/handlers.py

@@ -1,11 +1,12 @@
-# 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 draw.models import DrawResult
 from bot.state import BINDING_PENDING
+from bot.models import BotConfig
+
 
 @sync_to_async
 def get_client_by_telegram_id(telegram_id: int):
@@ -32,9 +33,11 @@ def create_binding_request(telegram_chat_id, client_card):
         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:
@@ -45,7 +48,7 @@ async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
 
     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"
@@ -55,8 +58,7 @@ async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
         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"
+            message_text += f"  • ID {inv.id} (*{inv.sum}*)\n"
     else:
         message_text += "  _Нет счетов_\n"
 
@@ -64,12 +66,8 @@ async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
     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 "неизвестно")
+            invoice_info = getattr(draw.participant.invoice, 'id', 'неизвестно') if draw.participant else 'неизвестно'
+            prize_info = getattr(draw.prize, 'reward', 'неизвестно')
             message_text += f"  • Лотерея: *{lottery_name}*\n    Счет: _{invoice_info}_\n    Выигрыш: *{prize_info}*\n\n"
     else:
         message_text += "  _Нет результатов розыгрышей_\n"
@@ -82,43 +80,37 @@ async def whoami(update: Update, context: ContextTypes.DEFAULT_TYPE):
 
     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 цифры).")
+        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)
+    BINDING_PENDING.discard(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()
@@ -132,9 +124,18 @@ async def show_all_invoices_callback(update: Update, context: ContextTypes.DEFAU
     message_text = "💳 *Все счета:*\n"
     if invoices:
         for inv in invoices:
-            invoice_str = str(inv).split('/')[0].strip()
-            message_text += f"  • {invoice_str} (*{inv.sum}*)\n"
+            message_text += f"  • ID {inv.id} (*{inv.sum}*)\n"
     else:
         message_text += "  _Нет счетов_\n"
 
     await query.edit_message_text(message_text, parse_mode="Markdown")
+
+
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    if not update.message:
+        return
+    bot_config = await sync_to_async(BotConfig.objects.filter(is_active=True).first)()
+    if bot_config and bot_config.active_welcome:
+        await update.message.reply_text(bot_config.active_welcome.welcome_message)
+    else:
+        await update.message.reply_text("Привет! Бот работает.")