Sort out keyboard demos a bit

Author: dannystaple

.gitignore

@@ -5,3 +5,4 @@ __pycache__
 .idea
 *.dll
 *.egg-info
+.vscode

demos/key_ctrl.py demos/pg_key_ctrl.pydemos/key_ctrl.py renamed to demos/pg_key_ctrl.py

@@ -1,4 +1,4 @@
-"""key_ctrl - key based arm controller"""
+"""Pygame key_ctrl - key based arm controller"""
 from functools import partial
 import logging
 import sys

demos/sh_key_ctrl.py

@@ -0,0 +1,65 @@
+"""Console keyboard based arm controller.
+Requires readchar
+"""
+from functools import partial
+import logging
+import sys
+
+from readchar import readchar, key
+
+import owi_maplin_usb_arm as usb_arm
+
+
+KEYMAP =  {
+    'z': usb_arm.BaseClockWise,
+    'x': usb_arm.BaseCtrClockWise,
+    'r': usb_arm.CloseGrips,
+    'f': usb_arm.OpenGrips,
+    'a': usb_arm.ShoulderDown,
+    'q': usb_arm.ShoulderUp,
+    's': usb_arm.ElbowDown,
+    'w': usb_arm.ElbowUp,
+    'd': usb_arm.WristDown,
+    'e': usb_arm.WristUp,
+    'l': usb_arm.LedOn
+}
+
+def handle_key(arm, delay, pressed_key):
+    def do_it():
+        if pressed_key in KEYMAP:
+            message = KEYMAP[pressed_key]
+            print("Key ", pressed_key, "Movement message", message)
+
+            arm.move(message, delay)
+    arm.safe_tell(do_it)
+
+
+def key_loop():
+    try:
+        arm = usb_arm.Arm()
+    except AttributeError:
+        print("Please make sure the arm is connected and turned on")
+        sys.exit(1)
+    handle = partial(handle_key, arm, 0.5)
+    exit_key = key.ESC
+
+    while True:
+        pressed_key = readchar()
+        if pressed_key == exit_key:
+            return
+        else:
+            handle(pressed_key)
+
+def main():
+    logging.basicConfig()
+    usb_arm.logger.setLevel(logging.DEBUG)
+    print("Press z/x to turn the base motor")
+    print("Press a/q to move the shoulder up/down")
+    print("Press w/s to move the elbow up/down")
+    print("Press e/d to move the wrist up/down")
+    print("Press r/f to close/open the grips")
+    print("Press l to toggle the LED")
+    print("Press ESC to exit")
+    key_loop()
+
+main()