121d967732
* Add coronavirus integration * Update homeassistant/components/coronavirus/manifest.json Co-Authored-By: Franck Nijhof <git@frenck.dev> Co-authored-by: Franck Nijhof <frenck@frenck.nl>
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
"""Sensor platform for the Corona virus."""
|
|
from homeassistant.const import ATTR_ATTRIBUTION
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from . import get_coordinator
|
|
from .const import ATTRIBUTION, OPTION_WORLDWIDE
|
|
|
|
|
|
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 ("confirmed", "recovered", "deaths", "current")
|
|
)
|
|
|
|
|
|
class CoronavirusSensor(Entity):
|
|
"""Sensor representing corona virus data."""
|
|
|
|
name = None
|
|
unique_id = None
|
|
|
|
def __init__(self, coordinator, country, info_type):
|
|
"""Initialize coronavirus sensor."""
|
|
if country == OPTION_WORLDWIDE:
|
|
self.name = f"Worldwide {info_type}"
|
|
else:
|
|
self.name = f"{coordinator.data[country].country} {info_type}"
|
|
self.unique_id = f"{country}-{info_type}"
|
|
self.coordinator = coordinator
|
|
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:
|
|
return sum(
|
|
getattr(case, self.info_type) for case in self.coordinator.data.values()
|
|
)
|
|
|
|
return getattr(self.coordinator.data[self.country], self.info_type)
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return unit of measurement."""
|
|
return "people"
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return device attributes."""
|
|
return {ATTR_ATTRIBUTION: ATTRIBUTION}
|
|
|
|
async def async_added_to_hass(self):
|
|
"""When entity is added to hass."""
|
|
self.coordinator.async_add_listener(self.async_write_ha_state)
|
|
|
|
async def async_will_remove_from_hass(self):
|
|
"""When entity will be removed from hass."""
|
|
self.coordinator.async_remove_listener(self.async_write_ha_state)
|