11 lines
357 B
Python
11 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
|