Fix similar tracks function (#848)
authorJozef Kruszynski <60214390+jozefKruszynski@users.noreply.github.com>
Thu, 17 Aug 2023 12:47:14 +0000 (13:47 +0100)
committerGitHub <noreply@github.com>
Thu, 17 Aug 2023 12:47:14 +0000 (14:47 +0200)
* Fix similar tracks function

* Fix lint issues

---------

Co-authored-by: jkruszynski <jozef@krush.at>
music_assistant/server/providers/tidal/__init__.py
music_assistant/server/providers/tidal/helpers.py

index a3869c2b4ecd6969f83c33091f9f890e9a0df97a..177cf087bceb7a8637050b8942c5b4bb4e8e7427 100644 (file)
@@ -310,13 +310,14 @@ class TidalProvider(MusicProvider):
             )
             yield track
 
-    async def get_similar_tracks(self, prov_track_id: str, limit=25) -> list[Track]:
+    async def get_similar_tracks(self, prov_track_id: str, limit=25) -> list[Track]:  # noqa: ARG002
         """Get similar tracks for given track id."""
         tidal_session = await self._get_tidal_session()
         async with self._throttler:
             return [
                 await self._parse_track(track_obj=track)
-                for track in await get_similar_tracks(tidal_session, prov_track_id, limit)
+                # Re-add limit here after tidalapi supports it, and remove noqa above
+                for track in await get_similar_tracks(tidal_session, prov_track_id)
             ]
 
     async def library_add(self, prov_item_id: str, media_type: MediaType):
index 7d6993536ffc233fb35146798bca80a9571f2a2e..86d90e22244f76a7429a048c615ea2516677025e 100644 (file)
@@ -261,12 +261,13 @@ async def create_playlist(
     return await asyncio.to_thread(inner)
 
 
-async def get_similar_tracks(session: TidalSession, prov_track_id, limit: int) -> list[TidalTrack]:
+async def get_similar_tracks(session: TidalSession, prov_track_id) -> list[TidalTrack]:
     """Async wrapper around the tidal Track.get_similar_tracks function."""
 
     def inner() -> list[TidalTrack]:
         try:
-            return TidalTrack(session, prov_track_id).get_track_radio(limit)
+            # Re-add limit here after tidalapi supports it
+            return TidalTrack(session, prov_track_id).get_track_radio()
         except HTTPError as err:
             if err.response.status_code == 404:
                 raise MediaNotFoundError(f"Track {prov_track_id} not found") from err