Migrated links, lyrics and properties

scripts/01_migrate_lyrics.py

@@ -0,0 +1,48 @@
+import json
+from collections.abc import Iterable
+from typing import TypedDict
+
+from folkugat_web.dal.sql import get_connection
+from folkugat_web.dal.sql.temes import lyrics as lyrics_dal
+from folkugat_web.model.temes import Lyrics
+
+query = """ SELECT id, lyrics FROM temes """
+LyricRowTuple = tuple[int, str]
+
+
+class OldLyricRowDict(TypedDict):
+    title: str
+    content: str
+
+
+def _build_lyric(tema_id: int, lyric_dict: OldLyricRowDict) -> Lyrics:
+    return Lyrics(
+        id=None,
+        tema_id=tema_id,
+        title=lyric_dict["title"],
+        content=lyric_dict["content"],
+    )
+
+
+def _build_lyrics(row: LyricRowTuple) -> list[Lyrics]:
+    tema_id = row[0]
+    lyric_dicts: list[OldLyricRowDict] = json.loads(row[1])
+    return [_build_lyric(tema_id, lyric_dict) for lyric_dict in lyric_dicts]
+
+
+with get_connection() as con:
+    cur = con.cursor()
+    _ = cur.execute(query)
+    iter_rows: Iterable[LyricRowTuple] = cur.fetchall()
+    rows = list(map(_build_lyrics, iter_rows))
+
+    _ = cur.execute("DELETE FROM tema_lyrics")
+    for lyrics in rows:
+        for lyric in lyrics:
+            _ = lyrics_dal.insert_lyric(lyric=lyric, con=con)
+
+    drop_query = """ ALTER TABLE temes DROP COLUMN lyrics"""
+    _ = cur.execute(drop_query)
+
+
+print("DONE!")