feat(telegram_bot): added dialogs folder
This commit is contained in:
@@ -0,0 +1,54 @@
|
||||
from typing import Any
|
||||
|
||||
from aiogram.fsm.context import FSMContext
|
||||
from aiogram.types import CallbackQuery
|
||||
from aiogram_dialog import Dialog, DialogManager, Window
|
||||
from aiogram_dialog.widgets.kbd import Button, ListGroup, ScrollingGroup
|
||||
from aiogram_dialog.widgets.text import Const, Format
|
||||
|
||||
from api.client import AdNovaClient
|
||||
from states.campaigns import CampaignsDailogState
|
||||
|
||||
|
||||
async def campaigns(**kwargs: dict[Any]) -> dict[str, Any]:
|
||||
state: FSMContext = kwargs["state"]
|
||||
state_data = await state.get_data()
|
||||
|
||||
async with AdNovaClient() as client:
|
||||
campaigns = await client.list_campaigns(state_data["advertiser_id"])
|
||||
|
||||
campaigns = [campaign.model_dump(mode="json") for campaign in campaigns]
|
||||
|
||||
return {
|
||||
"campaigns": campaigns,
|
||||
}
|
||||
|
||||
|
||||
async def campaign_detail_on_click(
|
||||
callback: CallbackQuery, button: Button, manager: DialogManager
|
||||
) -> None:
|
||||
await callback.answer()
|
||||
|
||||
|
||||
campaigns_dialog = Dialog(
|
||||
Window(
|
||||
Const("Campaigns:"),
|
||||
ScrollingGroup(
|
||||
ListGroup(
|
||||
Button(
|
||||
Format("{item[ad_title]}"),
|
||||
id="detail",
|
||||
on_click=campaign_detail_on_click,
|
||||
),
|
||||
id="campaigns",
|
||||
item_id_getter=lambda item: item["campaign_id"],
|
||||
items="campaigns",
|
||||
),
|
||||
id="pagination",
|
||||
width=1,
|
||||
height=4,
|
||||
),
|
||||
state=CampaignsDailogState.campaigns,
|
||||
getter=campaigns,
|
||||
),
|
||||
)
|
||||
@@ -0,0 +1,63 @@
|
||||
from aiogram.types import Message
|
||||
from aiogram_dialog import Dialog, DialogManager, Window
|
||||
from aiogram_dialog.widgets.input import ManagedTextInput, TextInput
|
||||
from aiogram_dialog.widgets.text import Const
|
||||
from pydantic import ValidationError
|
||||
|
||||
from api.client import AdNovaClient
|
||||
from api.errors import NotFoundError
|
||||
from api.schemas import Advertiser
|
||||
from states.start import StartDialogState
|
||||
|
||||
|
||||
def check_advertiser_id(advertiser_id: str) -> None:
|
||||
Advertiser.__pydantic_validator__.validate_assignment(
|
||||
Advertiser.model_construct(), "advertiser_id", advertiser_id
|
||||
)
|
||||
|
||||
try:
|
||||
client = AdNovaClient()
|
||||
client.sync_get_advertiser(advertiser_id)
|
||||
except NotFoundError:
|
||||
raise ValueError from None
|
||||
|
||||
|
||||
async def advertiser_id_on_error(
|
||||
message: Message,
|
||||
widget: ManagedTextInput,
|
||||
dialog_manager: DialogManager,
|
||||
error: object,
|
||||
) -> None:
|
||||
if isinstance(error, ValidationError):
|
||||
await message.answer("Invalid advertiser UUID.")
|
||||
elif isinstance(error, ValueError):
|
||||
await message.answer("Advertiser with this UUID not found.")
|
||||
|
||||
|
||||
async def advertiser_id_on_success(
|
||||
message: Message,
|
||||
widget: ManagedTextInput,
|
||||
dialog_manager: DialogManager,
|
||||
advertiser_id: str,
|
||||
) -> None:
|
||||
state = dialog_manager.middleware_data["state"]
|
||||
state_data = await state.get_data()
|
||||
state_data["advertiser_id"] = message.text
|
||||
await state.set_data(state_data)
|
||||
|
||||
await message.answer(f"Successfully authenticated as {message.text}")
|
||||
await dialog_manager.mark_closed()
|
||||
|
||||
|
||||
start_dialog = Dialog(
|
||||
Window(
|
||||
Const("Enter adveritser UUID:"),
|
||||
TextInput(
|
||||
id="advertiser_id",
|
||||
type_factory=check_advertiser_id,
|
||||
on_success=advertiser_id_on_success,
|
||||
on_error=advertiser_id_on_error,
|
||||
),
|
||||
state=StartDialogState.start,
|
||||
),
|
||||
)
|
||||
Reference in New Issue
Block a user