wasp: Add battery support

1 files changed, 23 insertions, 0 deletions

diff --git a/wasp/drivers/signal.py b/wasp/drivers/signal.py
new file mode 100644
index 0000000..18ce8bf
--- /dev/null
+++ b/wasp/drivers/signal.py
@@ -0,0 +1,23 @@
+class Signal(object):
+    '''Simplified Signal class
+
+    Note: The normal C implementation isn't working for the NRF port
+    '''
+
+    def __init__(self, pin, invert=False):
+        self.pin = pin
+        self.invert = invert
+
+    def __call__(self, v=None):
+        return self.value(v)
+
+    def value(self, v=None):
+        if v == None:
+            return self.invert ^ self.pin.value()
+        self.pin.value(self.invert ^ bool(v))
+
+    def on(self):
+        self.value(1)
+
+    def off(self):
+        self.value(0)