Add sensor platform to goalzero (#49835)
* Add sensor platform to goalzero * Tweak * Remove unused DATA_SCHEMA * Simplify * Remove last_reset * Update on reload
This commit is contained in:
parent
a3f1489785
commit
5cd4471c67
6 changed files with 174 additions and 5 deletions
59
homeassistant/components/goalzero/sensor.py
Normal file
59
homeassistant/components/goalzero/sensor.py
Normal file
|
@ -0,0 +1,59 @@
|
|||
"""Support for Goal Zero Yeti Sensors."""
|
||||
from homeassistant.components.sensor import ATTR_LAST_RESET, ATTR_STATE_CLASS
|
||||
from homeassistant.const import (
|
||||
ATTR_DEVICE_CLASS,
|
||||
ATTR_NAME,
|
||||
ATTR_UNIT_OF_MEASUREMENT,
|
||||
CONF_NAME,
|
||||
)
|
||||
|
||||
from . import YetiEntity
|
||||
from .const import (
|
||||
ATTR_DEFAULT_ENABLED,
|
||||
DATA_KEY_API,
|
||||
DATA_KEY_COORDINATOR,
|
||||
DOMAIN,
|
||||
SENSOR_DICT,
|
||||
)
|
||||
|
||||
|
||||
async def async_setup_entry(hass, entry, async_add_entities):
|
||||
"""Set up the Goal Zero Yeti sensor."""
|
||||
name = entry.data[CONF_NAME]
|
||||
goalzero_data = hass.data[DOMAIN][entry.entry_id]
|
||||
sensors = [
|
||||
YetiSensor(
|
||||
goalzero_data[DATA_KEY_API],
|
||||
goalzero_data[DATA_KEY_COORDINATOR],
|
||||
name,
|
||||
sensor_name,
|
||||
entry.entry_id,
|
||||
)
|
||||
for sensor_name in SENSOR_DICT
|
||||
]
|
||||
async_add_entities(sensors, True)
|
||||
|
||||
|
||||
class YetiSensor(YetiEntity):
|
||||
"""Representation of a Goal Zero Yeti sensor."""
|
||||
|
||||
def __init__(self, api, coordinator, name, sensor_name, server_unique_id):
|
||||
"""Initialize a Goal Zero Yeti sensor."""
|
||||
super().__init__(api, coordinator, name, server_unique_id)
|
||||
|
||||
self._condition = sensor_name
|
||||
|
||||
sensor = SENSOR_DICT[sensor_name]
|
||||
self._attr_name = f"{name} {sensor.get(ATTR_NAME)}"
|
||||
self._attr_unique_id = f"{self._server_unique_id}/{sensor_name}"
|
||||
self._attr_unit_of_measurement = sensor.get(ATTR_UNIT_OF_MEASUREMENT)
|
||||
self._attr_entity_registry_enabled_default = sensor.get(ATTR_DEFAULT_ENABLED)
|
||||
self._device_class = sensor.get(ATTR_DEVICE_CLASS)
|
||||
self._attr_last_reset = sensor.get(ATTR_LAST_RESET)
|
||||
self._attr_state_class = sensor.get(ATTR_STATE_CLASS)
|
||||
|
||||
@property
|
||||
def state(self):
|
||||
"""Return the state."""
|
||||
if self.api.data:
|
||||
return self.api.data[self._condition]
|
Loading…
Add table
Add a link
Reference in a new issue