Use attr** in meteoalarm (#61895)

Co-authored-by: epenet <epenet@users.noreply.github.com>
pull/62129/head
epenet 2021-12-16 23:01:31 +01:00 committed by GitHub
parent 863a139b6f
commit e20029d87f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 30 deletions

View File

@ -6,11 +6,11 @@ from meteoalertapi import Meteoalert
import voluptuous as vol
from homeassistant.components.binary_sensor import (
DEVICE_CLASS_SAFETY,
PLATFORM_SCHEMA,
BinarySensorDeviceClass,
BinarySensorEntity,
)
from homeassistant.const import ATTR_ATTRIBUTION, CONF_NAME
from homeassistant.const import CONF_NAME
import homeassistant.helpers.config_validation as cv
import homeassistant.util.dt as dt_util
@ -56,43 +56,23 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
class MeteoAlertBinarySensor(BinarySensorEntity):
"""Representation of a MeteoAlert binary sensor."""
_attr_attribution = ATTRIBUTION
_attr_device_class = BinarySensorDeviceClass.SAFETY
def __init__(self, api, name):
"""Initialize the MeteoAlert binary sensor."""
self._name = name
self._attributes = {}
self._state = None
self._attr_name = name
self._api = api
@property
def name(self):
"""Return the name of the binary sensor."""
return self._name
@property
def is_on(self):
"""Return the status of the binary sensor."""
return self._state
@property
def extra_state_attributes(self):
"""Return the state attributes."""
self._attributes[ATTR_ATTRIBUTION] = ATTRIBUTION
return self._attributes
@property
def device_class(self):
"""Return the device class of this binary sensor."""
return DEVICE_CLASS_SAFETY
def update(self):
"""Update device state."""
self._attributes = {}
self._state = False
self._attr_extra_state_attributes = None
self._attr_is_on = False
if alert := self._api.get_alert():
expiration_date = dt_util.parse_datetime(alert["expires"])
now = dt_util.utcnow()
if expiration_date > now:
self._attributes = alert
self._state = True
self._attr_extra_state_attributes = alert
self._attr_is_on = True