From 2f088fab5df93239dabdaa465a32a8097ce8318d Mon Sep 17 00:00:00 2001 From: Parker Date: Wed, 24 Jul 2024 20:28:04 -0500 Subject: Switch up search providers --- code/utils/custom_sources.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) (limited to 'code/utils') diff --git a/code/utils/custom_sources.py b/code/utils/custom_sources.py index 96ae3a9..6f5d65a 100644 --- a/code/utils/custom_sources.py +++ b/code/utils/custom_sources.py @@ -18,20 +18,27 @@ class CustomAudioTrack(DeferredAudioTrack): async def load( self, client ): # Load our 'actual' playback track using the metadata from this one. - ytsearch = f"ytsearch:{self.title} {self.author} audio" - results = await client.get_tracks(ytsearch) + dzsearch = f"dzsearch:{self.title} {self.author}" + results = await client.get_tracks(dzsearch) if not results.tracks or results.load_type in ( LoadType.EMPTY, LoadType.ERROR, ): - dzsearch = f"dzsearch:{self.title} {self.author}" - results = await client.get_tracks(dzsearch) + scsearch = f"scsearch:{self.title} {self.author}" + results = await client.get_tracks(scsearch) if not results.tracks or results.load_type in ( LoadType.EMPTY, LoadType.ERROR, ): - raise LoadError + ytsearch = f"ytsearch:{self.title} {self.author} audio" + results = await client.get_tracks(ytsearch) + + if not results.tracks or results.load_type in ( + LoadType.EMPTY, + LoadType.ERROR, + ): + raise LoadError first_track = results.tracks[0] # Grab the first track from the results. base64 = first_track.track # Extract the base64 string from the track. -- cgit v1.2.3-70-g09d2