Added stats

folkugat_web/dal/sql/playlists/query.py

@@ -1,10 +1,11 @@
-from collections.abc import Iterator
+from collections.abc import Iterable, Iterator
 from typing import TypedDict
 
 from folkugat_web.dal.sql import Connection, get_connection
 from folkugat_web.dal.sql.sessions import conversion as sessions_conversion
 from folkugat_web.model import playlists as model
 from folkugat_web.model.sessions import Session
+from folkugat_web.utils import groupby
 
 from . import conversion
 
@@ -55,14 +56,22 @@ def get_playlist_entries(
         return map(conversion.row_to_playlist_entry, cur.fetchall())
 
 
-def get_tune_sessions(tema_ids: list[int], con: Connection | None = None) -> Iterator[Session]:
-    query = """
+GetTuneSessionsRow = tuple[int, int, str, str, str, str | None, str | None, bool]
+
+
+def get_tune_sessions(tema_ids: list[int], con: Connection | None = None) -> dict[int, list[Session]]:
+    placeholders = ", ".join(["?" for _ in tema_ids])
+    query = f"""
       SELECT p.tema_id, s.id, s.date, s.start_time, s.end_time, s.venue_name, s.venue_url, s.is_live
       FROM playlists p JOIN sessions s ON p.session_id = s.id
-      WHERE p.tema_id IN :tema_ids
+      WHERE p.tema_id IN ({placeholders})
     """
-    data = dict(tema_ids=tuple(tema_ids))
     with get_connection(con) as con:
         cur = con.cursor()
-        _ = cur.execute(query, data)
-        return map(sessions_conversion.row_to_session, cur.fetchall())
+        _ = cur.execute(query, tema_ids)
+        result_rows: Iterable[GetTuneSessionsRow] = cur.fetchall()
+        return dict(groupby(
+            result_rows,
+            key_fn=lambda row: row[0],
+            group_fn=lambda rows: list(sessions_conversion.row_to_session(row[1:]) for row in rows)
+        ))