-
Notifications
You must be signed in to change notification settings - Fork 74
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add support for reading/writing the Storedge control registers
Extend the integration to create number & select entities for the Storedge control registers. Through these registers one can control the battery charge / discharge profile.
- Loading branch information
1 parent
8c6e089
commit 7e56eaf
Showing
4 changed files
with
356 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,122 @@ | ||
| import logging | ||
| from typing import Optional, Dict, Any | ||
|
|
||
| from .const import ( | ||
| DOMAIN, | ||
| ATTR_MANUFACTURER, | ||
| STORAGE_NUMBER_TYPES, | ||
| ) | ||
|
|
||
| from pymodbus.constants import Endian | ||
| from pymodbus.payload import BinaryPayloadBuilder | ||
|
|
||
| from homeassistant.const import CONF_NAME | ||
| from homeassistant.components.number import ( | ||
| PLATFORM_SCHEMA, | ||
| NumberEntity, | ||
| ) | ||
|
|
||
| from homeassistant.core import callback | ||
|
|
||
| _LOGGER = logging.getLogger(__name__) | ||
|
|
||
| async def async_setup_entry(hass, entry, async_add_entities) -> None: | ||
| hub_name = entry.data[CONF_NAME] | ||
| hub = hass.data[DOMAIN][hub_name]["hub"] | ||
|
|
||
| device_info = { | ||
| "identifiers": {(DOMAIN, hub_name)}, | ||
| "name": hub_name, | ||
| "manufacturer": ATTR_MANUFACTURER, | ||
| } | ||
|
|
||
| entities = [] | ||
|
|
||
| if hub.read_battery1 == True or hub.read_battery2 == True: | ||
| for number_info in STORAGE_NUMBER_TYPES: | ||
| number = SolarEdgeNumber( | ||
| hub_name, | ||
| hub, | ||
| device_info, | ||
| number_info[0], | ||
| number_info[1], | ||
| number_info[2], | ||
| number_info[3], | ||
| number_info[4], | ||
| ) | ||
| entities.append(number) | ||
|
|
||
| async_add_entities(entities) | ||
| return True | ||
|
|
||
| class SolarEdgeNumber(NumberEntity): | ||
| """Representation of an SolarEdge Modbus number.""" | ||
|
|
||
| def __init__(self, | ||
| platform_name, | ||
| hub, | ||
| device_info, | ||
| name, | ||
| key, | ||
| register, | ||
| fmt, | ||
| attrs | ||
| ) -> None: | ||
| """Initialize the selector.""" | ||
| self._platform_name = platform_name | ||
| self._hub = hub | ||
| self._device_info = device_info | ||
| self._name = name | ||
| self._key = key | ||
| self._register = register | ||
| self._fmt = fmt | ||
|
|
||
| self._attr_min_value = attrs["min"] | ||
| self._attr_max_value = attrs["max"] | ||
| if "unit" in attrs.keys(): | ||
| self._attr_unit_of_measurement = attrs["unit"] | ||
|
|
||
| async def async_added_to_hass(self) -> None: | ||
| """Register callbacks.""" | ||
| self._hub.async_add_solaredge_sensor(self._modbus_data_updated) | ||
|
|
||
| async def async_will_remove_from_hass(self) -> None: | ||
| self._hub.async_remove_solaredge_sensor(self._modbus_data_updated) | ||
|
|
||
| @callback | ||
| def _modbus_data_updated(self) -> None: | ||
| self.async_write_ha_state() | ||
|
|
||
| @property | ||
| def name(self) -> str: | ||
| """Return the name.""" | ||
| return f"{self._platform_name} ({self._name})" | ||
|
|
||
| @property | ||
| def unique_id(self) -> Optional[str]: | ||
| return f"{self._platform_name}_{self._key}" | ||
|
|
||
| @property | ||
| def should_poll(self) -> bool: | ||
| """Data is delivered by the hub""" | ||
| return False | ||
|
|
||
| @property | ||
| def value(self) -> float: | ||
| if self._key in self._hub.data: | ||
| return self._hub.data[self._key] | ||
|
|
||
| async def async_set_value(self, value: float) -> None: | ||
| """Change the selected value.""" | ||
| builder = BinaryPayloadBuilder(byteorder=Endian.Big, wordorder=Endian.Little) | ||
|
|
||
| if self._fmt == "i": | ||
| builder.add_32bit_uint(int(value)) | ||
| elif self._fmt == "f": | ||
| builder.add_32bit_float(float(value)) | ||
|
|
||
| self._hub.write_registers(unit=1, address=self._register, payload=builder.to_registers()) | ||
|
|
||
| self._hub.data[self._key] = value | ||
| self.async_write_ha_state() | ||
|
|
Oops, something went wrong.