Initial commit

1 files changed, 9 insertions, 0 deletions

diff --git a/draw.py b/draw.py
new file mode 100644
index 0000000..3330343
--- /dev/null
+++ b/draw.py
@@ -0,0 +1,9 @@
+from statemachine.contrib.diagram import DotGraphMachine
+from desmo_api.fsm import JailStateMachine
+
+
+if __name__ == "__main__":
+    graph = DotGraphMachine(JailStateMachine)
+    graph.graph_rankdir = "TB"
+    dot = graph()
+    dot.write_png("fsm-desmo.png")