Added correct printing of type nodes

2021-07-13 05:13:55 +02:00 · 2021-07-13 05:13:55 +02:00 · 1ba6aecdb9
commit 1ba6aecdb9
parent 128b3ec1e4
1 changed files with 4 additions and 1 deletions
--- a/state/devstate.py
+++ b/state/devstate.py
@ -28,8 +28,11 @@ class DevState(PyState):
            f.write("digraph main {\n")
            for n in sorted(self.nodes):
                if n in self.values:
+                    x = self.values[n]
+                    if isinstance(x, dict):
+                        x = f"{x.get('type')}"
                    f.write("\"a_%s\" [label=\"%s\"];\n" % (
-                        n, str(self.values[n]).replace('"', '\\"')))
+                        n, str(x).replace('"', '\\"')))
                else:
                    f.write("\"a_%s\" [label=\"\"];\n" % n)
            for i, e in sorted(list(self.edges.items())):