diff --git a/src/swingmusic/models/album.py b/src/swingmusic/models/album.py index f268226f..04015fc7 100644 --- a/src/swingmusic/models/album.py +++ b/src/swingmusic/models/album.py @@ -103,11 +103,17 @@ class Album: """ Checks if the album is a soundtrack. """ + title = self.og_title.lower() + keywords = ["motion picture", "soundtrack"] for keyword in keywords: - if keyword in self.og_title.lower(): + if keyword in title: return True + # if og_title ends with "the album" + if len(title) > 10 and title.endswith("the album"): + return True + return False def is_compilation(self) -> bool: