106 lines
2.9 KiB
Python
106 lines
2.9 KiB
Python
"""Fpl Entity class"""
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
from homeassistant.components.sensor import SensorEntity, STATE_CLASS_MEASUREMENT
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from homeassistant.const import (
|
|
CURRENCY_DOLLAR,
|
|
DEVICE_CLASS_ENERGY,
|
|
ENERGY_KILO_WATT_HOUR,
|
|
DEVICE_CLASS_MONETARY,
|
|
)
|
|
from .const import DOMAIN, VERSION, ATTRIBUTION
|
|
|
|
|
|
class FplEntity(CoordinatorEntity, SensorEntity):
|
|
"""FPL base entity"""
|
|
|
|
_attr_attribution = ATTRIBUTION
|
|
|
|
def __init__(self, coordinator, config_entry, account, sensorName):
|
|
super().__init__(coordinator)
|
|
self.config_entry = config_entry
|
|
self.account = account
|
|
self.sensorName = sensorName
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return the ID of this device."""
|
|
sensorName = self.sensorName.lower().replace(" ", "")
|
|
return f"{DOMAIN}{self.account}{sensorName}"
|
|
|
|
@property
|
|
def name(self):
|
|
return f"{DOMAIN.upper()} {self.account} {self.sensorName}"
|
|
|
|
@property
|
|
def device_info(self):
|
|
return {
|
|
"identifiers": {(DOMAIN, self.account)},
|
|
"name": f"FPL Account {self.account}",
|
|
"model": VERSION,
|
|
"manufacturer": "Florida Power & Light",
|
|
}
|
|
|
|
def customAttributes(self) -> dict:
|
|
"""override this method to set custom attributes"""
|
|
return {}
|
|
|
|
@property
|
|
def extra_state_attributes(self):
|
|
"""Return the state attributes."""
|
|
attributes = {
|
|
# "attribution": ATTRIBUTION,
|
|
# "integration": "FPL",
|
|
}
|
|
attributes.update(self.customAttributes())
|
|
return attributes
|
|
|
|
def getData(self, field):
|
|
"""call this method to retrieve sensor data"""
|
|
if self.coordinator.data is not None:
|
|
account = self.coordinator.data.get(self.account)
|
|
if account is not None:
|
|
return account.get(field, None)
|
|
return None
|
|
|
|
|
|
class FplEnergyEntity(FplEntity):
|
|
"""Represents a energy sensor"""
|
|
|
|
_attr_native_unit_of_measurement = ENERGY_KILO_WATT_HOUR
|
|
_attr_device_class = DEVICE_CLASS_ENERGY
|
|
_attr_icon = "mdi:flash"
|
|
_attr_state_class = STATE_CLASS_MEASUREMENT
|
|
|
|
@property
|
|
def last_reset(self) -> datetime:
|
|
"""Return the time when the sensor was last reset, if any."""
|
|
|
|
today = datetime.today()
|
|
yesterday = today - timedelta(days=1)
|
|
return datetime.combine(yesterday, datetime.min.time())
|
|
|
|
|
|
class FplMoneyEntity(FplEntity):
|
|
"""Represents a money sensor"""
|
|
|
|
_attr_native_unit_of_measurement = CURRENCY_DOLLAR
|
|
_attr_device_class = DEVICE_CLASS_MONETARY
|
|
_attr_icon = "mdi:currency-usd"
|
|
|
|
|
|
class FplDateEntity(FplEntity):
|
|
"""Represents a date or days"""
|
|
|
|
_attr_icon = "mdi:calendar"
|
|
|
|
|
|
class FplDayEntity(FplEntity):
|
|
"""Represents a date or days"""
|
|
|
|
_attr_native_unit_of_measurement = "days"
|
|
_attr_icon = "mdi:calendar"
|