init commit

server/light.py

@@ -0,0 +1,73 @@
+import RPi.GPIO as GPIO
+import time
+import database
+
+FUNCTIONS = ['green', 'orange', 'red']
+
+
+class Light:
+    def __init__(self, pin, name):
+        self.pin = pin
+        self.name = name
+        self.state = database.get_light_state(name)
+        self.setup()
+
+    def setup(self):
+        GPIO.setmode(GPIO.BCM)
+        GPIO.setup(self.pin, GPIO.OUT)
+        self._set(self.state)
+
+    def _set(self, state):
+        if state == self.state:
+            print('%s is equal %s' % (state, self.state,))
+
+        print('setting %s state to: %s' % (self.name, state,))
+        GPIO.output(self.pin, GPIO.LOW if state == GPIO.HIGH else GPIO.HIGH)
+        self.state = state
+        database.set_light_state(self.name, state)
+
+    def on(self):
+        self._set(GPIO.HIGH)
+
+    def off(self):
+        self._set(GPIO.LOW)
+
+    def flash(self, count=2, interval=0.2):
+        for _ in range(count):
+            self.on()
+            time.sleep(interval)
+            self.off()
+            time.sleep(interval)
+
+
+class Tower:
+    def __init__(self, pinList):
+        self.pins = pinList
+        self.setupLights()
+        self.green: Light
+        self.orange: Light
+        self.red: Light
+
+    def setupLights(self):
+        if len(FUNCTIONS) != len(self.pins):
+            raise ValueError("only %s pins provided, but %s available" %
+                             (len(self.pins), len(FUNCTIONS)))
+
+        for pin in self.pins:
+            light = Light(pin['pin'], pin['name'])
+            if light.name == 'green':
+                self.green = light
+            elif light.name == 'orange':
+                self.orange = light
+            elif light.name == 'red':
+                self.red = light
+
+    def on(self):
+        self.green.on()
+        self.orange.on()
+        self.red.on()
+
+    def off(self):
+        self.green.off()
+        self.orange.off()
+        self.red.off()