parker/guava
parker/guava
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Expand conditional checks for results, tracks, and load_type #12

Merged
PacketParker merged 1 commits from dev into main 2024-12-18 17:43:10 -06:00
Conversation 0 Commits 1 Files Changed 3 +54 -18
3 changed files with 54 additions and 18 deletions

24
code/cogs/play.py

@ -63,9 +63,15 @@ class Play(commands.Cog):
dzsearch = f"dzsearch:{query}"
results = await player.node.get_tracks(dzsearch)
# If Deezer returned nothing
if not (results and results.tracks) or results.load_type in (
LoadType.EMPTY,
LoadType.ERROR,
if (
not results
or not results.tracks
or not results.load_type
or results.load_type
in (
LoadType.EMPTY,
LoadType.ERROR,
)
):
if YOUTUBE_SUPPORT:
ytmsearch = f"ytmsearch:{query}"
@ -84,9 +90,15 @@ class Play(commands.Cog):
results = await player.node.get_tracks(query)
# If there are no results found, set results/embed to None, handled further down
if not (results and results.tracks) or results.load_type in (
LoadType.EMPTY,
LoadType.ERROR,
if (
not results
or not results.tracks
or not results.load_type
or results.load_type
in (
LoadType.EMPTY,
LoadType.ERROR,
)
):
results, embed = None, None

24
code/utils/ai_recommendations.py

@ -62,16 +62,28 @@ async def add_song_recommendations(
ytsearch = f"ytsearch:{song} by {artist} audio"
results = await player.node.get_tracks(ytsearch)
if not (results and results.tracks) or results.load_type in (
LoadType.EMPTY,
LoadType.ERROR,
if (
not results
or not results.tracks
or not results.load_type
or results.load_type
in (
LoadType.EMPTY,
LoadType.ERROR,
)
):
dzsearch = f"dzsearch:{song}"
results = await player.node.get_tracks(dzsearch)
if not (results and results.tracks) or results.load_type in (
LoadType.EMPTY,
LoadType.ERROR,
if (
not results
or not results.tracks
or not results.load_type
or results.load_type
in (
LoadType.EMPTY,
LoadType.ERROR,
)
):
continue

24
code/utils/custom_sources.py

@ -30,17 +30,29 @@ class CustomAudioTrack(DeferredAudioTrack):
): # Load our 'actual' playback track using the metadata from this one.
dzsearch = f"dzsearch:{self.title} {self.author}"
results = await client.get_tracks(dzsearch)
if not (results and results.tracks) or results.load_type in (
LoadType.EMPTY,
LoadType.ERROR,
if (
not results
or not results.tracks
or not results.load_type
or results.load_type
in (
LoadType.EMPTY,
LoadType.ERROR,
)
):
if YOUTUBE_SUPPORT:
ytmsearch = f"ytmsearch:{self.title} {self.author}"
results = await client.get_tracks(ytmsearch)
if not (results and results.tracks) or results.load_type in (
LoadType.EMPTY,
LoadType.ERROR,
if (
not results
or not results.tracks
or not results.load_type
or results.load_type
in (
LoadType.EMPTY,
LoadType.ERROR,
)
):
ytsearch = f"ytsearch:{self.title} {self.author} audio"
results = await client.get_tracks(ytsearch)