123 lines
4.4 KiB
Python
123 lines
4.4 KiB
Python
"""
|
|
Support for Asterisk Voicemail interface.
|
|
|
|
For more details about this component, please refer to the documentation at
|
|
https://home-assistant.io/components/asterisk_mbox/
|
|
"""
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers import discovery
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.dispatcher import (
|
|
async_dispatcher_send, dispatcher_connect)
|
|
|
|
REQUIREMENTS = ['asterisk_mbox==0.5.0']
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DOMAIN = 'asterisk_mbox'
|
|
|
|
SIGNAL_DISCOVER_PLATFORM = "asterisk_mbox.discover_platform"
|
|
SIGNAL_MESSAGE_REQUEST = 'asterisk_mbox.message_request'
|
|
SIGNAL_MESSAGE_UPDATE = 'asterisk_mbox.message_updated'
|
|
SIGNAL_CDR_UPDATE = 'asterisk_mbox.message_updated'
|
|
SIGNAL_CDR_REQUEST = 'asterisk_mbox.message_request'
|
|
|
|
CONFIG_SCHEMA = vol.Schema({
|
|
DOMAIN: vol.Schema({
|
|
vol.Required(CONF_HOST): cv.string,
|
|
vol.Required(CONF_PASSWORD): cv.string,
|
|
vol.Required(CONF_PORT): cv.port,
|
|
}),
|
|
}, extra=vol.ALLOW_EXTRA)
|
|
|
|
|
|
def setup(hass, config):
|
|
"""Set up for the Asterisk Voicemail box."""
|
|
conf = config.get(DOMAIN)
|
|
|
|
host = conf.get(CONF_HOST)
|
|
port = conf.get(CONF_PORT)
|
|
password = conf.get(CONF_PASSWORD)
|
|
|
|
hass.data[DOMAIN] = AsteriskData(hass, host, port, password, config)
|
|
|
|
return True
|
|
|
|
|
|
class AsteriskData:
|
|
"""Store Asterisk mailbox data."""
|
|
|
|
def __init__(self, hass, host, port, password, config):
|
|
"""Init the Asterisk data object."""
|
|
from asterisk_mbox import Client as asteriskClient
|
|
self.hass = hass
|
|
self.config = config
|
|
self.messages = None
|
|
self.cdr = None
|
|
|
|
dispatcher_connect(
|
|
self.hass, SIGNAL_MESSAGE_REQUEST, self._request_messages)
|
|
dispatcher_connect(
|
|
self.hass, SIGNAL_CDR_REQUEST, self._request_cdr)
|
|
dispatcher_connect(
|
|
self.hass, SIGNAL_DISCOVER_PLATFORM, self._discover_platform)
|
|
# Only connect after signal connection to ensure we don't miss any
|
|
self.client = asteriskClient(host, port, password, self.handle_data)
|
|
|
|
@callback
|
|
def _discover_platform(self, component):
|
|
_LOGGER.debug("Adding mailbox %s", component)
|
|
self.hass.async_create_task(discovery.async_load_platform(
|
|
self.hass, "mailbox", component, {}, self.config))
|
|
|
|
@callback
|
|
def handle_data(self, command, msg):
|
|
"""Handle changes to the mailbox."""
|
|
from asterisk_mbox.commands import (CMD_MESSAGE_LIST,
|
|
CMD_MESSAGE_CDR_AVAILABLE,
|
|
CMD_MESSAGE_CDR)
|
|
|
|
if command == CMD_MESSAGE_LIST:
|
|
_LOGGER.debug("AsteriskVM sent updated message list: Len %d",
|
|
len(msg))
|
|
old_messages = self.messages
|
|
self.messages = sorted(
|
|
msg, key=lambda item: item['info']['origtime'], reverse=True)
|
|
if not isinstance(old_messages, list):
|
|
async_dispatcher_send(self.hass, SIGNAL_DISCOVER_PLATFORM,
|
|
DOMAIN)
|
|
async_dispatcher_send(self.hass, SIGNAL_MESSAGE_UPDATE,
|
|
self.messages)
|
|
elif command == CMD_MESSAGE_CDR:
|
|
_LOGGER.debug("AsteriskVM sent updated CDR list: Len %d",
|
|
len(msg.get('entries', [])))
|
|
self.cdr = msg['entries']
|
|
async_dispatcher_send(self.hass, SIGNAL_CDR_UPDATE, self.cdr)
|
|
elif command == CMD_MESSAGE_CDR_AVAILABLE:
|
|
if not isinstance(self.cdr, list):
|
|
_LOGGER.debug("AsteriskVM adding CDR platform")
|
|
self.cdr = []
|
|
async_dispatcher_send(self.hass, SIGNAL_DISCOVER_PLATFORM,
|
|
"asterisk_cdr")
|
|
async_dispatcher_send(self.hass, SIGNAL_CDR_REQUEST)
|
|
else:
|
|
_LOGGER.debug("AsteriskVM sent unknown message '%d' len: %d",
|
|
command, len(msg))
|
|
|
|
@callback
|
|
def _request_messages(self):
|
|
"""Handle changes to the mailbox."""
|
|
_LOGGER.debug("Requesting message list")
|
|
self.client.messages()
|
|
|
|
@callback
|
|
def _request_cdr(self):
|
|
"""Handle changes to the CDR."""
|
|
_LOGGER.debug("Requesting CDR list")
|
|
self.client.get_cdr()
|