From: Jonathan Bangert Date: Wed, 9 Aug 2023 14:51:05 +0000 (+0200) Subject: Track available hasattr instead of isinstance X-Git-Url: https://git.kitaultman.com/?a=commitdiff_plain;h=5adc79905d9c07429eca1ebe194a9c9625bddb82;p=music-assistant-server.git Track available hasattr instead of isinstance --- diff --git a/music_assistant/server/providers/deezer/__init__.py b/music_assistant/server/providers/deezer/__init__.py index 7094d83b..887ebf7c 100644 --- a/music_assistant/server/providers/deezer/__init__.py +++ b/music_assistant/server/providers/deezer/__init__.py @@ -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(track, user_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