diff options
| author | Parker <contact@pkrm.dev> | 2024-12-15 22:44:38 -0600 |
|---|---|---|
| committer | Parker <contact@pkrm.dev> | 2024-12-15 22:44:38 -0600 |
| commit | 78ebd842755154c03c950ccbb8ec5b5033936ec6 (patch) | |
| tree | b12c99c84fe7c10bbd71b0a9ed6a2a7e28daf023 /code/utils/custom_sources.py | |
| parent | 2399336f6f0e01d69d535d96518d20845424d071 (diff) | |
Protect against `results` being `None`
Diffstat (limited to 'code/utils/custom_sources.py')
| -rw-r--r-- | code/utils/custom_sources.py | 8 |
1 files changed, 5 insertions, 3 deletions
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, ): |