From 347a6e0b964223fe32d537c1dc7ea474f28651ca Mon Sep 17 00:00:00 2001 From: "Zander M." Date: Mon, 11 Mar 2024 13:50:07 -0400 Subject: [PATCH] Refactor sourceName to use track_info object. --- pomice/pool.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/pomice/pool.py b/pomice/pool.py index 1446444..74f479d 100644 --- 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(