mirror of
https://github.com/RafaelSolVargas/Vulkan.git
synced 2025-10-29 16:57:23 +00:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from vulkanbot.music.Types import Provider
|
|
from vulkanbot.music.Spotify import SpotifySearch
|
|
from vulkanbot.music.utils import is_url
|
|
|
|
|
|
class Searcher():
|
|
"""Turn the user input into list of musics names, support youtube and spotify"""
|
|
|
|
def __init__(self) -> None:
|
|
self.__Spotify = SpotifySearch()
|
|
print(f'Spotify Connected: {self.__Spotify.connect()}')
|
|
|
|
def search(self, music: str) -> list:
|
|
"""Return a list with the song names or an URL
|
|
|
|
Arg -> User Input, a string with the
|
|
Return -> A list of musics names and Provider Type
|
|
"""
|
|
url_type = self.__identify_source(music)
|
|
|
|
if url_type == Provider.YouTube:
|
|
return [music], Provider.YouTube
|
|
|
|
elif url_type == Provider.Spotify:
|
|
musics = self.__Spotify.search(music)
|
|
return musics, Provider.Name
|
|
|
|
elif url_type == Provider.Name:
|
|
return [music], Provider.Name
|
|
|
|
elif url_type == Provider.Unknown:
|
|
return None, Provider.Unknown
|
|
|
|
def __identify_source(self, music) -> Provider:
|
|
"""Identify the provider of a music"""
|
|
if not is_url(music):
|
|
return Provider.Name
|
|
|
|
if "https://www.youtu" in music or "https://youtu.be" in music:
|
|
return Provider.YouTube
|
|
|
|
if "https://open.spotify.com" in music:
|
|
return Provider.Spotify
|
|
|
|
# If no match
|
|
return Provider.Unknown
|
|
|