evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve type hints in template (#74294)

Browse source
This commit is contained in:
epenet 2022-07-01 19:05:37 +02:00 • committed by GitHub
parent 72917f1d2c
commit 9211ba8371
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 33 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

13
homeassistant/components/template/binary_sensor.py
View file

@ -14,6 +14,7 @@ from homeassistant.components.binary_sensor import (
DOMAIN as BINARY_SENSOR_DOMAIN,
ENTITY_ID_FORMAT,
PLATFORM_SCHEMA,
BinarySensorDeviceClass,
BinarySensorEntity,
)
from homeassistant.const import (
@ -208,16 +209,18 @@ class BinarySensorTemplate(TemplateEntity, BinarySensorEntity, RestoreEntity):
ENTITY_ID_FORMAT, object_id, hass=hass
)
self._device_class = config.get(CONF_DEVICE_CLASS)
self._device_class: BinarySensorDeviceClass | None = config.get(
CONF_DEVICE_CLASS
)
self._template = config[CONF_STATE]
self._state = None
self._state: bool | None = None
self._delay_cancel = None
self._delay_on = None
self._delay_on_raw = config.get(CONF_DELAY_ON)
self._delay_off = None
self._delay_off_raw = config.get(CONF_DELAY_OFF)
async def async_added_to_hass(self):
async def async_added_to_hass(self) -> None:
"""Restore state and register callbacks."""
if (
(self._delay_on_raw is not None or self._delay_off_raw is not None)
@ -283,12 +286,12 @@ class BinarySensorTemplate(TemplateEntity, BinarySensorEntity, RestoreEntity):
self._delay_cancel = async_call_later(self.hass, delay, _set_state)
@property
def is_on(self):
def is_on(self) -> bool | None:
"""Return true if sensor is on."""
return self._state
@property
def device_class(self):
def device_class(self) -> BinarySensorDeviceClass | None:
"""Return the sensor class of the binary sensor."""
return self._device_class