alarm_control_panel.py

Go to the documentation of this file.

"""Support for Agent DVR Alarm Control Panels."""
 
from __future__ import annotations
 
from homeassistant.components.alarm_control_panel import (
    AlarmControlPanelEntity,
    AlarmControlPanelEntityFeature,
    AlarmControlPanelState,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from . import AgentDVRConfigEntry
from .const import DOMAIN as AGENT_DOMAIN
 
CONF_HOME_MODE_NAME = "home"
CONF_AWAY_MODE_NAME = "away"
CONF_NIGHT_MODE_NAME = "night"
 
CONST_ALARM_CONTROL_PANEL_NAME = "Alarm Panel"
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: AgentDVRConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up the Agent DVR Alarm Control Panels."""
    async_add_entities([AgentBaseStation(config_entry.runtime_data)])
 
 
class AgentBaseStation(AlarmControlPanelEntity):
    """Representation of an Agent DVR Alarm Control Panel."""
 
    _attr_supported_features = (
        AlarmControlPanelEntityFeature.ARM_HOME
        | AlarmControlPanelEntityFeature.ARM_AWAY
        | AlarmControlPanelEntityFeature.ARM_NIGHT
    )
    _attr_code_arm_required = False
    _attr_has_entity_name = True
    _attr_name = None
 
    def __init__(self, client):
        """Initialize the alarm control panel."""
        self._client_client = client
        self._attr_unique_id_attr_unique_id = f"{client.unique}_CP"
        self._attr_device_info_attr_device_info = DeviceInfo(
            identifiers={(AGENT_DOMAIN, client.unique)},
            name=f"{client.name} {CONST_ALARM_CONTROL_PANEL_NAME}",
            manufacturer="Agent",
            model=CONST_ALARM_CONTROL_PANEL_NAME,
            sw_version=client.version,
        )
 
    async def async_update(self) -> None:
        """Update the state of the device."""
        await self._client_client.update()
        self._attr_available_attr_available = self._client_client.is_available
        armed = self._client_client.is_armed
        if armed is None:
            self._attr_alarm_state_attr_alarm_state = None
            return
        if armed:
            prof = (await self._client_client.get_active_profile()).lower()
            self._attr_alarm_state_attr_alarm_state = AlarmControlPanelState.ARMED_AWAY
            if prof == CONF_HOME_MODE_NAME:
                self._attr_alarm_state_attr_alarm_state = AlarmControlPanelState.ARMED_HOME
            elif prof == CONF_NIGHT_MODE_NAME:
                self._attr_alarm_state_attr_alarm_state = AlarmControlPanelState.ARMED_NIGHT
        else:
            self._attr_alarm_state_attr_alarm_state = AlarmControlPanelState.DISARMED
 
    async def async_alarm_disarm(self, code: str | None = None) -> None:
        """Send disarm command."""
        await self._client_client.disarm()
        self._attr_alarm_state_attr_alarm_state = AlarmControlPanelState.DISARMED
 
    async def async_alarm_arm_away(self, code: str | None = None) -> None:
        """Send arm away command. Uses custom mode."""
        await self._client_client.arm()
        await self._client_client.set_active_profile(CONF_AWAY_MODE_NAME)
        self._attr_alarm_state_attr_alarm_state = AlarmControlPanelState.ARMED_AWAY
 
    async def async_alarm_arm_home(self, code: str | None = None) -> None:
        """Send arm home command. Uses custom mode."""
        await self._client_client.arm()
        await self._client_client.set_active_profile(CONF_HOME_MODE_NAME)
        self._attr_alarm_state_attr_alarm_state = AlarmControlPanelState.ARMED_HOME
 
    async def async_alarm_arm_night(self, code: str | None = None) -> None:
        """Send arm night command. Uses custom mode."""
        await self._client_client.arm()
        await self._client_client.set_active_profile(CONF_NIGHT_MODE_NAME)
        self._attr_alarm_state_attr_alarm_state = AlarmControlPanelState.ARMED_NIGHT