86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
"""Support for ZoneMinder camera streaming."""
|
|
import logging
|
|
from typing import Callable, List, Optional
|
|
|
|
from zoneminder.monitor import Monitor
|
|
|
|
from homeassistant.components.mjpeg.camera import (
|
|
CONF_MJPEG_URL,
|
|
CONF_STILL_IMAGE_URL,
|
|
MjpegCamera,
|
|
filter_urllib3_logging,
|
|
)
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_NAME, CONF_VERIFY_SSL
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from .common import get_client_from_data
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the ZoneMinder cameras."""
|
|
filter_urllib3_logging()
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: ConfigEntry,
|
|
async_add_entities: Callable[[List[Entity], Optional[bool]], None],
|
|
) -> None:
|
|
"""Set up the sensor config entry."""
|
|
zm_client = get_client_from_data(hass, config_entry.unique_id)
|
|
|
|
async_add_entities(
|
|
[
|
|
ZoneMinderCamera(monitor, zm_client.verify_ssl, config_entry)
|
|
for monitor in await hass.async_add_job(zm_client.get_monitors)
|
|
]
|
|
)
|
|
|
|
|
|
class ZoneMinderCamera(MjpegCamera):
|
|
"""Representation of a ZoneMinder Monitor Stream."""
|
|
|
|
def __init__(self, monitor: Monitor, verify_ssl: bool, config_entry: ConfigEntry):
|
|
"""Initialize as a subclass of MjpegCamera."""
|
|
device_info = {
|
|
CONF_NAME: monitor.name,
|
|
CONF_MJPEG_URL: monitor.mjpeg_image_url,
|
|
CONF_STILL_IMAGE_URL: monitor.still_image_url,
|
|
CONF_VERIFY_SSL: verify_ssl,
|
|
}
|
|
super().__init__(device_info)
|
|
self._is_recording = None
|
|
self._is_available = None
|
|
self._monitor = monitor
|
|
self._config_entry = config_entry
|
|
|
|
@property
|
|
def unique_id(self) -> Optional[str]:
|
|
"""Return a unique ID."""
|
|
return f"{self._config_entry.unique_id}_{self._monitor.id}_camera"
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""Update the recording state periodically."""
|
|
return True
|
|
|
|
def update(self):
|
|
"""Update our recording state from the ZM API."""
|
|
_LOGGER.debug("Updating camera state for monitor %i", self._monitor.id)
|
|
self._is_recording = self._monitor.is_recording
|
|
self._is_available = self._monitor.is_available
|
|
|
|
@property
|
|
def is_recording(self):
|
|
"""Return whether the monitor is in alarm mode."""
|
|
return self._is_recording
|
|
|
|
@property
|
|
def available(self):
|
|
"""Return True if entity is available."""
|
|
return self._is_available
|