75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
from aiogram import Router, F
|
|
from aiogram.fsm.context import FSMContext
|
|
from aiogram.types import Message, CallbackQuery, InlineKeyboardButton
|
|
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
|
from babel.support import LazyProxy
|
|
from logging import getLogger
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
|
from typing import TYPE_CHECKING
|
|
from ....model.settings import Settings
|
|
from ....model.user import UserBase
|
|
from ..context import ContextData, CallbackCommand
|
|
from ..common import get_send_message
|
|
from ....model.descriptors import EntityCaptionCallable
|
|
|
|
if TYPE_CHECKING:
|
|
from ....main import QBotApp
|
|
|
|
|
|
logger = getLogger(__name__)
|
|
router = Router()
|
|
|
|
|
|
@router.callback_query(ContextData.filter(F.command == CallbackCommand.MENU_ENTRY_ENTITIES))
|
|
async def menu_entry_entities(message: CallbackQuery, **kwargs):
|
|
|
|
callback_data: ContextData = kwargs["callback_data"]
|
|
state: FSMContext = kwargs["state"]
|
|
state_data = await state.get_data()
|
|
kwargs["state_data"] = state_data
|
|
|
|
stack = save_navigation_context(callback_data = callback_data, state_data = state_data)
|
|
|
|
await entities_menu(message = message, navigation_stack = stack, **kwargs)
|
|
|
|
|
|
async def entities_menu(message: Message | CallbackQuery,
|
|
app: "QBotApp",
|
|
state: FSMContext,
|
|
navigation_stack: list[ContextData],
|
|
**kwargs):
|
|
|
|
keyboard_builder = InlineKeyboardBuilder()
|
|
|
|
entity_metadata = app.entity_metadata
|
|
|
|
for entity in entity_metadata.entity_descriptors.values():
|
|
if entity.caption_plural.__class__ == EntityCaptionCallable:
|
|
caption = entity.caption_plural(entity) or entity.name
|
|
elif entity.caption_plural.__class__ == LazyProxy:
|
|
caption = f"{f"{entity.icon} " if entity.icon else ""}{entity.caption_plural.value or entity.name}"
|
|
else:
|
|
caption = f"{f"{entity.icon} " if entity.icon else ""}{entity.caption_plural or entity.name}"
|
|
|
|
keyboard_builder.row(
|
|
InlineKeyboardButton(
|
|
text = caption,
|
|
callback_data = ContextData(command = CallbackCommand.ENTITY_LIST, entity_name = entity.name).pack()))
|
|
|
|
context = pop_navigation_context(navigation_stack)
|
|
if context:
|
|
keyboard_builder.row(
|
|
InlineKeyboardButton(
|
|
text = (await Settings.get(Settings.APP_STRINGS_BACK_BTN)),
|
|
callback_data = context.pack()))
|
|
|
|
state_data = kwargs["state_data"]
|
|
await state.set_data(state_data)
|
|
|
|
send_message = get_send_message(message)
|
|
|
|
await send_message(text = (await Settings.get(Settings.APP_STRINGS_REFERENCES)), reply_markup = keyboard_builder.as_markup())
|
|
|
|
|
|
from ..navigation import save_navigation_context, pop_navigation_context
|