mirror of
https://github.com/RafaelSolVargas/Vulkan.git
synced 2025-10-29 16:57:23 +00:00
103 lines
3.4 KiB
Python
103 lines
3.4 KiB
Python
import re
|
|
from config import config
|
|
from yt_dlp import YoutubeDL
|
|
from yt_dlp.utils import ExtractorError, DownloadError
|
|
|
|
|
|
class Downloader():
|
|
"""Download music source from Youtube with a music name or Youtube URL"""
|
|
|
|
def __init__(self) -> None:
|
|
self.__YDL_OPTIONS = {'format': 'bestaudio/best',
|
|
'default_search': 'auto',
|
|
'playliststart': 0,
|
|
'extract_flat': True,
|
|
'playlistend': config.MAX_PLAYLIST_LENGTH,
|
|
'cookiefile': config.COOKIE_PATH
|
|
}
|
|
|
|
def download_one(self, music: str) -> list:
|
|
"""Download one music link from Youtube
|
|
|
|
Arg: Music url or music name to search
|
|
Return: List with the Youtube URL of the music
|
|
"""
|
|
if type(music) != str:
|
|
return
|
|
|
|
if self.__is_url(music): # If Url
|
|
info = self.__download_url(music, flat=True)
|
|
else: # If Title
|
|
info = self.__download_title(music)
|
|
|
|
return info
|
|
|
|
def download_many(self, music_list: list) -> list:
|
|
"""Download many music links from Youtube
|
|
|
|
Arg: List with names or music url to search
|
|
Return: List with the youtube URL of each music
|
|
"""
|
|
if type(music_list) != list:
|
|
return
|
|
|
|
musics_info = []
|
|
for music in music_list:
|
|
info = self.download_one(music)
|
|
musics_info.extend(info)
|
|
|
|
return musics_info
|
|
|
|
def download_full(self, link) -> dict:
|
|
"""Download the full music info with the video URL"""
|
|
info = self.__download_url(url=link, flat=False)
|
|
return info[0]
|
|
|
|
def __download_title(self, music_name: str) -> list:
|
|
"""Download and return a list with the music link in dict"""
|
|
with YoutubeDL(self.__YDL_OPTIONS) as ydl:
|
|
try:
|
|
result = ydl.extract_info(
|
|
f"ytsearch:{music_name}", download=False)
|
|
id = result['entries'][0]['id']
|
|
|
|
link = f"https://www.youtube.com/watch?v={id}"
|
|
return [link]
|
|
except Exception as e:
|
|
raise e
|
|
|
|
def __download_url(self, url: str, flat=True) -> list:
|
|
"""Download musics from Playlist URL or Music URL
|
|
|
|
Arg: URL from Youtube
|
|
Return: List of youtube links
|
|
"""
|
|
options = self.__YDL_OPTIONS
|
|
options['extract_flat'] = flat
|
|
with YoutubeDL(options) as ydl:
|
|
try:
|
|
result = ydl.extract_info(url, download=False)
|
|
|
|
musics_link = []
|
|
|
|
if result.get('entries'): # If got a dict of musics
|
|
for entry in result['entries']:
|
|
link = f"https://www.youtube.com/watch?v={entry['id']}"
|
|
musics_link.append(link)
|
|
else: # Or a single music
|
|
musics_link.append(result['original_url'])
|
|
|
|
return musics_link
|
|
except ExtractorError or DownloadError:
|
|
pass
|
|
|
|
def __is_url(self, string) -> bool:
|
|
"""Verify if a string is a url"""
|
|
regex = re.compile(
|
|
"http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+")
|
|
|
|
if re.search(regex, string):
|
|
return True
|
|
else:
|
|
return False
|