weyne/Vulkan
1
0
Fork 0
mirror of https://github.com/RafaelSolVargas/Vulkan.git synced 2025-10-29 16:57:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Chaging the downloader logic to work, keeping the same objective

Browse Source
This commit is contained in:
Rafael Vargas 2021-12-28 19:13:16 -04:00
parent 9602666bc8
commit c07f01d51a
1 changed files with 77 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

107
vulkanbot/music/Downloader.py
View File

@ -3,9 +3,11 @@ from config import config
from yt_dlp import YoutubeDL
from yt_dlp.utils import ExtractorError, DownloadError
from vulkanbot.music.Types import Provider
class Downloader():
"""Download musics direct URL or Source from Youtube using a music name or Youtube URL"""
"""Download musics direct URL and title or Source from Youtube using a music name or Youtube URL"""
def __init__(self) -> None:
self.__YDL_OPTIONS = {'format': 'bestaudio/best',
@ -15,24 +17,30 @@ class Downloader():
'playlistend': config.MAX_PLAYLIST_LENGTH,
}
def download_urls(self, musics_input) -> list:
def download_urls(self, musics_input, provider: Provider) -> list:
"""Download the musics direct URL from Youtube and return in a list
Arg: List with names or youtube url or a Unique String
Return: List with the direct youtube URL of each music
"""
if type(provider) != Provider:
return None
if type(musics_input) != list and type(musics_input) != str:
return
return None
if type(musics_input) == str: # Turn the string in a list
musics_input = [musics_input]
if provider == Provider.Name: # Send a list of names
musics_urls = self.__download_titles(musics_input)
print(musics_urls)
return musics_urls
musics_urls = []
for music in musics_input:
url = self.__download_one(music)
musics_urls.extend(url)
return musics_urls
elif provider == Provider.YouTube: # Send a URL or Title
print(musics_input)
url = self.__download_one(musics_input)
return url
else:
print('Erro no download')
return None
def download_source(self, url) -> dict:
"""Download musics full info and source from Music URL
@ -46,9 +54,12 @@ class Downloader():
try:
result = ydl.extract_info(url, download=False)
print('Resultado: ')
print(len(result))
return result
except ExtractorError or DownloadError:
pass
except (ExtractorError, DownloadError) as e: # Any type of error in download
print(e)
return None
def __download_one(self, music: str) -> list:
"""Download one music/playlist direct link from Youtube
@ -60,26 +71,36 @@ class Downloader():
return
if self.__is_url(music): # If Url
info = self.__download_links(music)
info = self.__download_links(music) # List of dict
else: # If Title
info = self.__download_title(music)
info = self.__download_titles(music) # List of dict
return info
def __download_title(self, music_name: str) -> list:
def __download_titles(self, musics_names: list) -> list:
"""Download a music direct URL using his name.
Arg: Music Name
Return: List with one item, the music direct URL
Return: List with one dict, containing the music direct URL and title
"""
if type(musics_names) == str: # Turn str into list
musics_names = [musics_names]
musics_info = []
with YoutubeDL(self.__YDL_OPTIONS) as ydl:
try:
search = f"ytsearch:{music_name}"
result = ydl.extract_info(search, download=False)
id = result['entries'][0]['id']
for name in musics_names:
search = f"ytsearch:{name}"
result = ydl.extract_info(search, download=False)
link = f"https://www.youtube.com/watch?v={id}"
return [link]
id = result['entries'][0]['id']
music_info = {
'url': f"https://www.youtube.com/watch?v={id}",
'title': result['entries'][0]['title']
}
musics_info.append(music_info)
return musics_info # Return a list
except Exception as e:
raise e
@ -87,24 +108,31 @@ class Downloader():
"""Download musics direct links from Playlist URL or Music URL
Arg_Url: URL from Youtube
Return: List of youtube information of each music
Return: List of dicts, with the title and url of each music
"""
options = self.__YDL_OPTIONS
options['extract_flat'] = True
with YoutubeDL(options) as ydl:
try:
result = ydl.extract_info(url, download=False)
musics_link = []
musics_info = []
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'])
music_info = {
'title': entry['title'],
'url': f"https://www.youtube.com/watch?v={entry['id']}"
}
return musics_link
musics_info.append(music_info)
else: # Or a single music
music_info = {
'url': result['original_url'],
'title': result['title']
}
musics_info.append(music_info)
return musics_info # Return a list
except ExtractorError or DownloadError:
pass
@ -117,3 +145,22 @@ class Downloader():
return True
else:
return False
"""
# url = 'https://open.spotify.com/playlist/64wCcaIp6DxNmMaDM8X0W3'
# url = 'https://open.spotify.com/album/6wEkHIUHNb1kZiV2nCnVoh'
# url = 'https://open.spotify.com/track/7wpnz7hje4FbnjZuWQtJHP'
# url = 'https://www.youtube.com/watch?v=FLQtLH33ZHM&ab_channel=Acoustic%26CoverTv'
# url = 'https://www.youtube.com/playlist?list=PLbbKJHHZR9ShYuKAr71cLJCFbYE-83vhS'
# url = 'https://www.youtube.com/watch?v=pFoc5XKkIIw&list=RDpFoc5XKkIIw&start_radio=1&ab_channel=heldenhaftig' # Custom
url = 'Bury The Light'
search = Searcher()
musicas, provider = search.search(url)
print(musicas, provider)
down = Downloader()
result = down.download_urls(musicas, provider)
#result = down.download_urls('Cheiro de tame impala', provider)
print(result)
"""