diff --git a/showtify.py b/showtify.py index 0e58263..ac1f2a0 100755 --- a/showtify.py +++ b/showtify.py @@ -143,15 +143,16 @@ def album_title(page_content): METADATA_CLASS_FOR_URL = [ - ("https://open.spotify.com/track/", TrackMetadata), - ("https://open.spotify.com/album/", AlbumMetadata), - ("https://open.spotify.com/playlist/", AlbumMetadata) + (r"https://open.spotify.com/track/.*", TrackMetadata), + (r"https://open.spotify.com/album/.*", AlbumMetadata), + (r"https://open.spotify.com/playlist/.*", AlbumMetadata), + (r"https://open.spotify.com/user/.*/playlist/.*", AlbumMetadata), ] def metadata_for_url(url): - for url_start, cls in METADATA_CLASS_FOR_URL: - if url.startswith(url_start): + for url_regex, cls in METADATA_CLASS_FOR_URL: + if re.fullmatch(url_regex, url): return cls.from_url(url) return None