163 lines
5.3 KiB
Python
163 lines
5.3 KiB
Python
from aiogram import Bot, F, Router
|
|
from aiogram.types import (
|
|
BufferedInputFile,
|
|
ChosenInlineResult,
|
|
InputMediaAudio,
|
|
URLInputFile,
|
|
)
|
|
|
|
from bot.modules.database import db
|
|
from bot.modules.deezer import deezer
|
|
from bot.modules.settings import UserSettings
|
|
from bot.modules.spotify import spotify
|
|
from bot.modules.youtube import AgeRestrictedError, youtube
|
|
from bot.modules.youtube.song import SongItem
|
|
from bot.utils.config import config
|
|
|
|
router = Router()
|
|
|
|
|
|
def not_strict_name(song, yt_song):
|
|
if "feat" in yt_song.name.lower():
|
|
return any(artist.lower() in yt_song.name.lower() for artist in song.artists)
|
|
else:
|
|
return False
|
|
|
|
|
|
@router.chosen_inline_result(F.result_id.startswith("spot::"))
|
|
async def on_new_chosen(
|
|
chosen_result: ChosenInlineResult, bot: Bot, settings: UserSettings
|
|
):
|
|
song = spotify.songs.from_id(chosen_result.result_id.removeprefix("spot::"))
|
|
|
|
bytestream = None
|
|
audio = None
|
|
|
|
yt_song: SongItem | None = youtube.songs.search_one(
|
|
song.full_name,
|
|
exact_match=True,
|
|
)
|
|
if settings["exact_spotify_search"].value == "yes":
|
|
if (
|
|
song.all_artists != yt_song.all_artists or song.name != yt_song.name
|
|
) and not not_strict_name(song, yt_song):
|
|
await bot.edit_message_caption(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
caption="🙄 Cannot find this song on YouTube, trying Deezer...",
|
|
reply_markup=None,
|
|
parse_mode="HTML",
|
|
)
|
|
yt_song = None
|
|
bytestream = False
|
|
|
|
try:
|
|
if bytestream is None:
|
|
bytestream = await yt_song.to_bytestream()
|
|
|
|
audio = await bot.send_audio(
|
|
chat_id=config.telegram.files_chat,
|
|
audio=BufferedInputFile(
|
|
file=bytestream.file,
|
|
filename=bytestream.filename,
|
|
),
|
|
thumbnail=URLInputFile(song.thumbnail),
|
|
performer=song.all_artists,
|
|
title=song.name,
|
|
duration=bytestream.duration,
|
|
)
|
|
db.youtube[yt_song.id] = audio.audio.file_id
|
|
|
|
except AgeRestrictedError:
|
|
await bot.edit_message_caption(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
caption="🔞 This song is age restricted, trying Deezer...",
|
|
reply_markup=None,
|
|
parse_mode="HTML",
|
|
)
|
|
yt_song = None
|
|
|
|
if not bytestream:
|
|
try:
|
|
deezer_song = await deezer.songs.search_one(
|
|
song.full_name,
|
|
)
|
|
|
|
bytestream = await (
|
|
await deezer.downloader.from_id(deezer_song.id)
|
|
).to_bytestream()
|
|
|
|
audio = await bot.send_audio(
|
|
chat_id=config.telegram.files_chat,
|
|
audio=BufferedInputFile(
|
|
file=bytestream.file,
|
|
filename=bytestream.filename,
|
|
),
|
|
thumbnail=URLInputFile(bytestream.song.thumbnail),
|
|
performer=bytestream.song.all_artists,
|
|
title=bytestream.song.name,
|
|
duration=bytestream.song.duration,
|
|
)
|
|
|
|
db.deezer[bytestream.song.id] = audio.audio.file_id
|
|
except Exception as e:
|
|
assert e
|
|
|
|
if audio:
|
|
if settings["exact_spotify_search"].value == "yes":
|
|
db.spotify[song.id] = audio.audio.file_id
|
|
|
|
await bot.edit_message_media(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
media=InputMediaAudio(media=audio.audio.file_id),
|
|
reply_markup=None,
|
|
)
|
|
|
|
else:
|
|
await bot.edit_message_caption(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
caption="🤷♂️ Cannot download this song",
|
|
reply_markup=None,
|
|
parse_mode="HTML",
|
|
)
|
|
|
|
if yt_song and settings["recode_youtube"].value == "yes":
|
|
await bot.edit_message_caption(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
caption="🔄 Recoding...",
|
|
reply_markup=None,
|
|
parse_mode="HTML",
|
|
)
|
|
await bytestream.rerender()
|
|
|
|
audio = await bot.send_audio(
|
|
chat_id=config.telegram.files_chat,
|
|
audio=BufferedInputFile(
|
|
file=bytestream.file,
|
|
filename=bytestream.filename,
|
|
),
|
|
thumbnail=URLInputFile(song.thumbnail),
|
|
performer=song.all_artists,
|
|
title=song.name,
|
|
duration=bytestream.duration,
|
|
)
|
|
db.youtube[yt_song.id] = audio.audio.file_id
|
|
db.recoded[yt_song.id] = True
|
|
|
|
if settings["exact_spotify_search"].value == "yes":
|
|
db.spotify[song.id] = audio.audio.file_id
|
|
db.recoded[song.id] = True
|
|
|
|
await bot.edit_message_caption(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
caption="",
|
|
reply_markup=None,
|
|
)
|
|
await bot.edit_message_media(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
media=InputMediaAudio(media=audio.audio.file_id),
|
|
)
|
|
elif yt_song and settings["recode_youtube"].value == "no":
|
|
db.recoded[yt_song.id] = audio.message_id
|
|
if settings["exact_spotify_search"].value == "yes":
|
|
db.recoded[song.id] = audio.message_id
|