diff --git a/streamrip/media.py b/streamrip/media.py index f9eba6a..9db8c35 100644 --- a/streamrip/media.py +++ b/streamrip/media.py @@ -1844,13 +1844,13 @@ class Playlist(Tracklist, Media): self.append(Track(self.client, id=track["id"])) else: for track in tracklist: - # TODO: This should be managed with .m3u files and alike. Arbitrary - # tracknumber tags might cause conflicts if the playlist files are - # inside of a library folder meta = TrackMetadata(track=track, source=self.client.source) - cover_url = get_cover_urls(track["album"], self.client.source)[ - kwargs.get("embed_cover_size", "large") - ] + cover_urls = get_cover_urls(track["album"], self.client.source) + cover_url = ( + cover_urls[kwargs.get("embed_cover_size", "large")] + if cover_urls is not None + else None + ) self.append( Track( diff --git a/streamrip/utils.py b/streamrip/utils.py index e9ccf90..6f34539 100644 --- a/streamrip/utils.py +++ b/streamrip/utils.py @@ -306,7 +306,7 @@ def get_container(quality: int, source: str) -> str: return "MP3" -def get_cover_urls(resp: dict, source: str) -> dict: +def get_cover_urls(resp: dict, source: str) -> Optional[dict]: """Parse a response dict containing cover info according to the source. :param resp: