add mailing function

app/bot/dialogs/flows/mailing/handlers.py

@@ -0,0 +1,54 @@
+import asyncio
+
+from aiogram.types import CallbackQuery, Message
+from aiogram_dialog import DialogManager
+from aiogram_dialog.widgets.input import MessageInput
+from aiogram_dialog.widgets.kbd import Button
+
+from app.infrastructure.database.crud import get_users
+
+
+async def message_data(
+    message: Message,
+    widget: MessageInput,
+    manager: DialogManager,
+):
+    manager.dialog_data["message"] = message
+    await manager.next()
+
+
+async def choose_recipients(
+    callback: CallbackQuery,
+    button: Button,
+    manager: DialogManager,
+):
+    manager.dialog_data["recipients"] = button.widget_id
+    await manager.next()
+
+
+async def confirm_mailing(
+    callback: CallbackQuery,
+    button: Button,
+    manager: DialogManager,
+):
+    print(manager.dialog_data["recipients"])
+    users = list(await get_users(
+        manager.middleware_data["session"],
+        exclude_admins=manager.dialog_data["recipients"] == "send_users",
+    ))
+
+    await asyncio.gather(
+        *[
+            callback.bot.copy_message(
+                chat_id=user.tg_id,
+                from_chat_id=manager.dialog_data["message"].chat.id,
+                message_id=manager.dialog_data["message"].message_id,
+            )
+            for user in users
+        ]
+    )
+    await callback.bot.send_message(
+        chat_id=callback.from_user.id,
+        text=f"Разослано сообщений: {len(users)}",
+    )
+    await manager.done()