diff --git a/prace/bakalarka/formatitko_commands.py b/prace/bakalarka/formatitko_commands.py index 26cb10b86c6186d7162528bab25dd6de86df36dc..a816c41e372edcb3775fff92a20c72716cc3bd01 100644 --- a/prace/bakalarka/formatitko_commands.py +++ b/prace/bakalarka/formatitko_commands.py @@ -1,6 +1,6 @@ import panflute as pf from formatitko.command_env import parse_string -from formatitko.elements import Slanted +from formatitko.elements import Slanted, FLineMarkup formatitko_commands={} def formatitko_command(f): @@ -45,6 +45,14 @@ def box(element, context, processor): 'lemma': "Lemma", 'def': "Definice", }[element.attributes["t"]] + color = { + 'fact': "red", + 'task': "orange", + 'algo': "brown", + 'theorem': "green", + 'lemma': "lightgreen", + 'def': "blue", + }.get(element.attributes["t"], "black") out = [] intro = [] if 'name' in element.attributes: @@ -56,12 +64,12 @@ def box(element, context, processor): for e in content: if not out: if isinstance(e, pf.Para): - out.append(pf.Para(intro, Slanted(*e.content))) + out.append(pf.Para(intro, *e.content)) else: raise NotImplemented() else: - out += make_emph(e, wrap_class=Slanted) - return out + out.append(e) + return [FLineMarkup(*out, color=color)] @formatitko_command def proof(element, context, processor):