diff --git a/bot/keyboards/inline/full_menu.py b/bot/keyboards/inline/full_menu.py index 6e1a798..639c268 100644 --- a/bot/keyboards/inline/full_menu.py +++ b/bot/keyboards/inline/full_menu.py @@ -2,21 +2,26 @@ from aiogram.utils.keyboard import (InlineKeyboardMarkup, InlineKeyboardButton, InlineKeyboardBuilder) from bot.factories.full_menu import FullMenuCallback +from bot.keyboards.inline import search_variants as sv + def get_full_menu_kb() -> InlineKeyboardMarkup: - buttons = [ - [ - InlineKeyboardButton( - text='⚙️ Settings', - callback_data=FullMenuCallback( - action='settings' - ).pack() - ), - InlineKeyboardButton( - text='🎵 Search in SoundCloud', - switch_inline_query_current_chat='sc::' - ) - ] - ] + buttons = (sv.get_search_variants( + query='', + services= + sv.soundcloud | + sv.spotify | + sv.deezer | + sv.youtube + ) + [ + [ + InlineKeyboardButton( + text='⚙️ Settings', + callback_data=FullMenuCallback( + action='settings' + ).pack() + ) + ], + ]) return InlineKeyboardBuilder(buttons).as_markup() diff --git a/bot/keyboards/inline/search_variants.py b/bot/keyboards/inline/search_variants.py index c07157f..5170348 100644 --- 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()