129 lines
3.8 KiB
Python
129 lines
3.8 KiB
Python
"""
|
|
Support for Arlo Alarm Control Panels.
|
|
|
|
For more details about this platform, please refer to the documentation at
|
|
https://home-assistant.io/components/alarm_control_panel.arlo/
|
|
"""
|
|
import asyncio
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.components.alarm_control_panel import (
|
|
AlarmControlPanel, PLATFORM_SCHEMA)
|
|
from homeassistant.components.arlo import (DATA_ARLO, CONF_ATTRIBUTION)
|
|
from homeassistant.const import (
|
|
ATTR_ATTRIBUTION, STATE_ALARM_ARMED_AWAY, STATE_ALARM_ARMED_HOME,
|
|
STATE_ALARM_DISARMED)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
ARMED = 'armed'
|
|
|
|
CONF_HOME_MODE_NAME = 'home_mode_name'
|
|
CONF_AWAY_MODE_NAME = 'away_mode_name'
|
|
|
|
DEPENDENCIES = ['arlo']
|
|
|
|
DISARMED = 'disarmed'
|
|
|
|
ICON = 'mdi:security'
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
|
vol.Optional(CONF_HOME_MODE_NAME, default=ARMED): cv.string,
|
|
vol.Optional(CONF_AWAY_MODE_NAME, default=ARMED): cv.string,
|
|
})
|
|
|
|
|
|
@asyncio.coroutine
|
|
def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
|
|
"""Set up the Arlo Alarm Control Panels."""
|
|
data = hass.data[DATA_ARLO]
|
|
|
|
if not data.base_stations:
|
|
return
|
|
|
|
home_mode_name = config.get(CONF_HOME_MODE_NAME)
|
|
away_mode_name = config.get(CONF_AWAY_MODE_NAME)
|
|
base_stations = []
|
|
for base_station in data.base_stations:
|
|
base_stations.append(ArloBaseStation(base_station, home_mode_name,
|
|
away_mode_name))
|
|
async_add_devices(base_stations, True)
|
|
|
|
|
|
class ArloBaseStation(AlarmControlPanel):
|
|
"""Representation of an Arlo Alarm Control Panel."""
|
|
|
|
def __init__(self, data, home_mode_name, away_mode_name):
|
|
"""Initialize the alarm control panel."""
|
|
self._base_station = data
|
|
self._home_mode_name = home_mode_name
|
|
self._away_mode_name = away_mode_name
|
|
self._state = None
|
|
|
|
@property
|
|
def icon(self):
|
|
"""Return icon."""
|
|
return ICON
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the device."""
|
|
return self._state
|
|
|
|
def update(self):
|
|
"""Update the state of the device."""
|
|
# PyArlo sometimes returns None for mode. So retry 3 times before
|
|
# returning None.
|
|
num_retries = 3
|
|
i = 0
|
|
while i < num_retries:
|
|
mode = self._base_station.mode
|
|
if mode:
|
|
self._state = self._get_state_from_mode(mode)
|
|
return
|
|
i += 1
|
|
self._state = None
|
|
|
|
@asyncio.coroutine
|
|
def async_alarm_disarm(self, code=None):
|
|
"""Send disarm command."""
|
|
self._base_station.mode = DISARMED
|
|
|
|
@asyncio.coroutine
|
|
def async_alarm_arm_away(self, code=None):
|
|
"""Send arm away command. Uses custom mode."""
|
|
self._base_station.mode = self._away_mode_name
|
|
|
|
@asyncio.coroutine
|
|
def async_alarm_arm_home(self, code=None):
|
|
"""Send arm home command. Uses custom mode."""
|
|
self._base_station.mode = self._home_mode_name
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the base station."""
|
|
return self._base_station.name
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the state attributes."""
|
|
return {
|
|
ATTR_ATTRIBUTION: CONF_ATTRIBUTION,
|
|
'device_id': self._base_station.device_id
|
|
}
|
|
|
|
def _get_state_from_mode(self, mode):
|
|
"""Convert Arlo mode to Home Assistant state."""
|
|
if mode == ARMED:
|
|
return STATE_ALARM_ARMED_AWAY
|
|
elif mode == DISARMED:
|
|
return STATE_ALARM_DISARMED
|
|
elif mode == self._home_mode_name:
|
|
return STATE_ALARM_ARMED_HOME
|
|
elif mode == self._away_mode_name:
|
|
return STATE_ALARM_ARMED_AWAY
|
|
return None
|