Home Assistant Git Exporter
This commit is contained in:
root
284
config/custom_components/apsystems_ecur/sensor.py
Normal file
284
config/custom_components/apsystems_ecur/sensor.py
Normal file
@@ -0,0 +1,284 @@
|
||||
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
|
||||
Reference in New Issue
Block a user