1
mirror of https://github.com/home-assistant/core synced 2024-10-13 14:52:10 +02:00
ha-core/homeassistant/components/blink/binary_sensor.py
epenet fbabad1d1d
Add binary_sensor setup type hints [a-f] (#63268)
Co-authored-by: epenet <epenet@users.noreply.github.com>
2022-01-03 13:10:41 +01:00

79 lines
2.4 KiB
Python

"""Support for Blink system camera control."""
from __future__ import annotations
from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass,
BinarySensorEntity,
BinarySensorEntityDescription,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import DeviceInfo, EntityCategory
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import (
DEFAULT_BRAND,
DOMAIN,
TYPE_BATTERY,
TYPE_CAMERA_ARMED,
TYPE_MOTION_DETECTED,
)
BINARY_SENSORS_TYPES: tuple[BinarySensorEntityDescription, ...] = (
BinarySensorEntityDescription(
key=TYPE_BATTERY,
name="Battery",
device_class=BinarySensorDeviceClass.BATTERY,
entity_category=EntityCategory.DIAGNOSTIC,
),
BinarySensorEntityDescription(
key=TYPE_CAMERA_ARMED,
name="Camera Armed",
),
BinarySensorEntityDescription(
key=TYPE_MOTION_DETECTED,
name="Motion Detected",
device_class=BinarySensorDeviceClass.MOTION,
),
)
async def async_setup_entry(
hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None:
"""Set up the blink binary sensors."""
data = hass.data[DOMAIN][config.entry_id]
entities = [
BlinkBinarySensor(data, camera, description)
for camera in data.cameras
for description in BINARY_SENSORS_TYPES
]
async_add_entities(entities)
class BlinkBinarySensor(BinarySensorEntity):
"""Representation of a Blink binary sensor."""
def __init__(self, data, camera, description: BinarySensorEntityDescription):
"""Initialize the sensor."""
self.data = data
self.entity_description = description
self._attr_name = f"{DOMAIN} {camera} {description.name}"
self._camera = data.cameras[camera]
self._attr_unique_id = f"{self._camera.serial}-{description.key}"
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, self._camera.serial)},
name=camera,
manufacturer=DEFAULT_BRAND,
model=self._camera.camera_type,
)
def update(self):
"""Update sensor state."""
self.data.refresh()
state = self._camera.attributes[self.entity_description.key]
if self.entity_description.key == TYPE_BATTERY:
state = state != "ok"
self._attr_is_on = state