jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 06:07:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added headers configuration variable to notify.rest component (#13674)

Browse Source 
* Added headers configuration variable to notify.rest component

* Fix code style
This commit is contained in:
Oleg 2018-04-04 17:35:33 +03:00 committed by Fabian Affolter
parent 4b2fdd243a
commit 9ce02d2717
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
homeassistant/components/notify/rest.py
View File

@ -12,7 +12,8 @@ import voluptuous as vol
from homeassistant.components.notify import ( from homeassistant.components.notify import (
ATTR_TARGET, ATTR_TITLE, ATTR_TITLE_DEFAULT, BaseNotificationService, ATTR_TARGET, ATTR_TITLE, ATTR_TITLE_DEFAULT, BaseNotificationService,
PLATFORM_SCHEMA) PLATFORM_SCHEMA)
from homeassistant.const import (CONF_RESOURCE, CONF_METHOD, CONF_NAME) from homeassistant.const import (CONF_RESOURCE, CONF_METHOD, CONF_NAME,
CONF_HEADERS)
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
CONF_DATA = 'data' CONF_DATA = 'data'
@ -29,6 +30,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
default=DEFAULT_MESSAGE_PARAM_NAME): cv.string, default=DEFAULT_MESSAGE_PARAM_NAME): cv.string,
vol.Optional(CONF_METHOD, default=DEFAULT_METHOD): vol.Optional(CONF_METHOD, default=DEFAULT_METHOD):
vol.In(['POST', 'GET', 'POST_JSON']), vol.In(['POST', 'GET', 'POST_JSON']),
vol.Optional(CONF_HEADERS): vol.Schema({cv.string: cv.string}),
vol.Optional(CONF_NAME): cv.string, vol.Optional(CONF_NAME): cv.string,
vol.Optional(CONF_TARGET_PARAMETER_NAME): cv.string, vol.Optional(CONF_TARGET_PARAMETER_NAME): cv.string,
vol.Optional(CONF_TITLE_PARAMETER_NAME): cv.string, vol.Optional(CONF_TITLE_PARAMETER_NAME): cv.string,
@ -43,6 +45,7 @@ def get_service(hass, config, discovery_info=None):
"""Get the RESTful notification service.""" """Get the RESTful notification service."""
resource = config.get(CONF_RESOURCE) resource = config.get(CONF_RESOURCE)
method = config.get(CONF_METHOD) method = config.get(CONF_METHOD)
headers = config.get(CONF_HEADERS)
message_param_name = config.get(CONF_MESSAGE_PARAMETER_NAME) message_param_name = config.get(CONF_MESSAGE_PARAMETER_NAME)
title_param_name = config.get(CONF_TITLE_PARAMETER_NAME) title_param_name = config.get(CONF_TITLE_PARAMETER_NAME)
target_param_name = config.get(CONF_TARGET_PARAMETER_NAME) target_param_name = config.get(CONF_TARGET_PARAMETER_NAME)
@ -50,19 +53,20 @@ def get_service(hass, config, discovery_info=None):
data_template = config.get(CONF_DATA_TEMPLATE) data_template = config.get(CONF_DATA_TEMPLATE)
return RestNotificationService( return RestNotificationService(
hass, resource, method, message_param_name, hass, resource, method, headers, message_param_name,
title_param_name, target_param_name, data, data_template) title_param_name, target_param_name, data, data_template)
class RestNotificationService(BaseNotificationService): class RestNotificationService(BaseNotificationService):
"""Implementation of a notification service for REST.""" """Implementation of a notification service for REST."""
def __init__(self, hass, resource, method, message_param_name, def __init__(self, hass, resource, method, headers, message_param_name,
title_param_name, target_param_name, data, data_template): title_param_name, target_param_name, data, data_template):
"""Initialize the service.""" """Initialize the service."""
self._resource = resource self._resource = resource
self._hass = hass self._hass = hass
self._method = method.upper() self._method = method.upper()
self._headers = headers
self._message_param_name = message_param_name self._message_param_name = message_param_name
self._title_param_name = title_param_name self._title_param_name = title_param_name
self._target_param_name = target_param_name self._target_param_name = target_param_name
@ -99,11 +103,14 @@ class RestNotificationService(BaseNotificationService):
data.update(_data_template_creator(self._data_template)) data.update(_data_template_creator(self._data_template))
if self._method == 'POST': if self._method == 'POST':
response = requests.post(self._resource, data=data, timeout=10) response = requests.post(self._resource, headers=self._headers,
data=data, timeout=10)
elif self._method == 'POST_JSON': elif self._method == 'POST_JSON':
response = requests.post(self._resource, json=data, timeout=10) response = requests.post(self._resource, headers=self._headers,
json=data, timeout=10)
else: # default GET else: # default GET
response = requests.get(self._resource, params=data, timeout=10) response = requests.get(self._resource, headers=self._headers,
params=data, timeout=10)
if response.status_code not in (200, 201): if response.status_code not in (200, 201):
_LOGGER.exception( _LOGGER.exception(