chore: generalize dot attributes

so that they're easier to extend
Hochfrequenz · Oct 29, 2024 · 68a0a21 · 68a0a21
1 parent d681046
commit 68a0a21
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/rebdhuhn/graphviz.py b/src/rebdhuhn/graphviz.py
@@ -160,7 +160,10 @@ def convert_graph_to_dot(ebd_graph: EbdGraph) -> str:
         f'<B><FONT POINT-SIZE="18">{ebd_graph.metadata.chapter}</FONT></B><BR/><BR/>'
         f'<B><FONT POINT-SIZE="16">{ebd_graph.metadata.sub_chapter}</FONT></B><BR/><BR/><BR/><BR/>'
     )
-    dot_code = "digraph D {\n" f'{ADD_INDENT}labelloc="t";\n{ADD_INDENT}label=<{header}>;\n'
+    dot_attributes: dict[str, str] = {f"'{ADD_INDENT}labelloc": '"t"', "label": f"<{header}>"}
+    dot_code = "digraph D {\n"
+    for dot_attr_key, dot_attr_value in dot_attributes.items():
+        dot_code += f"{dot_attr_key}={dot_attr_value};\n"
     assert len(nx_graph["Start"]) == 1, "Start node must have exactly one outgoing edge."
     dot_code += _convert_nodes_to_dot(ebd_graph, ADD_INDENT) + "\n\n"
     dot_code += "\n".join(_convert_edges_to_dot(ebd_graph, ADD_INDENT)) + "\n"