mirror of https://github.com/lumapu/ahoy.git
lumapu
4 years ago
committed by
GitHub
GitHub
6 changed files with 473 additions and 155 deletions
@ -0,0 +1,197 @@ |
|||||
|
#!/usr/bin/env python3 |
||||
|
# -*- coding: utf-8 -*- |
||||
|
|
||||
|
""" |
||||
|
Hoymiles output plugin library |
||||
|
""" |
||||
|
|
||||
|
import socket |
||||
|
from datetime import datetime, timezone |
||||
|
from hoymiles.decoders import StatusResponse |
||||
|
|
||||
|
try: |
||||
|
from influxdb_client import InfluxDBClient |
||||
|
except ModuleNotFoundError: |
||||
|
pass |
||||
|
|
||||
|
class OutputPluginFactory: |
||||
|
def __init__(self, **params): |
||||
|
""" |
||||
|
Initialize output plugin |
||||
|
|
||||
|
:param inverter_ser: The inverter serial |
||||
|
:type inverter_ser: str |
||||
|
:param inverter_name: The configured name for the inverter |
||||
|
:type inverter_name: str |
||||
|
""" |
||||
|
|
||||
|
self.inverter_ser = params.get('inverter_ser', '') |
||||
|
self.inverter_name = params.get('inverter_name', None) |
||||
|
|
||||
|
def store_status(self, response, **params): |
||||
|
""" |
||||
|
Default function |
||||
|
|
||||
|
:raises NotImplementedError: when the plugin does not implement store status data |
||||
|
""" |
||||
|
raise NotImplementedError('The current output plugin does not implement store_status') |
||||
|
|
||||
|
class InfluxOutputPlugin(OutputPluginFactory): |
||||
|
""" Influx2 output plugin """ |
||||
|
api = None |
||||
|
|
||||
|
def __init__(self, url, token, **params): |
||||
|
""" |
||||
|
Initialize InfluxOutputPlugin |
||||
|
|
||||
|
The following targets must be present in your InfluxDB. This does not |
||||
|
automatically create anything for You. |
||||
|
|
||||
|
:param str url: The url to connect this client to. Like http://localhost:8086 |
||||
|
:param str token: Influx2 access token which is allowed to write to bucket |
||||
|
:param org: Influx2 org, the token belongs to |
||||
|
:type org: str |
||||
|
:param bucket: Influx2 bucket to store data in (also known as retention policy) |
||||
|
:type bucket: str |
||||
|
:param measurement: Default measurement-prefix to use |
||||
|
:type measurement: str |
||||
|
""" |
||||
|
super().__init__(**params) |
||||
|
|
||||
|
self._bucket = params.get('bucket', 'hoymiles/autogen') |
||||
|
self._org = params.get('org', '') |
||||
|
self._measurement = params.get('measurement', |
||||
|
f'inverter,host={socket.gethostname()}') |
||||
|
|
||||
|
client = InfluxDBClient(url, token, bucket=self._bucket) |
||||
|
self.api = client.write_api() |
||||
|
|
||||
|
def store_status(self, response, **params): |
||||
|
""" |
||||
|
Publish StatusResponse object |
||||
|
|
||||
|
:param hoymiles.decoders.StatusResponse response: StatusResponse object |
||||
|
:type response: hoymiles.decoders.StatusResponse |
||||
|
:param measurement: Custom influx measurement name |
||||
|
:type measurement: str or None |
||||
|
|
||||
|
:raises ValueError: when response is not instance of StatusResponse |
||||
|
""" |
||||
|
|
||||
|
if not isinstance(response, StatusResponse): |
||||
|
raise ValueError('Data needs to be instance of StatusResponse') |
||||
|
|
||||
|
data = response.__dict__() |
||||
|
|
||||
|
measurement = self._measurement + f',location={data["inverter_ser"]}' |
||||
|
|
||||
|
data_stack = [] |
||||
|
|
||||
|
time_rx = datetime.now() |
||||
|
if 'time' in data and isinstance(data['time'], datetime): |
||||
|
time_rx = data['time'] |
||||
|
|
||||
|
# InfluxDB uses UTC |
||||
|
utctime = datetime.fromtimestamp(time_rx.timestamp(), tz=timezone.utc) |
||||
|
|
||||
|
# InfluxDB requires nanoseconds |
||||
|
ctime = int(utctime.timestamp() * 1e9) |
||||
|
|
||||
|
# AC Data |
||||
|
phase_id = 0 |
||||
|
for phase in data['phases']: |
||||
|
data_stack.append(f'{measurement},phase={phase_id},type=power value={phase["power"]} {ctime}') |
||||
|
data_stack.append(f'{measurement},phase={phase_id},type=voltage value={phase["voltage"]} {ctime}') |
||||
|
data_stack.append(f'{measurement},phase={phase_id},type=current value={phase["current"]} {ctime}') |
||||
|
phase_id = phase_id + 1 |
||||
|
|
||||
|
# DC Data |
||||
|
string_id = 0 |
||||
|
for string in data['strings']: |
||||
|
data_stack.append(f'{measurement},string={string_id},type=total value={string["energy_total"]/1000:.4f} {ctime}') |
||||
|
data_stack.append(f'{measurement},string={string_id},type=power value={string["power"]:.2f} {ctime}') |
||||
|
data_stack.append(f'{measurement},string={string_id},type=voltage value={string["voltage"]:.3f} {ctime}') |
||||
|
data_stack.append(f'{measurement},string={string_id},type=current value={string["current"]:3f} {ctime}') |
||||
|
string_id = string_id + 1 |
||||
|
# Global |
||||
|
data_stack.append(f'{measurement},type=frequency value={data["frequency"]:.3f} {ctime}') |
||||
|
data_stack.append(f'{measurement},type=temperature value={data["temperature"]:.2f} {ctime}') |
||||
|
|
||||
|
self.api.write(self._bucket, self._org, data_stack) |
||||
|
|
||||
|
try: |
||||
|
import paho.mqtt.client |
||||
|
except ModuleNotFoundError: |
||||
|
pass |
||||
|
|
||||
|
class MqttOutputPlugin(OutputPluginFactory): |
||||
|
""" Mqtt output plugin """ |
||||
|
client = None |
||||
|
|
||||
|
def __init__(self, *args, **params): |
||||
|
""" |
||||
|
Initialize MqttOutputPlugin |
||||
|
|
||||
|
:param host: Broker ip or hostname (defaults to: 127.0.0.1) |
||||
|
:type host: str |
||||
|
:param port: Broker port |
||||
|
:type port: int (defaults to: 1883) |
||||
|
:param user: Optional username to login to the broker |
||||
|
:type user: str or None |
||||
|
:param password: Optional passwort to login to the broker |
||||
|
:type password: str or None |
||||
|
:param topic: Topic prefix to use (defaults to: hoymiles/{inverter_ser}) |
||||
|
:type topic: str |
||||
|
|
||||
|
:param paho.mqtt.client.Client broker: mqtt-client instance |
||||
|
:param str inverter_ser: inverter serial |
||||
|
:param hoymiles.StatusResponse data: decoded inverter StatusResponse |
||||
|
:param topic: custom mqtt topic prefix (default: hoymiles/{inverter_ser}) |
||||
|
:type topic: str |
||||
|
""" |
||||
|
super().__init__(*args, **params) |
||||
|
|
||||
|
mqtt_client = paho.mqtt.client.Client() |
||||
|
mqtt_client.username_pw_set(params.get('user', None), params.get('password', None)) |
||||
|
mqtt_client.connect(params.get('host', '127.0.0.1'), params.get('port', 1883)) |
||||
|
mqtt_client.loop_start() |
||||
|
|
||||
|
self.client = mqtt_client |
||||
|
|
||||
|
def store_status(self, response, **params): |
||||
|
""" |
||||
|
Publish StatusResponse object |
||||
|
|
||||
|
:param hoymiles.decoders.StatusResponse response: StatusResponse object |
||||
|
:param topic: custom mqtt topic prefix (default: hoymiles/{inverter_ser}) |
||||
|
:type topic: str |
||||
|
|
||||
|
:raises ValueError: when response is not instance of StatusResponse |
||||
|
""" |
||||
|
|
||||
|
if not isinstance(response, StatusResponse): |
||||
|
raise ValueError('Data needs to be instance of StatusResponse') |
||||
|
|
||||
|
data = response.__dict__() |
||||
|
|
||||
|
topic = params.get('topic', f'hoymiles/{data["inverter_ser"]}') |
||||
|
|
||||
|
# AC Data |
||||
|
phase_id = 0 |
||||
|
for phase in data['phases']: |
||||
|
self.client.publish(f'{topic}/emeter/{phase_id}/power', phase['power']) |
||||
|
self.client.publish(f'{topic}/emeter/{phase_id}/voltage', phase['voltage']) |
||||
|
self.client.publish(f'{topic}/emeter/{phase_id}/current', phase['current']) |
||||
|
phase_id = phase_id + 1 |
||||
|
|
||||
|
# DC Data |
||||
|
string_id = 0 |
||||
|
for string in data['strings']: |
||||
|
self.client.publish(f'{topic}/emeter-dc/{string_id}/total', string['energy_total']/1000) |
||||
|
self.client.publish(f'{topic}/emeter-dc/{string_id}/power', string['power']) |
||||
|
self.client.publish(f'{topic}/emeter-dc/{string_id}/voltage', string['voltage']) |
||||
|
self.client.publish(f'{topic}/emeter-dc/{string_id}/current', string['current']) |
||||
|
string_id = string_id + 1 |
||||
|
# Global |
||||
|
self.client.publish(f'{topic}/frequency', data['frequency']) |
||||
|
self.client.publish(f'{topic}/temperature', data['temperature']) |
||||
@ -0,0 +1 @@ |
|||||
|
influxdb-client>=1.28.0 |
||||
Loading…
Reference in new issue