Trying python entry points

setup.py

@@ -11,4 +11,11 @@
     package_dir={'wsp_control': 'src'},
     install_requires=['RPi.GPIO', 'redis', 'gspread', 'raven'],
     scripts=['src/runwsp.py'],
+    entry_points = {
+        'my_ep_group_id': [
+            'check = wsp_control.functions:check',
+            'toggle = wsp_control.functions:toggle',
+            'override = wsp_control.functions:override',
+        ]
+    },
 )

src/functions.py

@@ -0,0 +1,32 @@
+#! /usr/bin/python
+import time
+import RPi.GPIO as GPIO
+from wsp_control.config import PIN
+
+GPIO.setwarnings(False)
+
+GPIO.setmode(GPIO.BCM)
+GPIO.setup(PIN.PUMP, GPIO.OUT)
+GPIO.setup(PIN.FLOW, GPIO.IN)
+
+def check():
+    if GPIO.input(PIN.PUMP):
+        print 'Pump is on'
+    else:
+        print 'Pump is off'
+
+def toggle():
+    while True:
+        print 'Switching', GPIO.input(PIN.PUMP)
+        GPIO.output(PIN.PUMP, not GPIO.input(PIN.PUMP))
+        time.sleep(2)
+
+def override():
+    print 'Switching on'
+    GPIO.output(PIN.PUMP, GPIO.HIGH)
+
+    if GPIO.input(PIN.PUMP):
+        print 'Pump is on'
+    else:
+        print 'Pump is off'
+

src/models.py

@@ -4,7 +4,7 @@
 import RPi.GPIO as GPIO
 from datetime import datetime
 
-from config import PIN, LITERS_PER_REV, TEMP_ENDPOINT, PUMP_ENDPOINT, AUTH
+from wsp_control.config import PIN, LITERS_PER_REV, TEMP_ENDPOINT, PUMP_ENDPOINT, AUTH
 
 
 logger = logging.getLogger(__name__)

src/runwsp.py

@@ -4,9 +4,9 @@
 import RPi.GPIO as GPIO
 from datetime import datetime
 
-from config import (PIN, UPLIFT_THRESHOLD, TEMP_CHECK_INTERVAL, PROBE_IN,
+from wsp_control.config import (PIN, UPLIFT_THRESHOLD, TEMP_CHECK_INTERVAL, PROBE_IN,
                     PROBE_OUT, PROBE_AIR)
-from models import Pump, DataLog, FlowMeter, Thermometer
+from wsp_control.models import Pump, DataLog, FlowMeter, Thermometer
 
 GPIO.setwarnings(False)
 # Choose numbering scheme