commit
82c5e2cf3c
|
@ -106,6 +106,7 @@ omit =
|
||||||
homeassistant/components/notify/pushbullet.py
|
homeassistant/components/notify/pushbullet.py
|
||||||
homeassistant/components/notify/pushetta.py
|
homeassistant/components/notify/pushetta.py
|
||||||
homeassistant/components/notify/pushover.py
|
homeassistant/components/notify/pushover.py
|
||||||
|
homeassistant/components/notify/rest.py
|
||||||
homeassistant/components/notify/slack.py
|
homeassistant/components/notify/slack.py
|
||||||
homeassistant/components/notify/smtp.py
|
homeassistant/components/notify/smtp.py
|
||||||
homeassistant/components/notify/syslog.py
|
homeassistant/components/notify/syslog.py
|
||||||
|
|
|
@ -0,0 +1,80 @@
|
||||||
|
"""
|
||||||
|
homeassistant.components.notify.rest
|
||||||
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||||
|
REST platform for notify component.
|
||||||
|
|
||||||
|
For more details about this platform, please refer to the documentation at
|
||||||
|
https://home-assistant.io/components/notify.rest/
|
||||||
|
"""
|
||||||
|
import logging
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from homeassistant.helpers import validate_config
|
||||||
|
from homeassistant.components.notify import (
|
||||||
|
DOMAIN, ATTR_TITLE, ATTR_TARGET, BaseNotificationService)
|
||||||
|
|
||||||
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
DEFAULT_METHOD = 'GET'
|
||||||
|
DEFAULT_MESSAGE_PARAM_NAME = 'message'
|
||||||
|
DEFAULT_TITLE_PARAM_NAME = None
|
||||||
|
DEFAULT_TARGET_PARAM_NAME = None
|
||||||
|
|
||||||
|
|
||||||
|
def get_service(hass, config):
|
||||||
|
""" Get the REST notification service. """
|
||||||
|
|
||||||
|
if not validate_config({DOMAIN: config},
|
||||||
|
{DOMAIN: ['resource', ]},
|
||||||
|
_LOGGER):
|
||||||
|
return None
|
||||||
|
|
||||||
|
method = config.get('method', DEFAULT_METHOD)
|
||||||
|
message_param_name = config.get('message_param_name',
|
||||||
|
DEFAULT_MESSAGE_PARAM_NAME)
|
||||||
|
title_param_name = config.get('title_param_name',
|
||||||
|
DEFAULT_TITLE_PARAM_NAME)
|
||||||
|
target_param_name = config.get('target_param_name',
|
||||||
|
DEFAULT_TARGET_PARAM_NAME)
|
||||||
|
|
||||||
|
return RestNotificationService(config['resource'], method,
|
||||||
|
message_param_name, title_param_name,
|
||||||
|
target_param_name)
|
||||||
|
|
||||||
|
|
||||||
|
# pylint: disable=too-few-public-methods, too-many-arguments
|
||||||
|
class RestNotificationService(BaseNotificationService):
|
||||||
|
""" Implements notification service for REST. """
|
||||||
|
|
||||||
|
def __init__(self, resource, method, message_param_name,
|
||||||
|
title_param_name, target_param_name):
|
||||||
|
self._resource = resource
|
||||||
|
self._method = method.upper()
|
||||||
|
self._message_param_name = message_param_name
|
||||||
|
self._title_param_name = title_param_name
|
||||||
|
self._target_param_name = target_param_name
|
||||||
|
|
||||||
|
def send_message(self, message="", **kwargs):
|
||||||
|
""" Send a message to a user. """
|
||||||
|
|
||||||
|
data = {
|
||||||
|
self._message_param_name: message
|
||||||
|
}
|
||||||
|
|
||||||
|
if self._title_param_name is not None:
|
||||||
|
data[self._title_param_name] = kwargs.get(ATTR_TITLE)
|
||||||
|
|
||||||
|
if self._target_param_name is not None:
|
||||||
|
data[self._title_param_name] = kwargs.get(ATTR_TARGET)
|
||||||
|
|
||||||
|
if self._method == 'POST':
|
||||||
|
response = requests.post(self._resource, data=data, timeout=10)
|
||||||
|
elif self._method == 'POST_JSON':
|
||||||
|
response = requests.post(self._resource, json=data, timeout=10)
|
||||||
|
else: # default GET
|
||||||
|
response = requests.get(self._resource, params=data, timeout=10)
|
||||||
|
|
||||||
|
if response.status_code not in (200, 201):
|
||||||
|
_LOGGER.exception(
|
||||||
|
"Error sending message. Response %d: %s:",
|
||||||
|
response.status_code, response.reason)
|
Loading…
Reference in New Issue