From 78ebd842755154c03c950ccbb8ec5b5033936ec6 Mon Sep 17 00:00:00 2001 From: Parker Date: Sun, 15 Dec 2024 22:44:38 -0600 Subject: Protect against `results` being `None` --- code/utils/custom_sources.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'code/utils/custom_sources.py') diff --git a/code/utils/custom_sources.py b/code/utils/custom_sources.py index 5cf2295..063b886 100644 --- a/code/utils/custom_sources.py +++ b/code/utils/custom_sources.py @@ -30,7 +30,7 @@ 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.tracks or results.load_type in ( + if not (results and results.tracks) or results.load_type in ( LoadType.EMPTY, LoadType.ERROR, ): @@ -38,14 +38,16 @@ class CustomAudioTrack(DeferredAudioTrack): ytmsearch = f"ytmsearch:{self.title} {self.author}" results = await client.get_tracks(ytmsearch) - if not results.tracks or results.load_type in ( + if not (results and results.tracks) or results.load_type in ( LoadType.EMPTY, LoadType.ERROR, ): ytsearch = f"ytsearch:{self.title} {self.author} audio" results = await client.get_tracks(ytsearch) - if not results.tracks or results.load_type in ( + if not ( + results and results.tracks + ) or results.load_type in ( LoadType.EMPTY, LoadType.ERROR, ): -- cgit v1.2.3-70-g09d2