Make script more reliable

rpiserver/main.py

@@ -1,55 +1,117 @@
 import os
 import logging
+DEBUG = os.environ.get('DEBUG', False)
 logging.basicConfig(
         format='[%(asctime)s] %(levelname)s %(module)s/%(funcName)s - %(message)s',
-        level=logging.DEBUG if os.environ.get('DEBUG') else logging.INFO)
-#logging.getLogger('aiohttp').setLevel(logging.DEBUG if os.environ.get('DEBUG') else logging.WARNING)
+        level=logging.DEBUG if DEBUG else logging.INFO)
 
 import asyncio
 from statistics import mean
+from signal import *
+import time
 
 import w1therm
 import relays
 import data
+import requests
 
 HI_SETPOINT = 40.0
-LO_SETPOINT = 30.0
+LO_SETPOINT = HI_SETPOINT - 5.0
 
+PUMP_RELAY = relays.RELAY1
 FAN_RELAY = relays.RELAY3
 
-async def main():
-    relays.init_relays()
+def controller_message(message):
+    payload = dict(misc=message)
+    r = requests.post('https://tbot.tannercollin.com/message', data=payload, timeout=10)
+    if r.status_code == 200:
+        return True
+    else:
+        logging.exception('Unable to communicate with controller! Message: ' + message)
+        return False
 
+
+def set_fan_on():
+    relays.set_relay(FAN_RELAY, relays.RELAY_ON)
+
+def set_fan_off():
+    relays.set_relay(FAN_RELAY, relays.RELAY_OFF)
+
+def set_pump_on():
+    relays.set_relay(PUMP_RELAY, relays.RELAY_ON)
+
+
+async def get_temp():
+    temps = await w1therm.get_temperatures()
+    if len(temps) != len(data.PROBES):
+        raise
+
+    temperature_log = []
+    for id_, temp in temps.items():
+        temperature_log.append('{}: {} C'.format(data.PROBES[id_], temp))
+    logging.info('Temperature probe ' + ', '.join(temperature_log))
+
+    temperature_list = list(temps.values())
+    temperature_mean = mean(temperature_list)
+    return temperature_mean
+
+async def run_thermostat():
     while True:
         try:
-            temps = await w1therm.get_temperatures()
-            if len(temps) != len(data.PROBES):
-                raise
+            temperature_mean = await get_temp()
         except:
-            logging.error('Problem reading temperature probes! Setting fan on and sleeping.')
+            logging.error('Problem reading temperature probes! Turning fan on and sleeping 60s.')
             relays.set_relay(FAN_RELAY, relays.RELAY_ON)
             await asyncio.sleep(60)
             continue
 
-        temperature_log = []
-        for id_, temp in temps.items():
-            temperature_log.append('{}: {} C'.format(data.PROBES[id_], temp))
-        logging.info('Temperature probe ' + ', '.join(temperature_log))
-
-        temperature_list = list(temps.values())
-        temperature_mean = mean(temperature_list)
-
         if temperature_mean > HI_SETPOINT:
             logging.info('Turning fan on')
-            relays.set_relay(FAN_RELAY, relays.RELAY_ON)
+            set_fan_on()
         elif temperature_mean < LO_SETPOINT:
             logging.info('Turning fan off')
-            relays.set_relay(FAN_RELAY, relays.RELAY_OFF)
+            set_fan_off()
 
         await asyncio.sleep(10)
 
+def run_on_exit(*args):
+    logging.info('Cryptosoak exiting, turning pump and fan on.')
+    controller_message('Cryptosoak exiting.')
+    set_pump_on()
+    set_fan_on()
+    exit()
+
+async def feed_watchdog():
+    while True:
+        with open('/dev/watchdog', 'w') as wdt:
+            wdt.write('1')
+        await asyncio.sleep(5)
+
+def init():
+    relays.init_relays()
+    set_pump_on()
+
+    for sig in (SIGABRT, SIGILL, SIGINT, SIGSEGV, SIGTERM):
+        signal(sig, run_on_exit)
+    logging.info('Signals initialized')
+
+async def main():
+    controller_message('Cryptosoak booting up...')
+
+    logging.info('Initialization complete')
+
+    await run_thermostat()
+
 if __name__ == '__main__':
+    init()
+
+    if not DEBUG:
+        logging.info('Waiting 60 seconds to boot...')
+        time.sleep(60)
+
     loop = asyncio.get_event_loop()
-    loop.run_until_complete(main())
+    loop.run_until_complete(main()).add_done_callback(run_on_exit)
+    if not DEBUG:
+        loop.run_until_complete(feed_watchdog())
     loop.close()
 

rpiserver/relays.py

@@ -31,7 +31,7 @@ async def test():
     for r in RELAYS:
         set_relay(r, RELAY_ON)
         await asyncio.sleep(0.5)
-        #set_relay(r, RELAY_OFF)
+        set_relay(r, RELAY_OFF)
 
 if __name__ == '__main__':
     init_relays()