tgbots/AnyMusicBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add YouTube recoding

Browse Source
This commit is contained in:
BarsTiger
2023-11-07 14:13:14 +02:00
parent 54006008d3
commit 06c327933f
13 changed files with 245 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/handlers/on_chosen/__init__.py
View File

@@ -1,5 +1,5 @@
from aiogram import Router
from . import spotify, deezer, youtube
from . import spotify, deezer, youtube, recode_cached, suppress_verify
router = Router()
@@ -7,6 +7,8 @@ router.include_routers(
spotify.router,
deezer.router,
youtube.router,
recode_cached.router,
suppress_verify.router,
)
__all__ = ['router']

101
bot/handlers/on_chosen/recode_cached.py Normal file
View File

@@ -0,0 +1,101 @@
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
await db.occasionally_write()

46
bot/handlers/on_chosen/spotify.py
View File

@@ -10,6 +10,7 @@ from bot.modules.youtube.song import SongItem
from bot.modules.deezer import deezer
from bot.utils.config import config
from bot.modules.database import db
from bot.modules.settings import UserSettings
router = Router()
@@ -22,13 +23,14 @@ def not_strict_name(song, yt_song):
@router.chosen_inline_result(F.result_id.startswith('spot::'))
async def on_new_chosen(chosen_result: ChosenInlineResult, bot: Bot):
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 = youtube.songs.search_one(
yt_song: SongItem | None = youtube.songs.search_one(
song.full_name,
exact_match=True,
)
@@ -40,6 +42,7 @@ async def on_new_chosen(chosen_result: ChosenInlineResult, bot: Bot):
reply_markup=None,
parse_mode='HTML',
)
yt_song = None
bytestream = False
try:
@@ -66,6 +69,7 @@ async def on_new_chosen(chosen_result: ChosenInlineResult, bot: Bot):
reply_markup=None,
parse_mode='HTML',
)
yt_song = None
if not bytestream:
try:
@@ -110,4 +114,42 @@ async def on_new_chosen(chosen_result: ChosenInlineResult, bot: Bot):
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.spotify[song.id] = audio.audio.file_id
db.recoded[yt_song.id] = True
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
db.recoded[song.id] = audio.message_id
await db.occasionally_write()

16
bot/handlers/on_chosen/suppress_verify.py Normal file
View File

@@ -0,0 +1,16 @@
from aiogram import Router, Bot, F
from aiogram.types import (
ChosenInlineResult,
)
router = Router()
@router.chosen_inline_result(
F.result_id.startswith('deezc::')
)
async def on_unneeded_cached_chosen(chosen_result: ChosenInlineResult, bot: Bot):
await bot.edit_message_reply_markup(
inline_message_id=chosen_result.inline_message_id,
reply_markup=None
)

36
bot/handlers/on_chosen/youtube.py
View File

@@ -7,12 +7,14 @@ from aiogram.types import (
from bot.modules.youtube import youtube, AgeRestrictedError
from bot.utils.config import config
from bot.modules.database import db
from bot.modules.settings import UserSettings
router = Router()
@router.chosen_inline_result(F.result_id.startswith('yt::'))
async def on_new_chosen(chosen_result: ChosenInlineResult, bot: Bot):
async def on_new_chosen(chosen_result: ChosenInlineResult, bot: Bot,
settings: UserSettings):
song = youtube.songs.from_id(chosen_result.result_id.removeprefix('yt::'))
try:
@@ -46,4 +48,36 @@ async def on_new_chosen(chosen_result: ChosenInlineResult, bot: Bot):
reply_markup=None
)
if settings['recode_youtube'].value == 'yes':
await bot.edit_message_caption(
inline_message_id=chosen_result.inline_message_id,
caption='🔄 Recoding...',
reply_markup=None
)
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[song.id] = audio.audio.file_id
db.recoded[song.id] = True
await bot.edit_message_media(
inline_message_id=chosen_result.inline_message_id,
media=InputMediaAudio(media=audio.audio.file_id),
reply_markup=None
)
else:
db.recoded[song.id] = audio.message_id
await db.occasionally_write()