Update menu

2023-11-06 18:27:03 +02:00
parent 130023c4b2
commit 54006008d3
2 changed files with 32 additions and 17 deletions
--- a/bot/keyboards/inline/search_variants.py
+++ b/bot/keyboards/inline/search_variants.py
@@ -16,10 +16,10 @@ spotify = {
 }


-def get_search_variants_kb(
+def get_search_variants(
        query: str,
        services: dict[str, str],
-) -> InlineKeyboardMarkup:
+) -> list[list[InlineKeyboardButton]]:
    buttons = [
        [
            InlineKeyboardButton(
@@ -29,4 +29,14 @@ def get_search_variants_kb(
        ] for key in services.keys()
    ]

-    return InlineKeyboardBuilder(buttons).as_markup()
+    return buttons
+
+
+def get_search_variants_kb(
+        query: str,
+        services: dict[str, str],
+) -> InlineKeyboardMarkup:
+    return InlineKeyboardBuilder(get_search_variants(
+        query,
+        services
+    )).as_markup()