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

Add timeout (#3304)

Browse Source
This commit is contained in:
Fabian Affolter 2016-09-11 09:21:01 +02:00 committed by GitHub
parent 9bbe7be684
commit 3f4d30c8da
1 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
homeassistant/components/notify/instapush.py
View File

@ -15,11 +15,15 @@ from homeassistant.components.notify import (
ATTR_TITLE, ATTR_TITLE_DEFAULT, PLATFORM_SCHEMA, BaseNotificationService) ATTR_TITLE, ATTR_TITLE_DEFAULT, PLATFORM_SCHEMA, BaseNotificationService)
from homeassistant.const import CONF_API_KEY from homeassistant.const import CONF_API_KEY
_LOGGER = logging.getLogger(__name__)
_RESOURCE = 'https://api.instapush.im/v1/'
CONF_APP_SECRET = 'app_secret' CONF_APP_SECRET = 'app_secret'
CONF_EVENT = 'event' CONF_EVENT = 'event'
CONF_TRACKER = 'tracker' CONF_TRACKER = 'tracker'
DEFAULT_TIMEOUT = 10
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_API_KEY): cv.string, vol.Required(CONF_API_KEY): cv.string,
vol.Required(CONF_APP_SECRET): cv.string, vol.Required(CONF_APP_SECRET): cv.string,
@ -28,18 +32,15 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
}) })
_LOGGER = logging.getLogger(__name__)
_RESOURCE = 'https://api.instapush.im/v1/'
def get_service(hass, config): def get_service(hass, config):
"""Get the instapush notification service.""" """Get the Instapush notification service."""
headers = {'x-instapush-appid': config[CONF_API_KEY], headers = {'x-instapush-appid': config[CONF_API_KEY],
'x-instapush-appsecret': config[CONF_APP_SECRET]} 'x-instapush-appsecret': config[CONF_APP_SECRET]}
try: try:
response = requests.get(_RESOURCE + 'events/list', response = requests.get(
headers=headers).json() '{}{}'.format(_RESOURCE, 'events/list'), headers=headers,
timeout=DEFAULT_TIMEOUT).json()
except ValueError: except ValueError:
_LOGGER.error('Unexpected answer from Instapush API.') _LOGGER.error('Unexpected answer from Instapush API.')
return None return None
@ -50,19 +51,18 @@ def get_service(hass, config):
if len([app for app in response if len([app for app in response
if app['title'] == config[CONF_EVENT]]) == 0: if app['title'] == config[CONF_EVENT]]) == 0:
_LOGGER.error( _LOGGER.error("No app match your given value. "
"No app match your given value. " "Please create an app at https://instapush.im")
"Please create an app at https://instapush.im")
return None return None
return InstapushNotificationService( return InstapushNotificationService(
config[CONF_API_KEY], config[CONF_APP_SECRET], config[CONF_EVENT], config.get(CONF_API_KEY), config.get(CONF_APP_SECRET),
config[CONF_TRACKER]) config.get(CONF_EVENT), config.get(CONF_TRACKER))
# pylint: disable=too-few-public-methods # pylint: disable=too-few-public-methods
class InstapushNotificationService(BaseNotificationService): class InstapushNotificationService(BaseNotificationService):
"""Implement the notification service for Instapush.""" """Implementation of the notification service for Instapush."""
def __init__(self, api_key, app_secret, event, tracker): def __init__(self, api_key, app_secret, event, tracker):
"""Initialize the service.""" """Initialize the service."""
@ -78,13 +78,15 @@ class InstapushNotificationService(BaseNotificationService):
def send_message(self, message="", **kwargs): def send_message(self, message="", **kwargs):
"""Send a message to a user.""" """Send a message to a user."""
title = kwargs.get(ATTR_TITLE, ATTR_TITLE_DEFAULT) title = kwargs.get(ATTR_TITLE, ATTR_TITLE_DEFAULT)
data = {"event": self._event, data = {
"trackers": {self._tracker: title + " : " + message}} 'event': self._event,
'trackers': {self._tracker: title + ' : ' + message}
}
response = requests.post(_RESOURCE + 'post', data=json.dumps(data), response = requests.post(
headers=self._headers) '{}{}'.format(_RESOURCE, 'post'), data=json.dumps(data),
headers=self._headers, timeout=DEFAULT_TIMEOUT)
if response.json()['status'] == 401: if response.json()['status'] == 401:
_LOGGER.error( _LOGGER.error(response.json()['msg'],
response.json()['msg'], "Please check your Instapush settings")
"Please check your details at https://instapush.im/")