Initial commit

folkugat_web/services/temes/write.py

@@ -0,0 +1,62 @@
+import dataclasses
+
+from folkugat_web.dal.sql import get_connection
+from folkugat_web.dal.sql.temes import query as temes_q
+from folkugat_web.dal.sql.temes import write as temes_w
+from folkugat_web.model import temes as model
+
+
+def update_title(tema_id: int, title: str) -> model.Tema:
+    with get_connection() as con:
+        tema = temes_q.get_tema_by_id(tema_id=tema_id, con=con)
+        if tema is None:
+            raise ValueError(f"No tune found with tema_id = {tema_id}!")
+
+        new_tema = dataclasses.replace(tema, title=title)
+        new_tema.compute_ngrams()
+
+        temes_w.update_tema(tema=new_tema, con=con)
+
+    return new_tema
+
+
+def update_lyric(tema_id: int, lyric_idx: int, lyric: model.Lyrics) -> model.Tema:
+    with get_connection() as con:
+        tema = temes_q.get_tema_by_id(tema_id=tema_id, con=con)
+        if tema is None:
+            raise ValueError(f"No tune found with tema_id = {tema_id}!")
+        if lyric_idx > len(tema.lyrics):
+            raise ValueError(f'Lyric index out of bounds')
+
+        tema.lyrics[lyric_idx] = lyric
+        temes_w.update_tema(tema=tema, con=con)
+
+    return tema
+
+
+def delete_lyric(tema_id: int, lyric_idx: int) -> model.Tema:
+    with get_connection() as con:
+        tema = temes_q.get_tema_by_id(tema_id=tema_id, con=con)
+        if tema is None:
+            raise ValueError(f"No tune found with tema_id = {tema_id}!")
+        if lyric_idx > len(tema.lyrics):
+            raise ValueError(f'Lyric index out of bounds')
+
+        del tema.lyrics[lyric_idx]
+        temes_w.update_tema(tema=tema, con=con)
+
+    return tema
+
+
+def add_lyric(tema_id: int) -> model.Tema:
+    with get_connection() as con:
+        tema = temes_q.get_tema_by_id(tema_id=tema_id, con=con)
+        if tema is None:
+            raise ValueError(f"No tune found with tema_id = {tema_id}!")
+        tema.lyrics.append(model.Lyrics(
+            title=tema.title,
+            content="",
+        ))
+        temes_w.update_tema(tema=tema, con=con)
+
+    return tema