electron simulator (#531)

1 files changed, 17 insertions, 0 deletions

diff --git a/lib/api/stitch_plan.py b/lib/api/stitch_plan.py
new file mode 100644
index 00000000..fd6bf9c9
--- /dev/null
+++ b/lib/api/stitch_plan.py
@@ -0,0 +1,17 @@
+from flask import Blueprint, g, jsonify
+
+from ..stitch_plan import patches_to_stitch_plan
+
+
+stitch_plan = Blueprint('stitch_plan', __name__)
+
+
+@stitch_plan.route('')
+def get_stitch_plan():
+    if not g.extension.get_elements():
+        return dict(colors=[], stitch_blocks=[], commands=[])
+
+    patches = g.extension.elements_to_patches(g.extension.elements)
+    stitch_plan = patches_to_stitch_plan(patches)
+
+    return jsonify(stitch_plan)