Afegir temes coocurrents
This commit is contained in:
@@ -3,7 +3,9 @@ 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.dal.sql.temes.conversion import row_to_tema
|
||||
from folkugat_web.model import playlists as model
|
||||
from folkugat_web.model import temes as temes_model
|
||||
from folkugat_web.model.sessions import Session
|
||||
from folkugat_web.utils import groupby
|
||||
|
||||
@@ -76,3 +78,36 @@ def get_tune_sessions(tema_ids: list[int], con: Connection | None = None) -> dic
|
||||
key_fn=lambda row: row[0],
|
||||
group_fn=lambda rows: list(sessions_conversion.row_to_session(row[1:]) for row in rows)
|
||||
))
|
||||
|
||||
|
||||
CommonlyPlayedTuneRow = tuple[int, str, str, str, str, int, int]
|
||||
|
||||
|
||||
def get_commonly_played_tunes(
|
||||
tema_id: int,
|
||||
con: Connection | None = None,
|
||||
) -> list[temes_model.CommonlyPlayedTema]:
|
||||
query = """
|
||||
SELECT
|
||||
id, title, alternatives, creation_date, modification_date, hidden, count
|
||||
FROM (
|
||||
SELECT tema_id, count(*) count FROM playlists p JOIN (
|
||||
SELECT session_id, set_id
|
||||
FROM playlists
|
||||
WHERE tema_id = ?
|
||||
) s
|
||||
ON p.session_id == s.session_id AND p.set_id == s.set_id
|
||||
WHERE tema_id != ?
|
||||
GROUP BY tema_id
|
||||
) common JOIN temes t ON common.tema_id == t.id
|
||||
"""
|
||||
with get_connection(con) as con:
|
||||
cur = con.cursor()
|
||||
_ = cur.execute(query, [tema_id, tema_id])
|
||||
result_rows: Iterable[CommonlyPlayedTuneRow] = cur.fetchall()
|
||||
return [
|
||||
temes_model.CommonlyPlayedTema(
|
||||
tema=row_to_tema(row[:6]),
|
||||
count=row[6],
|
||||
) for row in result_rows
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user