Improved lyrics rendering

folkugat_web/services/lilypond.py

@@ -1,17 +1,17 @@
 import asyncio
 import dataclasses
-import itertools
 import os
 import re
 from pathlib import Path
 from typing import Literal
 
 import aiofiles
-from folkugat_web.model.lilypond import (LyricsColumn, LyricsParagraph,
+from folkugat_web.model.lilypond import (LyricsLine, LyricsParagraph,
                                          LyricsText, RenderError)
 from folkugat_web.model.temes import Tema
 from folkugat_web.services import files as files_service
 from folkugat_web.templates import templates
+from folkugat_web.utils import batched
 
 SCORE_BEGINNING = "% --- SCORE BEGINNING --- %"
 
@@ -23,6 +23,9 @@ def source_to_single_score(source: str, tema: Tema) -> str:
         lyrics_text = build_lyrics(tema.lyrics[0].content)
     else:
         lyrics_text = None
+    print("AAAAAA")
+    print(lyrics_text)
+    print("AAAAAA")
     return templates.get_template("lilypond/single_score.ly").render(
         score_beginning=SCORE_BEGINNING,
         score_source=source,
@@ -77,13 +80,10 @@ async def render(source: str, fmt: RenderFormat, output_filename: Path | None =
     return output_filename, []
 
 
-def build_lyrics(text: str, max_cols: int = 2, min_pars: int = 2) -> LyricsText:
-    paragraphs = [LyricsParagraph(lines=par_str.splitlines()) for par_str in text.split("\n\n")]
-    n_cols = next(filter(lambda nc: len(paragraphs) // nc >= min_pars, range(max_cols, 0, -1)), 1)
-    n_long_cols = len(paragraphs) % n_cols
-    pars_per_col = [len(paragraphs) // n_cols + 1] * n_long_cols + [len(paragraphs) // n_cols] * (n_cols - n_long_cols)
-    acc_pars_per_col = [0] + list(itertools.accumulate(pars_per_col))
-    return LyricsText(columns=[
-        LyricsColumn(paragraphs=paragraphs[acc_pars_per_col[i]:acc_pars_per_col[i+1]])
-        for i in range(n_cols)
+def build_lyrics(text: str, max_cols: int = 3) -> LyricsText:
+    paragraphs = [LyricsParagraph(lines=par_str.splitlines()) for par_str in text.split("\n\n") if par_str]
+    print(list(batched(paragraphs, max_cols)))
+    return LyricsText(lines=[
+        LyricsLine(paragraphs=list(line_pars))
+        for line_pars in batched(paragraphs, max_cols)
     ])

folkugat_web/services/temes/lyrics.py

@@ -17,9 +17,11 @@ def _sub(pattern: str, sub: str) -> Callable[[str], str]:
 def _clean_string(lyrics_str: str) -> str:
     return (
         FnChain.transform(lyrics_str) |
-        _sub(r" *", " ") |
-        _sub(r"\s*\n\s*", "\n") |
-        _sub(r"\n*", "\n")
+        _sub(r"\t", " ") |
+        _sub(r" +", " ") |
+        _sub(r" \n", "\n") |
+        _sub(r"\n ", "\n") |
+        _sub(r"\n\n+", "\n\n")
     ).result()
 
 
@@ -37,9 +39,10 @@ def get_lyric_by_id(lyric_id: int, tema_id: int | None = None) -> model.Lyrics |
     return next(iter(lyrics_dal.get_lyrics(lyric_id=lyric_id, tema_id=tema_id)), None)
 
 
-def update_lyric(lyric: model.Lyrics):
-    lyric = dataclasses.replace(lyric, content=_clean_string(lyric.content))
+def update_lyric(lyric: model.Lyrics, title: str, content: str):
+    lyric = dataclasses.replace(lyric, title=title.strip(), content=_clean_string(content))
     lyrics_dal.update_lyric(lyric=lyric)
+    return lyric
 
 
 def delete_lyric(lyric_id: int, tema_id: int | None = None):