mirror of
https://github.com/lumapu/ahoy.git
synced 2025-06-12 23:51:39 +02:00
Integrate ahoy.py into hoymiles module
Finally get rid of ahoy.py and integrate functionallity into the module itself. Prepares for pipelines, adding pip installer or debian packaging. Improve configuration adds commandline switches for: * `--verbose, enabling verbose logging * `--log-transactions`, outbut all rf raw data Improve loop, now runs all queued commands per inverter within interval. Skip sleep when interval is allready due.
This commit is contained in:
Jan-Jonas Sämann
parent
6627eeb931
commit
d27f0c1148
4 changed files with 129 additions and 75 deletions
|
|
@ -1,212 +0,0 @@
|
|||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys
|
||||
import struct
|
||||
import re
|
||||
import time
|
||||
from datetime import datetime
|
||||
import argparse
|
||||
import hoymiles
|
||||
from RF24 import RF24, RF24_PA_LOW, RF24_PA_MAX, RF24_250KBPS, RF24_CRC_DISABLED, RF24_CRC_8, RF24_CRC_16
|
||||
import paho.mqtt.client
|
||||
import yaml
|
||||
from yaml.loader import SafeLoader
|
||||
|
||||
parser = argparse.ArgumentParser(description='Ahoy - Hoymiles solar inverter gateway')
|
||||
parser.add_argument("-c", "--config-file", nargs="?",
|
||||
help="configuration file")
|
||||
global_config = parser.parse_args()
|
||||
|
||||
if global_config.config_file:
|
||||
with open(global_config.config_file) as yf:
|
||||
cfg = yaml.load(yf, Loader=SafeLoader)
|
||||
else:
|
||||
with open(global_config.config_file) as yf:
|
||||
cfg = yaml.load('ahoy.yml', Loader=SafeLoader)
|
||||
|
||||
radio = RF24(22, 0, 1000000)
|
||||
hmradio = hoymiles.HoymilesNRF(device=radio)
|
||||
mqtt_client = None
|
||||
|
||||
command_queue = {}
|
||||
mqtt_command_topic_subs = []
|
||||
|
||||
hoymiles.HOYMILES_TRANSACTION_LOGGING=True
|
||||
hoymiles.HOYMILES_DEBUG_LOGGING=True
|
||||
|
||||
def main_loop():
|
||||
inverters = [
|
||||
inverter for inverter in ahoy_config.get('inverters', [])
|
||||
if not inverter.get('disabled', False)]
|
||||
|
||||
for inverter in inverters:
|
||||
if hoymiles.HOYMILES_DEBUG_LOGGING:
|
||||
print(f'Poll inverter {inverter["serial"]}')
|
||||
poll_inverter(inverter)
|
||||
|
||||
def poll_inverter(inverter):
|
||||
inverter_ser = inverter.get('serial')
|
||||
dtu_ser = ahoy_config.get('dtu', {}).get('serial')
|
||||
|
||||
if len(command_queue[str(inverter_ser)]) > 0:
|
||||
payload = command_queue[str(inverter_ser)].pop(0)
|
||||
else:
|
||||
payload = hoymiles.compose_set_time_payload()
|
||||
|
||||
payload_ttl = 4
|
||||
while payload_ttl > 0:
|
||||
payload_ttl = payload_ttl - 1
|
||||
com = hoymiles.InverterTransaction(
|
||||
radio=hmradio,
|
||||
dtu_ser=dtu_ser,
|
||||
inverter_ser=inverter_ser,
|
||||
request=next(hoymiles.compose_esb_packet(
|
||||
payload,
|
||||
seq=b'\x80',
|
||||
src=dtu_ser,
|
||||
dst=inverter_ser
|
||||
)))
|
||||
response = None
|
||||
while com.rxtx():
|
||||
try:
|
||||
response = com.get_payload()
|
||||
payload_ttl = 0
|
||||
except Exception as e:
|
||||
print(f'Error while retrieving data: {e}')
|
||||
pass
|
||||
|
||||
if response:
|
||||
dt = datetime.now()
|
||||
print(f'{dt} Payload: ' + hoymiles.hexify_payload(response))
|
||||
decoder = hoymiles.ResponseDecoder(response,
|
||||
request=com.request,
|
||||
inverter_ser=inverter_ser
|
||||
)
|
||||
result = decoder.decode()
|
||||
if isinstance(result, hoymiles.decoders.StatusResponse):
|
||||
data = result.__dict__()
|
||||
if hoymiles.HOYMILES_DEBUG_LOGGING:
|
||||
print(f'{dt} Decoded: {data["temperature"]}', end='')
|
||||
phase_id = 0
|
||||
for phase in data['phases']:
|
||||
print(f' phase{phase_id}=voltage:{phase["voltage"]}, current:{phase["current"]}, power:{phase["power"]}, frequency:{data["frequency"]}', end='')
|
||||
phase_id = phase_id + 1
|
||||
string_id = 0
|
||||
for string in data['strings']:
|
||||
print(f' string{string_id}=voltage:{string["voltage"]}, current:{string["current"]}, power:{string["power"]}, total:{string["energy_total"]/1000}, daily:{string["energy_daily"]}', end='')
|
||||
string_id = string_id + 1
|
||||
print()
|
||||
|
||||
if mqtt_client:
|
||||
mqtt_send_status(mqtt_client, inverter_ser, data,
|
||||
topic=inverter.get('mqtt', {}).get('topic', None))
|
||||
|
||||
def mqtt_send_status(broker, inverter_ser, data, topic=None):
|
||||
""" Publish StatusResponse object """
|
||||
|
||||
if not topic:
|
||||
topic = f'hoymiles/{inverter_ser}'
|
||||
|
||||
# AC Data
|
||||
phase_id = 0
|
||||
for phase in data['phases']:
|
||||
broker.publish(f'{topic}/emeter/{phase_id}/power', phase['power'])
|
||||
broker.publish(f'{topic}/emeter/{phase_id}/voltage', phase['voltage'])
|
||||
broker.publish(f'{topic}/emeter/{phase_id}/current', phase['current'])
|
||||
phase_id = phase_id + 1
|
||||
|
||||
# DC Data
|
||||
string_id = 0
|
||||
for string in data['strings']:
|
||||
broker.publish(f'{topic}/emeter-dc/{string_id}/total', string['energy_total']/1000)
|
||||
broker.publish(f'{topic}/emeter-dc/{string_id}/power', string['power'])
|
||||
broker.publish(f'{topic}/emeter-dc/{string_id}/voltage', string['voltage'])
|
||||
broker.publish(f'{topic}/emeter-dc/{string_id}/current', string['current'])
|
||||
string_id = string_id + 1
|
||||
# Global
|
||||
broker.publish(f'{topic}/frequency', data['frequency'])
|
||||
broker.publish(f'{topic}/temperature', data['temperature'])
|
||||
|
||||
def mqtt_on_command(client, userdata, message):
|
||||
"""
|
||||
Handle commands to topic
|
||||
hoymiles/{inverter_ser}/command
|
||||
frame a payload and put onto command_queue
|
||||
|
||||
Inverters must have mqtt.send_raw_enabled: true configured
|
||||
|
||||
This can be used to inject debug payloads
|
||||
The message must be in hexlified format
|
||||
|
||||
Use of variables:
|
||||
tttttttt gets expanded to a current int(time)
|
||||
|
||||
Example injects exactly the same as we normally use to poll data:
|
||||
mosquitto -h broker -t inverter_topic/command -m 800b00tttttttt0000000500000000
|
||||
|
||||
This allows for even faster hacking during runtime
|
||||
"""
|
||||
try:
|
||||
inverter_ser = next(
|
||||
item[0] for item in mqtt_command_topic_subs if item[1] == message.topic)
|
||||
except StopIteration:
|
||||
print('Unexpedtedly received mqtt message for {message.topic}')
|
||||
|
||||
if inverter_ser:
|
||||
p_message = message.payload.decode('utf-8').lower()
|
||||
|
||||
# Expand tttttttt to current time for use in hexlified payload
|
||||
expand_time = ''.join(f'{b:02x}' for b in struct.pack('>L', int(time.time())))
|
||||
p_message = p_message.replace('tttttttt', expand_time)
|
||||
|
||||
if (len(p_message) < 2048 \
|
||||
and len(p_message) % 2 == 0 \
|
||||
and re.match(r'^[a-f0-9]+$', p_message)):
|
||||
payload = bytes.fromhex(p_message)
|
||||
# commands must start with \x80
|
||||
if payload[0] == 0x80:
|
||||
command_queue[str(inverter_ser)].append(
|
||||
hoymiles.frame_payload(payload[1:]))
|
||||
|
||||
if __name__ == '__main__':
|
||||
ahoy_config = dict(cfg.get('ahoy', {}))
|
||||
|
||||
mqtt_config = ahoy_config.get('mqtt', [])
|
||||
if not mqtt_config.get('disabled', False):
|
||||
mqtt_client = paho.mqtt.client.Client()
|
||||
mqtt_client.username_pw_set(mqtt_config.get('user', None), mqtt_config.get('password', None))
|
||||
mqtt_client.connect(mqtt_config.get('host', '127.0.0.1'), mqtt_config.get('port', 1883))
|
||||
mqtt_client.loop_start()
|
||||
mqtt_client.on_message = mqtt_on_command
|
||||
|
||||
if not radio.begin():
|
||||
raise RuntimeError('Can\'t open radio')
|
||||
|
||||
inverters = [inverter.get('serial') for inverter in ahoy_config.get('inverters', [])]
|
||||
for inverter in ahoy_config.get('inverters', []):
|
||||
inverter_ser = inverter.get('serial')
|
||||
command_queue[str(inverter_ser)] = []
|
||||
|
||||
#
|
||||
# Enables and subscribe inverter to mqtt /command-Topic
|
||||
#
|
||||
if inverter.get('mqtt', {}).get('send_raw_enabled', False):
|
||||
topic_item = (
|
||||
str(inverter_ser),
|
||||
inverter.get('mqtt', {}).get('topic', f'hoymiles/{inverter_ser}') + '/command'
|
||||
)
|
||||
mqtt_client.subscribe(topic_item[1])
|
||||
mqtt_command_topic_subs.append(topic_item)
|
||||
|
||||
loop_interval = ahoy_config.get('interval', 1)
|
||||
try:
|
||||
while True:
|
||||
main_loop()
|
||||
|
||||
if loop_interval:
|
||||
time.sleep(time.time() % loop_interval)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
radio.powerDown()
|
||||
sys.exit()
|
||||
Loading…
Add table
Add a link
Reference in a new issue