huaweisolar.py

from datetime import datetime
import logging
import time
import huawei_solar
import paho.mqtt.client
import os
import json

version = "1.3.0"
FORMAT = (f'{version} - %(asctime)-15s %(threadName)-15s '
          '%(levelname)-8s %(module)-15s:%(lineno)-8s %(message)s')
logging.basicConfig(format=FORMAT)
log = logging.getLogger()
log.setLevel(logging.INFO)

inverter_ip = os.getenv('INVERTER_IP', '192.168.1.1')
mqtt_host = os.getenv('MQTT_IP', '192.168.1.1')

inverter = huawei_solar.HuaweiSolar(inverter_ip, port=int(
    os.getenv('INVERTER_PORT', "502")), slave=1)
inverter._slave = 1
inverter.wait = 1

vars_immediate_default = ['pv_01_voltage', 'pv_01_current', 'pv_02_voltage', 'pv_02_current', 'input_power', 'grid_voltage',
                          'grid_current', 'active_power', 'grid_A_voltage', 'active_grid_A_current', 'power_meter_active_power', 'storage_unit_1_total_charge']

immediate_results = {}


def get_day_start():
    now = datetime.now()
    return str(datetime(now.year, now.month, now.day))


def get_installation_date():
    (year, month, day) = int(os.getenv('INSTALLATION_YEAR')), int(
        os.getenv('INSTALLATION_MONTH')), int(os.getenv('INSTALLATION_DAY'))
    return str(datetime(year, month, day))


def try_modBus_variable(variable):
    global immediate_results

    try:
        result = inverter.get(variable)
        return {'value': result.value, 'unit': result.unit}
    except:
        log.warning(f"❌ Failed to get {variable}!")
        return immediate_results[variable]


def modbusAccess():
    global immediate_results

    vars_immediate = os.getenv('IMMEDIATE_VARS', ','.join(
        vars_immediate_default)).split(',')

    while True:
        immediate_results = {var: try_modBus_variable(
            var) for var in vars_immediate}

        log.info('✅ Obtained the results from Modbus!')
        immediate_results['day_start'] = get_day_start()
        immediate_results['installation_date'] = get_installation_date()
        log.info('✅ Added the start of the day to the object!')

        clientMQTT.publish(topic="huawei/node/solar",
                           payload=json.dumps(immediate_results), qos=1, retain=False)
        log.info('🚀 Publishing immediate results...')

        time.sleep(5)


def on_connect(client, userdata, flags, rc):
    if rc == 0:
        client.connected_flag = True  # set flag
        log.info("MQTT OK!")
    else:
        log.info("MQTT FAILURE. ERROR CODE: %s", rc)


paho.mqtt.client.Client.connected_flag = False  # create flag in class
broker_port = int(os.getenv("MQTT_PORT", "1883"))

clientMQTT = paho.mqtt.client.Client()
clientMQTT.on_connect = on_connect  # bind call back function
clientMQTT.loop_start()
log.info("Connecting to MQTT broker: %s:%d ", mqtt_host, broker_port)
clientMQTT.username_pw_set(username=os.getenv('MQTT_USER', 'user'),
                           password=os.getenv('MQTT_PASS', 'password'))
clientMQTT.connect(mqtt_host, broker_port)  # connect to broker
while not clientMQTT.connected_flag:  # wait in loop
    # log.info("...")
    pass
time.sleep(1)
log.info("START MODBUS...")

try:
    modbusAccess()
except:
    log.error("❌ Error! Stopping MQTT!")
    clientMQTT.loop_stop()
    exit(1)