38d095aa18
* Define entity attributes as entity class variables * Example coronavirus integration * Example verisure * Cleanup/typing fixes * Fix Coronavirus * Revert "Fix Coronavirus" This reverts commit 060843860fe300f8448d0d2476de8963d5ddf5a2. * Revert "Cleanup/typing fixes" This reverts commit 659b79e75a97007f7181064e446c3e988c2d54bb. * Define entity attributes as entity class variables (attr alternative) * Example coronavirus * Example nut * Example verisure * Mark private * Cleanup after all reverting/cherrypicking/merging * Implement all entity properties * Update coronavirus example * Update nut example * Update verisure example * Lets not talk about this one... * Fix multiple class attribute
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
"""Sensor platform for the Corona virus."""
|
|
from homeassistant.components.sensor import SensorEntity
|
|
from homeassistant.const import ATTR_ATTRIBUTION
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from . import get_coordinator
|
|
from .const import ATTRIBUTION, OPTION_WORLDWIDE
|
|
|
|
SENSORS = {
|
|
"confirmed": "mdi:emoticon-neutral-outline",
|
|
"current": "mdi:emoticon-sad-outline",
|
|
"recovered": "mdi:emoticon-happy-outline",
|
|
"deaths": "mdi:emoticon-cry-outline",
|
|
}
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
"""Defer sensor setup to the shared sensor module."""
|
|
coordinator = await get_coordinator(hass)
|
|
|
|
async_add_entities(
|
|
CoronavirusSensor(coordinator, config_entry.data["country"], info_type)
|
|
for info_type in SENSORS
|
|
)
|
|
|
|
|
|
class CoronavirusSensor(CoordinatorEntity, SensorEntity):
|
|
"""Sensor representing corona virus data."""
|
|
|
|
_attr_unit_of_measurement = "people"
|
|
|
|
def __init__(self, coordinator, country, info_type):
|
|
"""Initialize coronavirus sensor."""
|
|
super().__init__(coordinator)
|
|
self._attr_extra_state_attributes = {ATTR_ATTRIBUTION: ATTRIBUTION}
|
|
self._attr_icon = SENSORS[info_type]
|
|
self._attr_unique_id = f"{country}-{info_type}"
|
|
if country == OPTION_WORLDWIDE:
|
|
self._attr_name = f"Worldwide Coronavirus {info_type}"
|
|
else:
|
|
self._attr_name = (
|
|
f"{coordinator.data[country].country} Coronavirus {info_type}"
|
|
)
|
|
|
|
self.country = country
|
|
self.info_type = info_type
|
|
|
|
@property
|
|
def available(self):
|
|
"""Return if sensor is available."""
|
|
return self.coordinator.last_update_success and (
|
|
self.country in self.coordinator.data or self.country == OPTION_WORLDWIDE
|
|
)
|
|
|
|
@property
|
|
def state(self):
|
|
"""State of the sensor."""
|
|
if self.country == OPTION_WORLDWIDE:
|
|
sum_cases = 0
|
|
for case in self.coordinator.data.values():
|
|
value = getattr(case, self.info_type)
|
|
if value is None:
|
|
continue
|
|
sum_cases += value
|
|
|
|
return sum_cases
|
|
|
|
return getattr(self.coordinator.data[self.country], self.info_type)
|