telegram-bot/telegram_helpers.py

10 lines
357 B
Python

def build_menu(buttons: [],
n_cols: int,
header_buttons: [] = None,
footer_buttons: [] = None):
menu = [buttons[i:i + n_cols] for i in range(0, len(buttons), n_cols)]
if header_buttons:
menu.insert(0, [header_buttons])
if footer_buttons:
menu.append([footer_buttons])
return menu