radcontrold.py

#!/usr/bin/env python
import sys
import logging
from configparser import ConfigParser
from os.path import expanduser
from socket import gethostname
from time import sleep

from eq3bt import Thermostat, Mode
from bluepy.btle import BTLEException
from mqttwrapper import run_script

log = logging.getLogger("radcontrold")


def callback(topic, payload, config):
    log.debug("%s %s", topic, payload)
    room = topic.split("/")[2]

    mode = {
        b'0': Mode.Closed,
        b'1': Mode.Open,
    }.get(payload)
    if mode is None:
        log.warning("Ignoring invalid payload on %s", topic)
        return

    addresses = config['radiators'].get(room, "")
    if not addresses:
        # Control message is for a radiator we're not responsible for.
        log.debug("No EQ3 addresses in config for %s", room)
        return

    success = True
    for address in addresses.split(","):
        for attempt in range(10):
            try:
                Thermostat(address).mode = mode
                log.info("Set %s in %s to %s", address, room, mode)
                break
            except BTLEException:
                log.warning("Couldn't set mode %s for %s in %s", mode, address, room)
            sleep(1)
        else:
            success = False
    # Only post acknowledgment to MQTT topic if all thermostats were controlled.
    if success:
        return [
            ("{}/ack".format(topic), payload)
        ]


def main():
    formatter = "[%(asctime)s] %(name)s %(levelname)s - %(message)s"
    logging.basicConfig(level=logging.DEBUG, format=formatter)
    logging.getLogger('eq3bt').setLevel(logging.ERROR)

    hostname = gethostname().split(".")[0]
    config = ConfigParser()
    config.read(expanduser("~/.config/radcontrold/{}.ini".format(hostname)))
    if not config.has_section('radiators') or len(config['radiators']) == 0:
        log.warning("No config for {}, exiting.".format(hostname))
        sys.exit(0)

    run_script(callback, broker=config['mqtt']['broker'], topics=['control/radiator/+/active'], config=config)


if __name__ == '__main__':
    main()