Refactor sourceName to use track_info object.

2024-03-11 13:50:07 -04:00 · 2024-03-11 13:50:07 -04:00 · 347a6e0b96
parent bb12e33584
commit 347a6e0b96
1 changed files with 2 additions and 7 deletions
--- a/pomice/pool.py
+++ b/pomice/pool.py
@ -546,18 +546,13 @@ class Node:
            query=f"encodedTrack={quote(identifier)}",
        )

-        if self._version.major >= 4:
-            track_info = data["info"]
-            track_type = data["info"]["sourceName"]
-        else:
-            track_info = data
-            track_type = data["sourceName"]
+        track_info = data["info"] if self._version.major >= 4 else data

        return Track(
            track_id=identifier,
            ctx=ctx,
            info=track_info,
-            track_type=TrackType(track_type),
+            track_type=TrackType(track_info["sourceName"]),
        )

    async def get_tracks(