Patch decimal return type serialization

AudiusProject · Feb 14, 2024 · 8b0b180 · 8b0b180
1 parent 7029d77
commit 8b0b180
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/discovery-provider/src/queries/get_remixable_tracks.py b/packages/discovery-provider/src/queries/get_remixable_tracks.py
@@ -67,7 +67,8 @@ def get_remixable_tracks(args):
         tracks = []
         for result in results:
             track = result[0]
-            score = result[-1]
+            # Convert decimal to int
+            score = int(result[-1])
             track = helpers.model_to_dictionary(track)
             track["score"] = score
             tracks.append(track)

diff --git a/packages/discovery-provider/src/queries/get_top_playlists.py b/packages/discovery-provider/src/queries/get_top_playlists.py
@@ -138,7 +138,8 @@ def get_top_playlists_sql(kind: TopPlaylistKind, args: GetTopPlaylistsArgs):
             for result in playlist_results:
                 # The playlist is the portion of the query result before repost_count and score
                 playlist = result[0:-2]
-                score = result[-1]
+                # Convert decimal to int
+                score = int(result[-1])
 
                 # Convert the playlist row tuple into a dictionary keyed by column name
                 playlist = helpers.tuple_to_model_dictionary(playlist, Playlist)