-#!/usr/bin/python
+#!/usr/bin/python3
+from collections import deque
import getopt
+import os
import paho.mqtt.client as mqtt
import sys
import time
debug_mode = False
+log_name = '/run/burrow-auto'
+log_file = None
+indent = 0
def debug(msg):
if debug_mode:
- print msg
+ print(("\t" * indent) + msg, file=sys.stderr)
+ elif log_file is not None:
+ print(("\t" * indent) + msg, file=log_file)
+
+def debug_open():
+ if not debug_mode:
+ global log_file
+ log_file = open(log_name + '.new', 'w')
+
+def debug_close():
+ if debug_mode:
+ debug("=" * 80)
+ else:
+ global log_file
+ log_file.close()
+ log_file = None
+ os.rename(log_name + '.new', log_name)
+
+def diff(x, y):
+ if x is None or y is None:
+ return None
+ else:
+ return x - y
class State:
def __init__(self):
- self.attrs = {}
+ self.attrs = {}
+ self.hyst_state = {}
+ self.running_averages = {}
def update(self):
- self.now = time.time()
- tm = time.localtime(self.now)
- self.year = tm.tm_year
- self.month = tm.tm_mon
- self.day = tm.tm_mday
- self.hour = tm.tm_hour
- self.min = tm.tm_min
- self.wday = tm.tm_wday
+ self.now = time.time()
+ tm = time.localtime(self.now)
+ self.year = tm.tm_year
+ self.month = tm.tm_mon
+ self.day = tm.tm_mday
+ self.hour = tm.tm_hour
+ self.min = tm.tm_min
+ self.wday = tm.tm_wday
+ debug("[{}-{:02d}-{:02d} {:02d}:{:02d}:{:02d} wday={}]".format(
+ self.year, self.month, self.day,
+ self.hour, self.min, tm.tm_sec,
+ self.wday
+ ))
def get_sensor(self, key):
topic = "burrow/" + key
- if topic in self.attrs:
- s = self.attrs[topic].split(" ")
- if len(s) >= 2 and s[1] < self.now - 120:
- return None
- return float(s[0])
- else:
- return None
+ if topic in self.attrs:
+ s = self.attrs[topic].split(" ")
+ if len(s) >= 2 and int(s[1]) < self.now - 120:
+ debug("< {} EXPIRED".format(key))
+ return None
+ else:
+ debug("< {} = {}".format(key, s[0]))
+ return float(s[0])
+ else:
+ debug("< {} UNDEFINED".format(key))
+ return None
def set(self, key, val):
global mq
topic = "burrow/" + key
- debug("Setting {} to {}".format(topic, val))
- mq.publish(topic, val, qos=1, retain=True)
+ debug("> {} = {}".format(topic, val))
+ mq.publish(topic, val, qos=1, retain=True)
-st = State()
+ def send(self, key, val):
+ global mq
+ topic = "burrow/" + key
+ debug("> {} := {}".format(topic, val))
+ mq.publish(topic, val, qos=1, retain=False)
-def on_connect(mq, userdata, flags, rc):
- mq.subscribe("burrow/#")
+ def auto_enabled(self, key):
+ topic = "burrow/auto/" + key
+ if topic in self.attrs:
+ return self.attrs[topic] != '0'
+ else:
+ return True
+
+ def hysteresis(self, key, value, low, high):
+ old_state = self.hyst_state.get(key, 0)
+ if value is None:
+ new_state = 0
+ elif old_state <= 0:
+ if value >= high:
+ new_state = 1
+ else:
+ new_state = -1
+ else:
+ if value <= low:
+ new_state = -1
+ else:
+ new_state = 1
+ self.hyst_state[key] = new_state
+ return new_state
+
+ def update_average(self, key, window_seconds):
+ if key not in self.running_averages:
+ self.running_averages[key] = (deque(), 0, 0, None)
+ (history, sum, count, avg) = self.running_averages[key]
+
+ while len(history) > 0 and history[0][0] <= self.now - window_seconds:
+ if history[0][1] is not None:
+ sum -= history[0][1]
+ count -= 1
+ history.popleft()
+
+ curr = self.get_sensor(key)
+ history.append((self.now, curr))
+ if curr is not None:
+ sum += curr
+ count += 1
+
+ if count > len(history) // 2:
+ avg = sum / count
+ else:
+ avg = None
+
+ self.running_averages[key] = (history, sum, count, avg)
+ if avg is None:
+ debug("= avg NONE ({} samples, {} non-null)".format(len(history), count))
+ else:
+ debug("= avg {:.6} ({} samples, {} non-null)".format(avg, len(history), count))
+ self.set("avg/" + key, "{:.6} {}".format(avg, int(self.now)))
+
+ def get_sensor_avg(self, key):
+ val = self.running_averages[key][3]
+ if val is None:
+ debug("< {} = avg NONE".format(key))
+ else:
+ debug("< {} = avg {:.6}".format(key, val))
+ return val
+
+st = State()
-def on_message(mq, userdata, msg):
- global st
- debug("Message {}: {}".format(msg.topic, msg.payload))
- st.attrs[msg.topic] = msg.payload
def auto_loft_fan():
global st
- lt = st.get_sensor("temp/loft")
- if lt is not None and lt >= 30:
- fs = 3
- elif st.hour in range(10, 20):
- fs = 2
+ lt = st.get_sensor_avg("temp/loft")
+ out = st.get_sensor_avg('air/outside-intake')
+
+ if lt is None or out is None:
+ fs = 0
+ elif st.hysteresis('lf_out_cold', out, 5, 6) < 0:
+ fs = 0
+ elif st.hysteresis('lf_out_cool', out, 14, 15) < 0:
+ if st.min in range(10, 15):
+ fs = 2
+ else:
+ fs = 0
+ elif ac_is_on() > 0:
+ fs = 3
+ elif st.hysteresis('lf_loft_hot', lt, 25, 26) > 0:
+ if st.hysteresis('lf_loft_hotter_than_out', lt, out - 1, out + 1) > 0:
+ fs = 3
+ else:
+ fs = 1
else:
- fs = 1
+ if st.min in range(10, 15):
+ fs = 2
+ else:
+ fs = 0
+
st.set("loft/fan", fs)
+
def auto_circ():
global st
- if st.hour in range(19, 23):
+ if st.hour in range(7, 24):
c = 1
else:
c = 0;
st.set("loft/circulation", c)
+
+def ac_is_on():
+ # Heuristics to tell if AC is currently on
+
+ tie = st.get_sensor_avg('air/inside-exhaust')
+ tmix = st.get_sensor_avg('air/mixed')
+ if tie is None or tmix is None:
+ return 0
+
+ # Mixed air is significantly colder than inside exhaust
+ ac_on = -st.hysteresis('ac_off', tmix, tie - 5, tie - 4)
+
+ # FIXME: It might also mean that the loft is cold
+
+ return ac_on
+
+
+def auto_air():
+ global st
+ tii = st.get_sensor_avg('air/inside-intake')
+ tie = st.get_sensor_avg('air/inside-exhaust')
+ toi = st.get_sensor_avg('air/outside-intake')
+ tmix = st.get_sensor_avg('air/mixed')
+ house_warm = st.hysteresis('house_warm', tii, 23.5, 24.5)
+ house_hot = st.hysteresis('house_hot', tii, 24.5, 25)
+ ac_on = ac_is_on()
+
+ # XXX: Temporarily disabled
+ #house_warm = -1
+ #house_hot = -1
+ #ac_on = -1
+
+ # Do we want to bypass the heat exchanger?
+ outside_warmer = st.hysteresis('outside_warmer', diff(toi, tii), -0.5, 0.5)
+ if (house_warm > 0) and (outside_warmer > 0) or \
+ (house_warm < 0) and (outside_warmer < 0):
+ st.set('air/bypass', 0)
+ else:
+ st.set('air/bypass', 1)
+
+ # Is mixed air colder than air from the inside?
+ mixed_warmer = st.hysteresis('mixed_warmer', diff(tmix, tii), -1, 0)
+
+ # Do we want to boost heat exchanger fan?
+ if ac_on > 0 or (house_hot > 0 and mixed_warmer < 0):
+ st.set('air/exchanger-fan', 255)
+ else:
+ st.set('air/exchanger-fan', 0)
+
+ debug("Air: house_warm={} house_hot={} ac_on={} outside_warmer={} mixed_warmer={}".format(house_warm, house_hot, ac_on, outside_warmer, mixed_warmer))
+
+ if ac_on != 0:
+ st.set("air/ac-on", "{} {}".format(1 if ac_on > 0 else 0, int(st.now)))
+ else:
+ st.set("air/ac-on", "")
+
+
+def auto_aircon():
+ global st
+ tii = st.get_sensor_avg('air/inside-intake')
+ tie = st.get_sensor_avg('air/inside-exhaust')
+ house_hot = st.hysteresis('ac_house_hot', tii, 23.5, 24)
+ outside_hot = st.hysteresis('ac_outside_hot', tie, 23, 24)
+ ac_on = ac_is_on()
+
+ if house_hot > 0 and outside_hot > 0:
+ want_ac = 1
+ else:
+ want_ac = -1
+
+ if not hasattr(st, 'last_ac_change'):
+ st.last_ac_change = st.now
+ need_wait = st.last_ac_change + 300 - st.now # FIXME: Increase
+
+ if need_wait > 0:
+ action = f"wait({need_wait:.0f})"
+ elif ac_on == 0:
+ action = "need-data"
+ elif ac_on != want_ac:
+ action = "change"
+ st.send('air/aircon-remote', 'p')
+ st.last_ac_change = st.now
+ else:
+ action = "none"
+
+ debug("AC: house_hot={} outside_hot={} ac_on={} want_ac={} action={}".format(house_hot, outside_hot, ac_on, want_ac, action))
+
+
+def on_connect(mq, userdata, flags, rc):
+ mq.subscribe("burrow/#")
+
+def on_message(mq, userdata, msg):
+ global st
+ # debug("Message {}: {}".format(msg.topic, msg.payload))
+ st.attrs[msg.topic] = msg.payload.decode('utf-8')
+
opts, args = getopt.gnu_getopt(sys.argv[1:], "", ["debug"])
for opt in opts:
o, arg = opt
mq = mqtt.Client()
mq.on_connect = on_connect
mq.on_message = on_message
-mq.will_set("status/auto", "failed", retain=True)
-mq.connect("127.0.0.1")
+mq.will_set("status/auto", "dead", retain=True)
+mq.connect("burrow-mqtt")
mq.publish("status/auto", "ok", retain=True)
mq.loop_start()
+# Heuristic delay to get all attributes from MQTT
+time.sleep(3)
+
+checks = [
+ ('loft-fan', auto_loft_fan),
+ ('circ', auto_circ),
+ ('air', auto_air),
+ ('aircon', auto_aircon),
+]
+
while True:
+ debug_open()
st.update()
- auto_loft_fan()
- auto_circ()
+
+ debug("averages")
+ indent += 1
+ st.update_average('air/outside-intake', 180)
+ st.update_average('air/inside-intake', 180)
+ st.update_average('air/inside-exhaust', 180)
+ st.update_average('air/mixed', 180)
+ st.update_average('temp/loft', 180)
+ indent -= 1
+
+ for name, func in checks:
+ if st.auto_enabled(name):
+ debug(name)
+ indent += 1
+ func()
+ indent -= 1
+ else:
+ debug("{} DISABLED".format(name))
+
+ debug_close()
time.sleep(10)