59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
"""Simplepush notification service."""
|
|
from simplepush import send, send_encrypted
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.notify import (
|
|
ATTR_TITLE,
|
|
ATTR_TITLE_DEFAULT,
|
|
PLATFORM_SCHEMA,
|
|
BaseNotificationService,
|
|
)
|
|
from homeassistant.const import CONF_EVENT, CONF_PASSWORD
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
ATTR_ENCRYPTED = "encrypted"
|
|
|
|
CONF_DEVICE_KEY = "device_key"
|
|
CONF_SALT = "salt"
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_DEVICE_KEY): cv.string,
|
|
vol.Optional(CONF_EVENT): cv.string,
|
|
vol.Inclusive(CONF_PASSWORD, ATTR_ENCRYPTED): cv.string,
|
|
vol.Inclusive(CONF_SALT, ATTR_ENCRYPTED): cv.string,
|
|
}
|
|
)
|
|
|
|
|
|
def get_service(hass, config, discovery_info=None):
|
|
"""Get the Simplepush notification service."""
|
|
return SimplePushNotificationService(config)
|
|
|
|
|
|
class SimplePushNotificationService(BaseNotificationService):
|
|
"""Implementation of the notification service for Simplepush."""
|
|
|
|
def __init__(self, config):
|
|
"""Initialize the Simplepush notification service."""
|
|
self._device_key = config.get(CONF_DEVICE_KEY)
|
|
self._event = config.get(CONF_EVENT)
|
|
self._password = config.get(CONF_PASSWORD)
|
|
self._salt = config.get(CONF_SALT)
|
|
|
|
def send_message(self, message="", **kwargs):
|
|
"""Send a message to a Simplepush user."""
|
|
title = kwargs.get(ATTR_TITLE, ATTR_TITLE_DEFAULT)
|
|
|
|
if self._password:
|
|
send_encrypted(
|
|
self._device_key,
|
|
self._password,
|
|
self._salt,
|
|
title,
|
|
message,
|
|
event=self._event,
|
|
)
|
|
else:
|
|
send(self._device_key, title, message, event=self._event)
|