Vladless
32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
from aiogram.types import InlineKeyboardMarkup
|
|
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
|
|
|
|
|
def insert_hook_buttons(builder: InlineKeyboardBuilder, buttons: list) -> InlineKeyboardBuilder:
|
|
"""
|
|
Вставляет кнопки из хуков в существующий builder (вставка после указанной кнопки через `after`)
|
|
"""
|
|
markup = builder.as_markup()
|
|
new_rows = markup.inline_keyboard.copy()
|
|
|
|
for module in buttons:
|
|
if isinstance(module, dict) and "after" in module and "button" in module:
|
|
after = module["after"]
|
|
button = module["button"]
|
|
|
|
insert_pos = -1
|
|
for i, row in enumerate(new_rows):
|
|
if any(btn.callback_data == after for btn in row):
|
|
insert_pos = i + 1
|
|
break
|
|
|
|
if 0 <= insert_pos <= len(new_rows):
|
|
new_rows.insert(insert_pos, [button])
|
|
else:
|
|
new_rows.append([button])
|
|
else:
|
|
button = module.get("button") if isinstance(module, dict) else module
|
|
new_rows.append([button])
|
|
|
|
return InlineKeyboardBuilder.from_markup(InlineKeyboardMarkup(inline_keyboard=new_rows))
|