escape double quotes in graphviz labels

Signed-off-by: Nathanael DEMACON <quantumsheep@users.noreply.github.com>
caesar0301 · Dec 28, 2024 · 3167a02 · 3167a02
1 parent 7e1daf1
commit 3167a02
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/treelib/tree.py b/treelib/tree.py
@@ -1097,7 +1097,8 @@ def to_graphviz(
                 sorting=sorting,
             ):
                 nid = self[n].identifier
-                state = '"{0}" [label="{1}", shape={2}]'.format(nid, self[n].tag, shape)
+                tag = str(self[n].tag).translate(str.maketrans({'"': r"\""}))
+                state = '"{0}" [label="{1}", shape={2}]'.format(nid, tag, shape)
                 nodes.append(state)
 
                 for c in self.children(nid):