Make script more reliable

rpiserver/main.py

@@ -1,23 +1,46 @@
 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 = HI_SETPOINT - 10.0
+LO_SETPOINT = HI_SETPOINT - 5.0
 
 PUMP_RELAY = relays.RELAY1
 FAN_RELAY = relays.RELAY3
 
+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):
@@ -32,11 +55,7 @@ async def get_temp():
     temperature_mean = mean(temperature_list)
     return temperature_mean
 
-
-async def main():
-    relays.init_relays()
-    relays.set_relay(PUMP_RELAY, relays.RELAY_ON)
-
+async def run_thermostat():
     while True:
         try:
             temperature_mean = await get_temp()
@@ -48,15 +67,51 @@ async def main():
 
         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()