Did some refactor, add YouTube as search variant
This commit is contained in:
1
bot/results/common/__init__.py
Normal file
1
bot/results/common/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
pass
|
||||
39
bot/results/common/search.py
Normal file
39
bot/results/common/search.py
Normal file
@@ -0,0 +1,39 @@
|
||||
from aiogram.types import (
|
||||
InlineQueryResultDocument, InlineQueryResultCachedAudio,
|
||||
InlineKeyboardMarkup, InlineKeyboardButton
|
||||
)
|
||||
|
||||
from bot.modules.database.db import DBDict
|
||||
|
||||
from bot.modules.common.song import BaseSongItem
|
||||
from typing import TypeVar
|
||||
|
||||
|
||||
BaseSongT = TypeVar('BaseSongT', bound=BaseSongItem)
|
||||
|
||||
|
||||
async def get_common_search_result(
|
||||
audio: BaseSongT,
|
||||
db_table: DBDict,
|
||||
service_id: str
|
||||
) -> InlineQueryResultDocument | InlineQueryResultCachedAudio:
|
||||
return (
|
||||
InlineQueryResultDocument(
|
||||
id=f'{service_id}::' + audio.id,
|
||||
title=audio.name,
|
||||
description=audio.all_artists,
|
||||
thumb_url=audio.thumbnail,
|
||||
document_url=audio.preview_url or audio.thumbnail,
|
||||
mime_type='application/zip',
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
inline_keyboard=[
|
||||
[InlineKeyboardButton(text='Downloading...', callback_data='.')]
|
||||
]
|
||||
),
|
||||
caption=audio.full_name,
|
||||
) if audio.id not in list(db_table.keys()) else
|
||||
InlineQueryResultCachedAudio(
|
||||
id=f'{service_id}c::' + audio.id,
|
||||
audio_file_id=db_table[audio.id],
|
||||
)
|
||||
)
|
||||
Reference in New Issue
Block a user