fix: return playlist count as dictionary

metabrainz · Sep 16, 2024 · 2e1eb3b · 2e1eb3b
1 parent f59367a
commit 2e1eb3b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/listenbrainz/db/playlist.py b/listenbrainz/db/playlist.py
@@ -985,12 +985,12 @@ def get_playlist_recordings_metadata(mb_curs, ts_curs, playlist: Playlist) -> Pl
     return playlist
 
 
-def get_playlist_count(ts_conn, creator_ids: List[str]):
+def get_playlist_count(ts_conn, creator_ids: List[str]) -> dict:
     query = text("""
         SELECT creator_id, COUNT(*) as count
           FROM playlist.playlist
          WHERE creator_id IN :creator_ids
       GROUP BY creator_id
     """)
     result = ts_conn.execute(query, {"creator_ids": tuple(creator_ids)})
-    return result.fetchall()
+    return {row[0]: row[1] for row in result.fetchall()}