70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
|
"""
|
||
|
|
Join platform for notify component.
|
||
|
|
|
||
|
|
For more details about this platform, please refer to the documentation at
|
||
|
https://home-assistant.io/components/notify.join/
|
||
|
|
"""
|
||
|
|
import logging
|
||
|
|
import voluptuous as vol
|
||
|
|
from homeassistant.components.notify import (
|
||
|
ATTR_DATA, ATTR_TITLE, ATTR_TITLE_DEFAULT, PLATFORM_SCHEMA,
|
||
|
|
BaseNotificationService)
|
||
|
|
from homeassistant.const import CONF_API_KEY
|
||
|
|
import homeassistant.helpers.config_validation as cv
|
||
|
|
|
||
|
|
REQUIREMENTS = ['python-join-api==0.0.2']
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
CONF_DEVICE_ID = 'device_id'
|
||
|
|
CONF_DEVICE_IDS = 'device_ids'
|
||
|
|
CONF_DEVICE_NAMES = 'device_names'
|
||
|
|
|
||
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
||
|
|
vol.Required(CONF_API_KEY): cv.string,
|
||
|
|
vol.Optional(CONF_DEVICE_ID): cv.string,
|
||
|
|
vol.Optional(CONF_DEVICE_IDS): cv.string,
|
||
|
|
vol.Optional(CONF_DEVICE_NAMES): cv.string,
|
||
|
|
})
|
||
|
|
|
||
|
|
|
||
|
def get_service(hass, config, discovery_info=None):
|
||
|
|
"""Get the Join notification service."""
|
||
|
|
api_key = config.get(CONF_API_KEY)
|
||
|
|
device_id = config.get(CONF_DEVICE_ID)
|
||
|
|
device_ids = config.get(CONF_DEVICE_IDS)
|
||
|
|
device_names = config.get(CONF_DEVICE_NAMES)
|
||
|
|
if api_key:
|
||
|
|
from pyjoin import get_devices
|
||
|
|
if not get_devices(api_key):
|
||
|
|
_LOGGER.error("Error connecting to Join. Check the API key")
|
||
|
|
return False
|
||
|
|
if device_id is None and device_ids is None and device_names is None:
|
||
|
|
_LOGGER.error("No device was provided. Please specify device_id"
|
||
|
|
", device_ids, or device_names")
|
||
|
|
return False
|
||
|
|
return JoinNotificationService(api_key, device_id,
|
||
|
|
device_ids, device_names)
|
||
|
|
|
||
|
|
|
||
|
|
class JoinNotificationService(BaseNotificationService):
|
||
|
|
"""Implement the notification service for Join."""
|
||
|
|
|
||
|
|
def __init__(self, api_key, device_id, device_ids, device_names):
|
||
|
|
"""Initialize the service."""
|
||
|
|
self._api_key = api_key
|
||
|
|
self._device_id = device_id
|
||
|
|
self._device_ids = device_ids
|
||
|
|
self._device_names = device_names
|
||
|
|
|
||
|
|
def send_message(self, message="", **kwargs):
|
||
|
|
"""Send a message to a user."""
|
||
|
|
from pyjoin import send_notification
|
||
|
title = kwargs.get(ATTR_TITLE, ATTR_TITLE_DEFAULT)
|
||
|
|
data = kwargs.get(ATTR_DATA) or {}
|
||
|
|
send_notification(
|
||
|
|
device_id=self._device_id, device_ids=self._device_ids,
|
||
|
|
device_names=self._device_names, text=message, title=title,
|
||
|
|
icon=data.get('icon'), smallicon=data.get('smallicon'),
|
||
|
|
vibration=data.get('vibration'), api_key=self._api_key)
|