Merge pull request #2 from orionrobots/web_arm_2021

dannystaple · web-flow · commit 25060e5f6f9e · 2021-03-02T08:39:31.000Z
Add web control code for youtube tutorial
diff --git a/demos/web_arm/arm_hello.py b/demos/web_arm/arm_hello.py
@@ -0,0 +1,16 @@
+import usb_arm
+from flask import Flask, Response
+
+
+arm = usb_arm.Arm()
+app = Flask(__name__)
+
+@app.route('/')
+def flash():
+  def inner():
+      yield "starting</br>\n"
+      arm.move(usb_arm.LedOn)
+      yield "completed\n"
+  return Response(inner())
+
+app.run(host="0.0.0.0")
diff --git a/demos/web_arm/arm_server.py b/demos/web_arm/arm_server.py
@@ -0,0 +1,35 @@
+import usb_arm
+from flask import Flask, Response, request
+
+
+arm = usb_arm.Arm()
+app = Flask(__name__)
+
+movemap = {
+    "GripsClose": usb_arm.GripsClose,
+    "CloseGrips": usb_arm.CloseGrips,
+    "GripsOpen": usb_arm.GripsOpen,
+    "OpenGrips": usb_arm.OpenGrips,
+    "Stop": usb_arm.Stop,
+    "WristUp": usb_arm.WristUp,
+    "WristDown": usb_arm.WristDown,
+    "ElbowUp": usb_arm.ElbowUp,
+    "ElbowDown": usb_arm.ElbowDown,
+    "ShoulderUp": usb_arm.ShoulderUp,
+    "ShoulderDown": usb_arm.ShoulderDown,
+    "BaseClockWise": usb_arm.BaseClockWise,
+    "BaseCtrClockWise": usb_arm.BaseCtrClockWise,
+    "LedOn": usb_arm.LedOn
+}
+
+@app.route('/move', methods=['POST'])
+def move():
+    pattern = movemap[request.form['pattern']]
+
+    def inner():
+            yield "starting<br>\n"
+            arm.move(pattern)
+            yield "completed\n"
+    return Response(inner())
+
+app.run(host="0.0.0.0")
diff --git a/demos/web_arm/arm_server2.py b/demos/web_arm/arm_server2.py
@@ -0,0 +1,39 @@
+import usb_arm
+from flask import Flask, Response, request, render_template
+
+
+arm = usb_arm.Arm()
+app = Flask(__name__)
+
+movemap = {
+    "GripsClose": usb_arm.GripsClose,
+    "CloseGrips": usb_arm.CloseGrips,
+    "GripsOpen": usb_arm.GripsOpen,
+    "OpenGrips": usb_arm.OpenGrips,
+    "Stop": usb_arm.Stop,
+    "WristUp": usb_arm.WristUp,
+    "WristDown": usb_arm.WristDown,
+    "ElbowUp": usb_arm.ElbowUp,
+    "ElbowDown": usb_arm.ElbowDown,
+    "ShoulderUp": usb_arm.ShoulderUp,
+    "ShoulderDown": usb_arm.ShoulderDown,
+    "BaseClockWise": usb_arm.BaseClockWise,
+    "BaseCtrClockWise": usb_arm.BaseCtrClockWise,
+    "LedOn": usb_arm.LedOn
+}
+
+@app.route('/')
+def index():
+  return render_template('index.html', patterns=movemap)
+
+@app.route('/move', methods=['POST'])
+def move():
+    pattern = movemap[request.form['pattern']]
+
+    def inner():
+            yield "starting<br>\n"
+            arm.move(pattern)
+            yield "completed\n"
+    return Response(inner())
+
+app.run(host="0.0.0.0")
diff --git a/demos/web_arm/templates/index.html b/demos/web_arm/templates/index.html
@@ -0,0 +1,13 @@
+<html>
+<head>
+  <title>Robot Arm</title>
+  <script src="https://code.jquery.com/jquery-3.3.1.min.js"></script>
+</head>
+<body>
+  <ul>
+    {% for name in patterns %}
+    <li><a href="#" onclick="$.post('/move', {'pattern': '{{ name }}'});">{{ name }}</a></li>
+    {% endfor %}	
+  </ul>
+</body>
+</html>
