Track available hasattr instead of isinstance
authorJonathan Bangert <jonathan@bangert.dk>
Wed, 9 Aug 2023 14:51:05 +0000 (16:51 +0200)
committerJonathan Bangert <jonathan@bangert.dk>
Wed, 9 Aug 2023 14:51:05 +0000 (16:51 +0200)
music_assistant/server/providers/deezer/__init__.py

index 7094d83b7eb8e2552e940dfe6853f358d4439c3f..887ebf7ce58eee9b3b30990c53207b3e02fc5d68 100644 (file)
@@ -622,7 +622,7 @@ class DeezerProvider(MusicProvider):  # pylint: disable=W0223
                     item_id=str(track.id),
                     provider_domain=self.domain,
                     provider_instance=self.instance_id,
-                    available=self.track_available(trackuser_country),
+                    available=self.track_available(track=track, user_country=user_country),
                     url=track.link,
                     isrc=isrc,
                 )
@@ -701,11 +701,9 @@ class DeezerProvider(MusicProvider):  # pylint: disable=W0223
 
         raise NotImplementedError("Unsupported contenttype")
 
-    def track_available(
-        self, track: deezer.Track, user_country: str, track_type: Track | PlaylistTrack | AlbumTrack
-    ) -> bool:
+    def track_available(self, track: deezer.Track, user_country: str) -> bool:
         """Check if a given track is available in the users country."""
-        if isinstance(track_type, Track):
+        if hasattr(track, "available_countries"):
             return user_country in track.available_countries
         return True