From 319fdc62f2bb944c441cb39057ff5b60e829fe51 Mon Sep 17 00:00:00 2001
From: Parker <contact@pkrm.dev>
Date: Fri, 29 Nov 2024 22:25:17 -0600
Subject: [PATCH] Check for IndexError in case artworkUrl doesn't exist

---
 code/utils/custom_sources.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/code/utils/custom_sources.py b/code/utils/custom_sources.py
index 50f0228..ed00c66 100644
--- a/code/utils/custom_sources.py
+++ b/code/utils/custom_sources.py
@@ -126,6 +126,10 @@ class SpotifySource(Source):
         for track in metadata["tracks"][
             "items"
         ]:  # Loop through each track in the playlist.
+            try:
+                artwork_url = track["track"]["album"]["images"][0]["url"]
+            except IndexError:
+                artwork_url = None
             tracks.append(
                 CustomAudioTrack(
                     {  # Create an instance of our CustomAudioTrack.
@@ -139,9 +143,7 @@ class SpotifySource(Source):
                         "title": track["track"]["name"],
                         "uri": track["track"]["external_urls"]["spotify"],
                         "duration": track["track"]["duration_ms"],
-                        "artworkUrl": track["track"]["album"]["images"][0][
-                            "url"
-                        ],
+                        "artworkUrl": artwork_url,
                     },
                     requster=user,
                 )