101 lines
2.8 KiB
Python
101 lines
2.8 KiB
Python
from aiogram import Router, Bot, F
|
|
from aiogram.types import (
|
|
BufferedInputFile, InputMediaAudio,
|
|
ChosenInlineResult,
|
|
)
|
|
|
|
from bot.modules.youtube.downloader import YouTubeBytestream
|
|
|
|
from bot.utils.config import config
|
|
from bot.modules.database import db
|
|
from bot.modules.settings import UserSettings
|
|
|
|
from io import BytesIO
|
|
|
|
router = Router()
|
|
|
|
|
|
@router.chosen_inline_result(
|
|
F.result_id.startswith('spotc::') | F.result_id.startswith('ytc::')
|
|
)
|
|
async def on_cached_chosen(chosen_result: ChosenInlineResult, bot: Bot,
|
|
settings: UserSettings):
|
|
if settings['recode_youtube'].value != 'yes':
|
|
await bot.edit_message_reply_markup(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
reply_markup=None
|
|
)
|
|
return
|
|
|
|
if (
|
|
type(
|
|
db.recoded.get(
|
|
song_id := chosen_result.result_id
|
|
.removeprefix('spotc::')
|
|
.removeprefix('ytc::')
|
|
)
|
|
) in [bool, type(None)]
|
|
):
|
|
await bot.edit_message_reply_markup(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
reply_markup=None
|
|
)
|
|
return
|
|
|
|
await bot.edit_message_caption(
|
|
inline_message_id=chosen_result.inline_message_id,
|
|
caption='🔄 Recoding...',
|
|
reply_markup=None
|
|
)
|
|
|
|
message = await bot.forward_message(
|
|
config.telegram.files_chat,
|
|
config.telegram.files_chat,
|
|
db.recoded[song_id]
|
|
)
|
|
|
|
song_io: BytesIO = await bot.download( # type: ignore
|
|
destination=BytesIO(),
|
|
file=message.audio.file_id,
|
|
)
|
|
await message.delete()
|
|
|
|
bytestream = YouTubeBytestream.from_bytestream(
|
|
bytestream=song_io,
|
|
filename=message.audio.file_name,
|
|
duration=message.audio.duration,
|
|
)
|
|
|
|
await bytestream.rerender()
|
|
|
|
audio = await bot.send_audio(
|
|
chat_id=config.telegram.files_chat,
|
|
audio=BufferedInputFile(
|
|
file=bytestream.file,
|
|
filename=bytestream.filename,
|
|
),
|
|
thumbnail=BufferedInputFile(
|
|
file=(await bot.download(message.audio.thumbnail.file_id)).read(),
|
|
filename='thumbnail.jpg'
|
|
),
|
|
performer=message.audio.performer,
|
|
title=message.audio.title,
|
|
duration=bytestream.duration,
|
|
)
|
|
|
|
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)
|
|
)
|
|
|
|
if chosen_result.result_id.startswith('spotc::'):
|
|
db.spotify[song_id] = audio.audio.file_id
|
|
else:
|
|
db.youtube[song_id] = audio.audio.file_id
|
|
db.recoded[song_id] = True
|