285 lines
8.8 KiB
Python
285 lines
8.8 KiB
Python
from datetime import timedelta, datetime, date
|
|
import logging
|
|
|
|
import async_timeout
|
|
|
|
from homeassistant.util import dt as dt_util
|
|
from homeassistant.components.sensor import SensorEntity
|
|
from homeassistant.helpers.entity import EntityCategory
|
|
from homeassistant.helpers.update_coordinator import (
|
|
CoordinatorEntity,
|
|
)
|
|
|
|
from homeassistant.components.sensor import (
|
|
SensorDeviceClass,
|
|
SensorStateClass,
|
|
)
|
|
|
|
from .const import (
|
|
DOMAIN,
|
|
SOLAR_ICON,
|
|
FREQ_ICON,
|
|
SIGNAL_ICON
|
|
)
|
|
|
|
from homeassistant.const import (
|
|
UnitOfPower,
|
|
UnitOfEnergy,
|
|
UnitOfTemperature,
|
|
UnitOfElectricPotential,
|
|
UnitOfFrequency,
|
|
PERCENTAGE
|
|
)
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
async def async_setup_entry(hass, config, add_entities, discovery_info=None):
|
|
|
|
ecu = hass.data[DOMAIN].get("ecu")
|
|
coordinator = hass.data[DOMAIN].get("coordinator")
|
|
|
|
sensors = [
|
|
APSystemsECUSensor(coordinator, ecu, "current_power",
|
|
label="Current Power",
|
|
unit=UnitOfPower.WATT,
|
|
devclass=SensorDeviceClass.POWER,
|
|
icon=SOLAR_ICON,
|
|
stateclass=SensorStateClass.MEASUREMENT
|
|
),
|
|
APSystemsECUSensor(coordinator, ecu, "today_energy",
|
|
label="Today Energy",
|
|
unit=UnitOfEnergy.KILO_WATT_HOUR,
|
|
devclass=SensorDeviceClass.ENERGY,
|
|
icon=SOLAR_ICON,
|
|
stateclass=SensorStateClass.TOTAL_INCREASING
|
|
),
|
|
APSystemsECUSensor(coordinator, ecu, "lifetime_energy",
|
|
label="Lifetime Energy",
|
|
unit=UnitOfEnergy.KILO_WATT_HOUR,
|
|
devclass=SensorDeviceClass.ENERGY,
|
|
icon=SOLAR_ICON,
|
|
stateclass=SensorStateClass.TOTAL_INCREASING
|
|
),
|
|
APSystemsECUSensor(coordinator, ecu, "qty_of_inverters",
|
|
label="Inverters",
|
|
icon=SOLAR_ICON,
|
|
entity_category=EntityCategory.DIAGNOSTIC
|
|
),
|
|
APSystemsECUSensor(coordinator, ecu, "qty_of_online_inverters",
|
|
label="Inverters Online",
|
|
icon=SOLAR_ICON,
|
|
entity_category=EntityCategory.DIAGNOSTIC
|
|
),
|
|
]
|
|
|
|
inverters = coordinator.data.get("inverters", {})
|
|
for uid,inv_data in inverters.items():
|
|
_LOGGER.debug(f"Inverter {uid} {inv_data.get('channel_qty')}")
|
|
# https://github.com/ksheumaker/homeassistant-apsystems_ecur/issues/110
|
|
if inv_data.get("channel_qty") != None:
|
|
sensors.extend([
|
|
APSystemsECUInverterSensor(coordinator, ecu, uid, "temperature",
|
|
label="Temperature",
|
|
unit=UnitOfTemperature.CELSIUS,
|
|
devclass=SensorDeviceClass.TEMPERATURE,
|
|
stateclass=SensorStateClass.MEASUREMENT,
|
|
entity_category=EntityCategory.DIAGNOSTIC
|
|
),
|
|
APSystemsECUInverterSensor(coordinator, ecu, uid, "frequency",
|
|
label="Frequency",
|
|
unit=UnitOfFrequency.HERTZ,
|
|
stateclass=SensorStateClass.MEASUREMENT,
|
|
devclass=SensorDeviceClass.FREQUENCY,
|
|
icon=FREQ_ICON,
|
|
entity_category=EntityCategory.DIAGNOSTIC
|
|
),
|
|
APSystemsECUInverterSensor(coordinator, ecu, uid, "voltage",
|
|
label="Voltage",
|
|
unit=UnitOfElectricPotential.VOLT,
|
|
stateclass=SensorStateClass.MEASUREMENT,
|
|
devclass=SensorDeviceClass.VOLTAGE, entity_category=EntityCategory.DIAGNOSTIC
|
|
),
|
|
APSystemsECUInverterSensor(coordinator, ecu, uid, "signal",
|
|
label="Signal",
|
|
unit=PERCENTAGE,
|
|
stateclass=SensorStateClass.MEASUREMENT,
|
|
devclass=SensorDeviceClass.SIGNAL_STRENGTH,
|
|
icon=SIGNAL_ICON,
|
|
entity_category=EntityCategory.DIAGNOSTIC
|
|
)
|
|
])
|
|
for i in range(0, inv_data.get("channel_qty", 0)):
|
|
sensors.append(
|
|
APSystemsECUInverterSensor(coordinator, ecu, uid, f"power",
|
|
index=i, label=f"Power Ch {i+1}",
|
|
unit=UnitOfPower.WATT,
|
|
devclass=SensorDeviceClass.POWER,
|
|
icon=SOLAR_ICON,
|
|
stateclass=SensorStateClass.MEASUREMENT
|
|
)
|
|
)
|
|
add_entities(sensors)
|
|
|
|
|
|
class APSystemsECUInverterSensor(CoordinatorEntity, SensorEntity):
|
|
def __init__(self, coordinator, ecu, uid, field, index=0, label=None, icon=None, unit=None, devclass=None, stateclass=None, entity_category=None):
|
|
|
|
super().__init__(coordinator)
|
|
|
|
self.coordinator = coordinator
|
|
|
|
self._index = index
|
|
self._uid = uid
|
|
self._ecu = ecu
|
|
self._field = field
|
|
self._devclass = devclass
|
|
self._label = label
|
|
if not label:
|
|
self._label = field
|
|
self._icon = icon
|
|
self._unit = unit
|
|
self._stateclass = stateclass
|
|
self._entity_category = entity_category
|
|
|
|
self._name = f"Inverter {self._uid} {self._label}"
|
|
self._state = None
|
|
|
|
@property
|
|
def unique_id(self):
|
|
field = self._field
|
|
if self._index != None:
|
|
field = f"{field}_{self._index}"
|
|
return f"{self._ecu.ecu.ecu_id}_{self._uid}_{field}"
|
|
|
|
@property
|
|
def device_class(self):
|
|
return self._devclass
|
|
|
|
@property
|
|
def name(self):
|
|
return self._name
|
|
|
|
@property
|
|
def state(self):
|
|
_LOGGER.debug(f"State called for {self._field}")
|
|
if self._field == "voltage":
|
|
return self.coordinator.data.get("inverters", {}).get(self._uid, {}).get("voltage", [])[0]
|
|
elif self._field == "power":
|
|
_LOGGER.debug(f"POWER {self._uid} {self._index}")
|
|
return self.coordinator.data.get("inverters", {}).get(self._uid, {}).get("power", [])[self._index]
|
|
else:
|
|
return self.coordinator.data.get("inverters", {}).get(self._uid, {}).get(self._field)
|
|
|
|
@property
|
|
def icon(self):
|
|
return self._icon
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
return self._unit
|
|
|
|
|
|
@property
|
|
def extra_state_attributes(self):
|
|
attrs = {
|
|
"ecu_id" : self._ecu.ecu.ecu_id,
|
|
"inverter_uid" : self._uid,
|
|
"last_update" : self._ecu.ecu.last_update,
|
|
}
|
|
return attrs
|
|
|
|
@property
|
|
def state_class(self):
|
|
_LOGGER.debug(f"State class {self._stateclass} - {self._field}")
|
|
return self._stateclass
|
|
|
|
@property
|
|
def device_info(self):
|
|
parent = f"inverter_{self._uid}"
|
|
return {
|
|
"identifiers": {
|
|
(DOMAIN, parent),
|
|
}
|
|
}
|
|
|
|
@property
|
|
def entity_category(self):
|
|
return self._entity_category
|
|
|
|
class APSystemsECUSensor(CoordinatorEntity, SensorEntity):
|
|
|
|
def __init__(self, coordinator, ecu, field, label=None, icon=None, unit=None, devclass=None, stateclass=None, entity_category=None):
|
|
|
|
super().__init__(coordinator)
|
|
|
|
self.coordinator = coordinator
|
|
|
|
self._ecu = ecu
|
|
self._field = field
|
|
self._label = label
|
|
if not label:
|
|
self._label = field
|
|
self._icon = icon
|
|
self._unit = unit
|
|
self._devclass = devclass
|
|
self._stateclass = stateclass
|
|
self._entity_category = entity_category
|
|
|
|
self._name = f"ECU {self._label}"
|
|
self._state = None
|
|
|
|
@property
|
|
def unique_id(self):
|
|
return f"{self._ecu.ecu.ecu_id}_{self._field}"
|
|
|
|
@property
|
|
def name(self):
|
|
return self._name
|
|
|
|
@property
|
|
def device_class(self):
|
|
return self._devclass
|
|
|
|
@property
|
|
def state(self):
|
|
_LOGGER.debug(f"State called for {self._field}")
|
|
return self.coordinator.data.get(self._field)
|
|
|
|
@property
|
|
def icon(self):
|
|
return self._icon
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
return self._unit
|
|
|
|
@property
|
|
def extra_state_attributes(self):
|
|
|
|
attrs = {
|
|
"ecu_id" : self._ecu.ecu.ecu_id,
|
|
"Firmware" : self._ecu.ecu.firmware,
|
|
"Timezone" : self._ecu.ecu.timezone,
|
|
"last_update" : self._ecu.ecu.last_update
|
|
}
|
|
return attrs
|
|
|
|
@property
|
|
def state_class(self):
|
|
_LOGGER.debug(f"State class {self._stateclass} - {self._field}")
|
|
return self._stateclass
|
|
|
|
@property
|
|
def device_info(self):
|
|
parent = f"ecu_{self._ecu.ecu.ecu_id}"
|
|
return {
|
|
"identifiers": {
|
|
(DOMAIN, parent),
|
|
}
|
|
}
|
|
|
|
@property
|
|
def entity_category(self):
|
|
return self._entity_category
|