93 lines
3.6 KiB
Python
93 lines
3.6 KiB
Python
from aiogram.fsm.context import FSMContext
|
|
from aiogram.types import Message, CallbackQuery
|
|
|
|
from .context import ContextData, CallbackCommand
|
|
|
|
|
|
async def save_navigation_context(callback_data: ContextData, state: FSMContext) -> list[ContextData]:
|
|
data = await state.get_data()
|
|
stack = [ContextData.unpack(item) for item in data.get("navigation_stack", [])]
|
|
data_nc = data.get("navigation_context")
|
|
navigation_context = ContextData.unpack(data_nc) if data_nc else None
|
|
if callback_data.back:
|
|
callback_data.back = False
|
|
if stack:
|
|
stack.pop()
|
|
else:
|
|
if stack and navigation_context and navigation_context.command == callback_data.command:
|
|
navigation_context = callback_data
|
|
elif navigation_context:
|
|
stack.append(navigation_context)
|
|
|
|
await state.update_data({"navigation_stack": [item.pack() for item in stack],
|
|
"navigation_context": callback_data.pack()})
|
|
return stack
|
|
|
|
|
|
def pop_navigation_context(stack: list[ContextData]) -> ContextData | None:
|
|
if stack:
|
|
data = stack[-1]
|
|
data.back = True
|
|
return data
|
|
|
|
|
|
async def get_navigation_context(state: FSMContext) -> tuple[list[ContextData], ContextData | None]:
|
|
data = await state.get_data()
|
|
data_nc = data.get("navigation_context")
|
|
context = ContextData.unpack(data_nc) if data_nc else None
|
|
return ([ContextData.unpack(item) for item in data.get("navigation_stack", [])],
|
|
context)
|
|
|
|
|
|
async def clear_state(state: FSMContext, clear_nav: bool = False):
|
|
if clear_nav:
|
|
await state.clear()
|
|
else:
|
|
state_data = await state.get_data()
|
|
stack = state_data.get("navigation_stack")
|
|
context = state_data.get("navigation_context")
|
|
update_data = {}
|
|
if stack:
|
|
update_data["navigation_stack"] = stack
|
|
if context:
|
|
update_data["navigation_context"] = context
|
|
await state.clear()
|
|
await state.update_data(update_data)
|
|
|
|
|
|
async def route_callback(message: Message | CallbackQuery, back: bool = True, **kwargs):
|
|
|
|
stack, context = await get_navigation_context(kwargs["state"])
|
|
if back:
|
|
context = pop_navigation_context(stack)
|
|
stack = await save_navigation_context(callback_data = context, state = kwargs["state"])
|
|
kwargs.update({"callback_data": context, "navigation_stack": stack})
|
|
if context:
|
|
if context.command == CallbackCommand.MENU_ENTRY_MAIN:
|
|
await main_menu(message, **kwargs)
|
|
elif context.command == CallbackCommand.MENU_ENTRY_SETTINGS:
|
|
await settings_menu(message, **kwargs)
|
|
elif context.command == CallbackCommand.MENU_ENTRY_PARAMETERS:
|
|
await parameters_menu(message, **kwargs)
|
|
elif context.command == CallbackCommand.MENU_ENTRY_LANGUAGE:
|
|
await language_menu(message, **kwargs)
|
|
elif context.command == CallbackCommand.MENU_ENTRY_ENTITIES:
|
|
await entities_menu(message, **kwargs)
|
|
elif context.command == CallbackCommand.ENTITY_LIST:
|
|
await entity_list(message, **kwargs)
|
|
elif context.command == CallbackCommand.ENTITY_ITEM:
|
|
await entity_item(message, **kwargs)
|
|
|
|
else:
|
|
raise ValueError(f"Unknown command {context.command}")
|
|
else:
|
|
raise ValueError("No navigation context")
|
|
|
|
|
|
from .menu.main import main_menu
|
|
from .menu.settings import settings_menu
|
|
from .menu.parameters import parameters_menu
|
|
from .menu.language import language_menu
|
|
from .menu.entities import entities_menu
|
|
from .forms.entity_list import entity_list
|
|
from .forms.entity_form import entity_item |