101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
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_id: 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}!")
|
|
|
|
tema.lyrics.replace(lyric_id, lyric)
|
|
temes_w.update_tema(tema=tema, con=con)
|
|
|
|
return tema
|
|
|
|
|
|
def delete_lyric(tema_id: int, lyric_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.delete(lyric_id)
|
|
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(
|
|
id=None,
|
|
title=tema.title,
|
|
content="",
|
|
))
|
|
temes_w.update_tema(tema=tema, con=con)
|
|
|
|
return tema
|
|
|
|
|
|
def update_link(tema_id: int, link_id: int, link: model.Link) -> 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.links.replace(link_id, link)
|
|
temes_w.update_tema(tema=tema, con=con)
|
|
|
|
return tema
|
|
|
|
|
|
def delete_link(tema_id: int, link_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.links.delete(link_id)
|
|
temes_w.update_tema(tema=tema, con=con)
|
|
|
|
return tema
|
|
|
|
|
|
def add_link(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.links.append(model.Link(
|
|
id=None,
|
|
title="",
|
|
url="",
|
|
content_type=model.ContentType.OTHER,
|
|
link_type=None,
|
|
))
|
|
temes_w.update_tema(tema=tema, con=con)
|
|
|
|
return tema
|