Do some more basic housekeeping to get to a building working version 0.0.1

library/grow/pump.py

@@ -0,0 +1,78 @@
+import time
+import atexit
+import RPi.GPIO as GPIO
+
+PUMP_1_PIN = 17
+PUMP_2_PIN = 27
+PUMP_3_PIN = 22
+
+_is_setup = False
+
+
+class Pump(object):
+    __slots__ = 'out1', 'out2', 'out3'
+
+    def __init__(self, out1, out2, out3):
+        self.out1 = out1
+        self.out2 = out2
+        self.out3 = out3
+
+    def __repr__(self):
+        fmt = """Pump 1: {out1}
+Pump 1: {out2} Ohms
+Pump 1: {out3} Ohms"""
+        
+        return fmt.format(
+            out1=self.out1,
+            out2=self.out2,
+            out3=self.out3)
+
+    __str__ = __repr__
+
+
+def setup():
+    global _is_setup
+    global _pump
+    
+    if _is_setup:
+        return
+    _is_setup = True
+
+    GPIO.setwarnings(False)
+    GPIO.setmode(GPIO.BCM)
+    GPIO.setup(PUMP_1_PIN, GPIO.OUT)
+    GPIO.input(PUMP_1_PIN, 0)
+    GPIO.setup(PUMP_2_PIN, GPIO.OUT)
+    GPIO.input(PUMP_2_PIN, 0)
+    GPIO.setup(PUMP_3_PIN, GPIO.OUT)
+    GPIO.input(PUMP_3_PIN, 0)
+    
+    atexit.register(cleanup)
+
+
+def set_pump_on(channel):
+    """Set wet point for a moisture channel."""
+    _pump[channel] = 1
+
+def set_pump_off(channel, value):
+    """Set wet point for a moisture channel."""
+    _pump[channel] = 0
+
+def read_pump(channel):
+    """Get current value for the additional ADC pin."""
+    setup()
+    return _pump[channel]
+
+def cleanup():
+    GPIO.output(PUMP_1_PIN, 0)
+    GPIO.output(PUMP_2_PIN, 0)
+    GPIO.output(PUMP_3_PIN, 0)
+
+def read_all():
+    """Return pump state"""
+    setup()
+    in1 = _pump[1]
+    in2 = _pump[2]
+    in3 = _pump[3]
+
+    return _pump(in1, in2, in3)