airzone_mqtt.py

#!/usr/bin/env python3
import argparse

import airzone
import json

import paho.mqtt.client as mqtt
import time
import logging
import os, sys

AZ_TOPIC_STATUS = "airzone/status"
AZ_TOPIC_ACTIONS = "airzone/actions"


def action(args):
    extra_args = {"use_rtu_framer": args.rtuframer}

    if 'log' in args:
        logging.basicConfig(level=logging.DEBUG, filename=args.log, format="%(asctime)s %(message)s", filemode="w")
    else:
        logging.basicConfig(stream=sys.stdout, level=logging.DEBUG, format="%(asctime)s %(message)s")

    logger = logging.getLogger()

    logger.warning("successfully started the script")

    az = airzone.airzone_factory(args.address, args.port, args.machine, args.system, **extra_args)

    zones = { z.name : z for z in az.zones }
    logger.warning("Airzone zones : {}".format(zones.keys()))

    def on_connect(client, userdata, flags, rc):
        client.subscribe("{}/#".format(AZ_TOPIC_ACTIONS))

    def on_message(client, userdata, message):
        topic = message.topic
        msg = message.payload
        aTopic = topic.replace('{}/'.format(AZ_TOPIC_ACTIONS), '')
        if aTopic == "mode":
            az.operation_mode = int(msg)
            send_new_states()
            return
        zone, action = aTopic.split('/')
        if zone not in zones.keys():
            logger.error('zone "{}" unknown'.format(zone))
            return
        if action == "status":
            logger.info('Zone "{}", status to {}'.format(zone, int(msg)))
            if int(msg):
                zones[zone].turn_on()
                return
            else:
                zones[zone].turn_off()
                return
        if action == "temp":
            logger.info('Zone "{}", setting temp point to {}'.format(zone, float(msg)))
            zones[zone].signal_temperature_value = float(msg)
            return

    def send_new_states():
        az.retrieve_machine_state()
        for z in zones:
            zones[z].retrieve_zone_state()
        jsonStatus = az.toJSON()
        status = json.loads(jsonStatus)
        #logger.debug(status)
        mqttc.publish('{}/mode'.format(AZ_TOPIC_STATUS), payload=status['mode'], retain=True)
        for z in status['zones']:
            #logger.debug(z)
            topic_prefix = '{}/{}'.format(AZ_TOPIC_STATUS, z['name'])
	    # 2022-05-29 15:01:11,628 {'air_demand': 0, 'can_fullfill': 0,
	    # 'humidity': 32, 'id': 1, 'name': 'Grenier', 'temp':
	    # 23.700000762939453, 'temp_request': 29.5}
            for s in ["on", "air_demand", "can_fullfill", "humidity", "temp", "temp_request"]:
                mqttc.publish('{}/{}'.format(topic_prefix, s), payload=z[s], retain=True)

    # MQTT Client init
    mqttc = mqtt.Client(client_id=args.mqtt_client)
    if args.mqtt_username and args.mqtt_password:
        logger.info('MQTT username and password provided')
        mqttc.username_pw_set(args.mqtt_username, password=args.mqtt_password)
    mqttc.connect(args.mqtt_address, port=args.mqtt_port, keepalive=60)

    mqttc.on_message = on_message
    mqttc.on_connect = on_connect

    #start the client
    mqttc.loop_start()

    # Monitoring loop
    while True:
        try:
            send_new_states()
            time.sleep(10)
        except (EOFError, SystemExit, KeyboardInterrupt):
            mqttc.disconnect()
            sys.exit(1)


parser = argparse.ArgumentParser(prog='airzone')
parser.add_argument("address", type=str, help="serial device or ip address for localapi")
parser.add_argument("port", type=str, help="serial tcp port or http port for localapi")
parser.add_argument("mqtt_address", type=str, help="mqtt ip address")
parser.add_argument("mqtt_client", type=str, help="mqtt client id")
parser.add_argument("--mqtt_port", type=int, help="mqtt ip port", default=1883)
parser.add_argument("--mqtt-username", type=str, help="MQTT username")
parser.add_argument("--mqtt-password", type=str, help="MQTT password")
parser.add_argument("--log", type=str, help="logging file")
parser.add_argument("--machine", type=int, default=1, help="Machine number where connect")
parser.add_argument("--system", choices=['innobus', 'aido', 'localapi'], default='localapi', help="Type of Airzone System")
parser.add_argument("--state", choices=['str', 'raw', 'json'], default='json',
                    help="Get the formatted state, or the raw machine state")
parser.add_argument("--rtuframer", type=bool, default= False, help="use rtu framer for modbus.")
parser.set_defaults(func=action)

args = parser.parse_args()
args.func(args)