mirror of
https://github.com/koloideal/Quizzi.git
synced 2026-06-10 10:25:28 +03:00
74 lines
3.2 KiB
Python
74 lines
3.2 KiB
Python
from aiogram.types import CallbackQuery, Message
|
|
from aiogram_dialog import Dialog, DialogManager, StartMode, Window
|
|
from aiogram_dialog.widgets.input import MessageInput
|
|
from aiogram_dialog.widgets.kbd import Button, Row
|
|
from aiogram_dialog.widgets.text import Const
|
|
from dishka import FromDishka
|
|
from dishka.integrations.aiogram_dialog import inject
|
|
|
|
from trudex.application.bot.admin_dialogs.states import AdminBroadcastSG, AdminMenuSG
|
|
from trudex.infrastructure.database.dao.user import UserDAO
|
|
from trudex.infrastructure.utils.broadcast import broadcast_message
|
|
|
|
|
|
async def on_broadcast_input(message: Message, _widget: MessageInput, manager: DialogManager):
|
|
manager.dialog_data["broadcast_message_id"] = message.message_id
|
|
manager.dialog_data["broadcast_chat_id"] = message.chat.id
|
|
await manager.switch_to(AdminBroadcastSG.broadcast_confirm)
|
|
|
|
|
|
@inject
|
|
async def on_broadcast_confirm(_callback: CallbackQuery, _button: Button, manager: DialogManager, user_dao: FromDishka[UserDAO]):
|
|
message_id = manager.dialog_data.get("broadcast_message_id")
|
|
chat_id = manager.dialog_data.get("broadcast_chat_id")
|
|
|
|
if not message_id or not chat_id or not _callback.message:
|
|
await _callback.answer("Ошибка: сообщение не найдено")
|
|
return
|
|
|
|
await _callback.message.answer("⏳ Рассылка началась...")
|
|
|
|
bot = _callback.bot
|
|
if not bot:
|
|
await _callback.answer("Ошибка: бот не найден")
|
|
return
|
|
|
|
stats = await broadcast_message(bot, message_id, chat_id, user_dao)
|
|
|
|
stats_text = (
|
|
f"✅ <b>Рассылка завершена</b>\n\n"
|
|
f"Всего пользователей: {stats.total}\n"
|
|
f"Успешно отправлено: {stats.success}\n"
|
|
f"Не удалось отправить: {stats.failed}"
|
|
)
|
|
|
|
await _callback.message.answer(stats_text)
|
|
await manager.done()
|
|
|
|
|
|
async def on_broadcast_cancel(_callback: CallbackQuery, _button: Button, manager: DialogManager):
|
|
await _callback.answer("Рассылка отменена")
|
|
await manager.start(AdminMenuSG.main, mode=StartMode.RESET_STACK)
|
|
|
|
|
|
async def on_back_to_main(_callback: CallbackQuery, _button: Button, manager: DialogManager):
|
|
await manager.start(AdminMenuSG.main, mode=StartMode.RESET_STACK)
|
|
|
|
|
|
broadcast_dialog = Dialog(
|
|
Window(
|
|
Const("<b>📢 Рассылка</b>\n\nОтправьте сообщение, которое хотите разослать всем пользователям:"),
|
|
MessageInput(on_broadcast_input),
|
|
Button(Const("◀️ Отмена"), id="back", on_click=on_back_to_main),
|
|
state=AdminBroadcastSG.broadcast_input,
|
|
),
|
|
Window(
|
|
Const("<b>⚠️ Подтверждение рассылки</b>\n\nВы уверены, что хотите отправить это сообщение всем пользователям?"),
|
|
Row(
|
|
Button(Const("✅ Да"), id="broadcast_confirm", on_click=on_broadcast_confirm),
|
|
Button(Const("❌ Нет"), id="broadcast_cancel", on_click=on_broadcast_cancel),
|
|
),
|
|
state=AdminBroadcastSG.broadcast_confirm,
|
|
),
|
|
)
|