core/homeassistant/components/alarm_control_panel/verisure.py

"""
homeassistant.components.alarm_control_panel.verisure
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Interfaces with Verisure alarm control panel.
"""
import logging

import homeassistant.components.verisure as verisure
import homeassistant.components.alarm_control_panel as alarm

from homeassistant.const import (
    STATE_UNKNOWN,
    STATE_ALARM_DISARMED, STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMED_AWAY)

_LOGGER = logging.getLogger(__name__)


def setup_platform(hass, config, add_devices, discovery_info=None):
    """ Sets up the Verisure platform. """

    if not verisure.MY_PAGES:
        _LOGGER.error('A connection has not been made to Verisure mypages.')
        return False

    alarms = []

    alarms.extend([
        VerisureAlarm(value)
        for value in verisure.get_alarm_status().values()
        if verisure.SHOW_ALARM
        ])

    add_devices(alarms)


class VerisureAlarm(alarm.AlarmControlPanel):
    """ represents a Verisure alarm status within home assistant. """

    def __init__(self, alarm_status):
        self._id = alarm_status.id
        self._device = verisure.MY_PAGES.DEVICE_ALARM
        self._state = STATE_UNKNOWN

    @property
    def name(self):
        """ Returns the name of the device. """
        return 'Alarm {}'.format(self._id)

    @property
    def state(self):
        """ Returns the state of the device. """
        return self._state

    @property
    def code_format(self):
        """ Four digit code required"""
        return '^\\d{4}$'

    def update(self):
        ''' update alarm status '''
        verisure.update()

        if verisure.STATUS[self._device][self._id].status == 'unarmed':
            self._state = STATE_ALARM_DISARMED
        elif verisure.STATUS[self._device][self._id].status == 'armedhome':
            self._state = STATE_ALARM_ARMED_HOME
        elif verisure.STATUS[self._device][self._id].status == 'armedaway':
            self._state = STATE_ALARM_ARMED_AWAY
        elif verisure.STATUS[self._device][self._id].status != 'pending':
            _LOGGER.error(
                'Unknown alarm state %s',
                verisure.STATUS[self._device][self._id].status)

    def alarm_disarm(self, code=None):
        """ Send disarm command. """
        verisure.MY_PAGES.set_alarm_status(
            code,
            verisure.MY_PAGES.ALARM_DISARMED)
        _LOGGER.warning('disarming')

    def alarm_arm_home(self, code=None):
        """ Send arm home command. """
        verisure.MY_PAGES.set_alarm_status(
            code,
            verisure.MY_PAGES.ALARM_ARMED_HOME)
        _LOGGER.warning('arming home')

    def alarm_arm_away(self, code=None):
        """ Send arm away command. """
        verisure.MY_PAGES.set_alarm_status(
            code,
            verisure.MY_PAGES.ALARM_ARMED_AWAY)
        _LOGGER.warning('arming away')

    def alarm_trigger(self, code=None):
        """ Send alarm trigger command. No code needed. """
        return