sensor.py

Go to the documentation of this file.

"""Support for ZoneMinder sensors."""
 
from __future__ import annotations
 
import logging
 
import voluptuous as vol
from zoneminder.monitor import Monitor, TimePeriod
from zoneminder.zm import ZoneMinder
 
from homeassistant.components.sensor import (
    PLATFORM_SCHEMA as SENSOR_PLATFORM_SCHEMA,
    SensorEntity,
    SensorEntityDescription,
)
from homeassistant.const import CONF_MONITORED_CONDITIONS
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
from . import DOMAIN as ZONEMINDER_DOMAIN
 
_LOGGER = logging.getLogger(__name__)
 
CONF_INCLUDE_ARCHIVED = "include_archived"
 
DEFAULT_INCLUDE_ARCHIVED = False
 
SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
    SensorEntityDescription(
        key="all",
        name="Events",
    ),
    SensorEntityDescription(
        key="hour",
        name="Events Last Hour",
    ),
    SensorEntityDescription(
        key="day",
        name="Events Last Day",
    ),
    SensorEntityDescription(
        key="week",
        name="Events Last Week",
    ),
    SensorEntityDescription(
        key="month",
        name="Events Last Month",
    ),
)
 
SENSOR_KEYS: list[str] = [desc.key for desc in SENSOR_TYPES]
 
PLATFORM_SCHEMA = SENSOR_PLATFORM_SCHEMA.extend(
    {
        vol.Optional(
            CONF_INCLUDE_ARCHIVED, default=DEFAULT_INCLUDE_ARCHIVED
        ): cv.boolean,
        vol.Optional(CONF_MONITORED_CONDITIONS, default=["all"]): vol.All(
            cv.ensure_list, [vol.In(SENSOR_KEYS)]
        ),
    }
)
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up the ZoneMinder sensor platform."""
    include_archived = config[CONF_INCLUDE_ARCHIVED]
    monitored_conditions = config[CONF_MONITORED_CONDITIONS]
 
    sensors: list[SensorEntity] = []
    zm_client: ZoneMinder
    for zm_client in hass.data[ZONEMINDER_DOMAIN].values():
        if not (monitors := zm_client.get_monitors()):
            raise PlatformNotReady(
                "Sensor could not fetch any monitors from ZoneMinder"
            )
 
        for monitor in monitors:
            sensors.append(ZMSensorMonitors(monitor))
 
            sensors.extend(
                [
                    ZMSensorEvents(monitor, include_archived, description)
                    for description in SENSOR_TYPES
                    if description.key in monitored_conditions
                ]
            )
 
        sensors.append(ZMSensorRunState(zm_client))
    add_entities(sensors)
 
 
class ZMSensorMonitors(SensorEntity):
    """Get the status of each ZoneMinder monitor."""
 
    def __init__(self, monitor: Monitor) -> None:
        """Initialize monitor sensor."""
        self._monitor_monitor = monitor
        self._attr_available_attr_available = False
        self._attr_name_attr_name = f"{self._monitor.name} Status"
 
    def update(self) -> None:
        """Update the sensor."""
        if not (state := self._monitor_monitor.function):
            self._attr_native_value_attr_native_value = None
        else:
            self._attr_native_value_attr_native_value = state.value
        self._attr_available_attr_available = self._monitor_monitor.is_available
 
 
class ZMSensorEvents(SensorEntity):
    """Get the number of events for each monitor."""
 
    _attr_native_unit_of_measurement = "Events"
 
    def __init__(
        self,
        monitor: Monitor,
        include_archived: bool,
        description: SensorEntityDescription,
    ) -> None:
        """Initialize event sensor."""
        self.entity_descriptionentity_description = description
 
        self._monitor_monitor = monitor
        self._include_archived_include_archived = include_archived
        self.time_periodtime_period = TimePeriod.get_time_period(description.key)
        self._attr_name_attr_name = f"{monitor.name} {self.time_period.title}"
 
    def update(self) -> None:
        """Update the sensor."""
        self._attr_native_value_attr_native_value = self._monitor_monitor.get_events(
            self.time_periodtime_period, self._include_archived_include_archived
        )
 
 
class ZMSensorRunState(SensorEntity):
    """Get the ZoneMinder run state."""
 
    _attr_name = "Run State"
 
    def __init__(self, client: ZoneMinder) -> None:
        """Initialize run state sensor."""
        self._attr_available_attr_available = False
        self._client_client = client
 
    def update(self) -> None:
        """Update the sensor."""
        self._attr_native_value_attr_native_value = self._client_client.get_active_state()
        self._attr_available_attr_available = self._client_client.is_available