Add useful comments on the code

This commit is contained in:
stef-devs 2022-06-22 20:32:58 +02:00
parent 7ae0f1a348
commit b26dbe8678
1 changed files with 17 additions and 2 deletions

View File

@ -47,8 +47,9 @@ logging.basicConfig(
load_dotenv() load_dotenv()
API_TOKEN = os.getenv('BOT_TOKEN') API_TOKEN = os.getenv('BOT_TOKEN')
bot = telebot.TeleBot(API_TOKEN) bot = telebot.TeleBot(API_TOKEN)
#Start bot
logging.info("Il bot si è avviato con successo!") logging.info("Il bot si è avviato con successo!")
@ -72,7 +73,9 @@ def select_music(pm):
def music_step(pm): def music_step(pm):
if pm.text.startswith('https://www.youtube.com/') or pm.text.startswith('https://youtu.be/'): text = pm.text
#Check if the link is correct (Youtube and Spotify links only)
if text.startswith('https://www.youtube.com/') or text.startswith('https://youtu.be/'):
ytdl_opts = { ytdl_opts = {
'format': 'bestaudio/best', 'format': 'bestaudio/best',
'outtmpl': 'music_cache/%(title)s.%(ext)s', 'outtmpl': 'music_cache/%(title)s.%(ext)s',
@ -82,6 +85,18 @@ def music_step(pm):
'preferredquality': '192', 'preferredquality': '192',
}], }],
} }
with YoutubeDL(ytdl_opts) as ydl:
url = pm.text
info = ydl.extract_info(url, download=False)
name = info.get('title')
id = info.get('id')
ydl.download([id])
send_message = "🎶 La canzone <b>" + name + \
"</b> è stata scaricata con successo!"
bot.send_message(pm.chat.id, send_message, parse_mode="HTML")
bot.send_audio(pm.chat.id, audio=open("music_cache/" + name + '.mp3', 'rb'))
os.remove("music_cache/" + name + '.mp3')
#If the link for spotify ends with /album, /user, /playlist ecc.. send a error message
elif pm.text.startswith('https://open.spotify.com/track/'): elif pm.text.startswith('https://open.spotify.com/track/'):
ytdl_opts = { ytdl_opts = {
'format': 'bestaudio/best', 'format': 'bestaudio/best',