1 files changed, 9 insertions, 13 deletions
diff --git a/pybot/drivers/hotplug/led.py b/pybot/drivers/hotplug/led.py
index 0703d54..fde6b3e 100644
--- a/pybot/drivers/hotplug/led.py
+++ b/pybot/drivers/hotplug/led.py
@@ -2,18 +2,14 @@
 RD_VERSION = 0x00
 TURN = 0x01
 
-f1 = {
-    'name': 'getVersion',
-    'call': RD_VERSION,
-    'params': 0,
-    'read': 3
-}
+def getVersion(dev):
+    dev.send([RD_VERSION])
+    raw = dev.read(3)
+    return raw[1] + raw[2] * 256
 
-f2 = {
-    'name': 'turn',
-    'call': TURN,
-    'params': 1,
-    'read': 1
-}
+def turn(dev, on):
+    msg = [TURN, on]
+    dev.send(msg)
+    raw = dev.read(1)
+    return raw[0]
 
-FUNCTIONS = [f1, f2]