1
mirror of https://github.com/home-assistant/core synced 2024-09-03 08:14:07 +02:00
ha-core/homeassistant/components/blink/camera.py
2022-04-26 19:22:03 -10:00

98 lines
3.0 KiB
Python

"""Support for Blink system camera."""
from __future__ import annotations
import logging
from requests.exceptions import ChunkedEncodingError
from homeassistant.components.camera import Camera
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_platform
from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DEFAULT_BRAND, DOMAIN, SERVICE_TRIGGER
_LOGGER = logging.getLogger(__name__)
ATTR_VIDEO_CLIP = "video"
ATTR_IMAGE = "image"
async def async_setup_entry(
hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None:
"""Set up a Blink Camera."""
data = hass.data[DOMAIN][config.entry_id]
entities = [
BlinkCamera(data, name, camera) for name, camera in data.cameras.items()
]
async_add_entities(entities)
platform = entity_platform.async_get_current_platform()
platform.async_register_entity_service(SERVICE_TRIGGER, {}, "trigger_camera")
class BlinkCamera(Camera):
"""An implementation of a Blink Camera."""
def __init__(self, data, name, camera):
"""Initialize a camera."""
super().__init__()
self.data = data
self._attr_name = f"{DOMAIN} {name}"
self._camera = camera
self._attr_unique_id = f"{camera.serial}-camera"
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, camera.serial)},
name=name,
manufacturer=DEFAULT_BRAND,
model=camera.camera_type,
)
_LOGGER.debug("Initialized blink camera %s", self.name)
@property
def extra_state_attributes(self):
"""Return the camera attributes."""
return self._camera.attributes
def enable_motion_detection(self):
"""Enable motion detection for the camera."""
self._camera.arm = True
self.data.refresh()
def disable_motion_detection(self):
"""Disable motion detection for the camera."""
self._camera.arm = False
self.data.refresh()
@property
def motion_detection_enabled(self):
"""Return the state of the camera."""
return self._camera.arm
@property
def brand(self):
"""Return the camera brand."""
return DEFAULT_BRAND
def trigger_camera(self):
"""Trigger camera to take a snapshot."""
self._camera.snap_picture()
self.data.refresh()
def camera_image(
self, width: int | None = None, height: int | None = None
) -> bytes | None:
"""Return a still image response from the camera."""
try:
return self._camera.image_from_cache.content
except ChunkedEncodingError:
_LOGGER.debug("Could not retrieve image for %s", self._camera.name)
return None
except TypeError:
_LOGGER.debug("No cached image for %s", self._camera.name)
return None