nambrogi
/
pomice
mirror of https://github.com/cloudwithax/pomice.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added better query parsing to get_tracks() to make it easier to search up keywords rather than having the end user implementing it themself

This commit is contained in:
cloudwithax 2021-10-09 17:30:16 -04:00
parent cb46150ce2
commit a900f77424
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pomice/pool.py
View File

@ -40,6 +40,9 @@ SPOTIFY_URL_REGEX = re.compile(
DISCORD_MP3_URL_REGEX = re.compile( DISCORD_MP3_URL_REGEX = re.compile(
r"https?://cdn.discordapp.com/attachments/(?P<channel_id>[0-9]+)/(?P<message_id>[0-9]+)/(?P<file>[a-zA-Z0-9_.]+)+" r"https?://cdn.discordapp.com/attachments/(?P<channel_id>[0-9]+)/(?P<message_id>[0-9]+)/(?P<file>[a-zA-Z0-9_.]+)+"
) )
URL_REGEX = re.compile(
r'https?://(?:www\.)?.+'
)
class Node: class Node:
@ -265,6 +268,10 @@ class Node:
You can also pass in a discord.py Context object to get a You can also pass in a discord.py Context object to get a
Context object on any track you search. Context object on any track you search.
""" """
if not URL_REGEX.match(query):
query = f"{search_type}:{query}"
if spotify_url_check := SPOTIFY_URL_REGEX.match(query): if spotify_url_check := SPOTIFY_URL_REGEX.match(query):
if not self._spotify_client_id and not self._spotify_client_secret: if not self._spotify_client_id and not self._spotify_client_secret:
raise InvalidSpotifyClientAuthorization( raise InvalidSpotifyClientAuthorization(