97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
"""
|
|
Instapush notification service.
|
|
|
|
For more details about this platform, please refer to the documentation at
|
|
https://home-assistant.io/components/notify.instapush/
|
|
"""
|
|
import json
|
|
import logging
|
|
|
|
import requests
|
|
import voluptuous as vol
|
|
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.components.notify import (
|
|
ATTR_TITLE, ATTR_TITLE_DEFAULT, PLATFORM_SCHEMA, BaseNotificationService)
|
|
from homeassistant.const import (
|
|
CONF_API_KEY, HTTP_HEADER_CONTENT_TYPE, CONTENT_TYPE_JSON)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
_RESOURCE = 'https://api.instapush.im/v1/'
|
|
|
|
CONF_APP_SECRET = 'app_secret'
|
|
CONF_EVENT = 'event'
|
|
CONF_TRACKER = 'tracker'
|
|
|
|
DEFAULT_TIMEOUT = 10
|
|
|
|
HTTP_HEADER_APPID = 'x-instapush-appid'
|
|
HTTP_HEADER_APPSECRET = 'x-instapush-appsecret'
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
|
vol.Required(CONF_API_KEY): cv.string,
|
|
vol.Required(CONF_APP_SECRET): cv.string,
|
|
vol.Required(CONF_EVENT): cv.string,
|
|
vol.Required(CONF_TRACKER): cv.string,
|
|
})
|
|
|
|
|
|
def get_service(hass, config, discovery_info=None):
|
|
"""Get the Instapush notification service."""
|
|
headers = {
|
|
HTTP_HEADER_APPID: config[CONF_API_KEY],
|
|
HTTP_HEADER_APPSECRET: config[CONF_APP_SECRET],
|
|
}
|
|
|
|
try:
|
|
response = requests.get(
|
|
'{}{}'.format(_RESOURCE, 'events/list'), headers=headers,
|
|
timeout=DEFAULT_TIMEOUT).json()
|
|
except ValueError:
|
|
_LOGGER.error('Unexpected answer from Instapush API')
|
|
return None
|
|
|
|
if 'error' in response:
|
|
_LOGGER.error(response['msg'])
|
|
return None
|
|
|
|
if not [app for app in response if app['title'] == config[CONF_EVENT]]:
|
|
_LOGGER.error("No app match your given value")
|
|
return None
|
|
|
|
return InstapushNotificationService(
|
|
config.get(CONF_API_KEY), config.get(CONF_APP_SECRET),
|
|
config.get(CONF_EVENT), config.get(CONF_TRACKER))
|
|
|
|
|
|
class InstapushNotificationService(BaseNotificationService):
|
|
"""Implementation of the notification service for Instapush."""
|
|
|
|
def __init__(self, api_key, app_secret, event, tracker):
|
|
"""Initialize the service."""
|
|
self._api_key = api_key
|
|
self._app_secret = app_secret
|
|
self._event = event
|
|
self._tracker = tracker
|
|
self._headers = {
|
|
HTTP_HEADER_APPID: self._api_key,
|
|
HTTP_HEADER_APPSECRET: self._app_secret,
|
|
HTTP_HEADER_CONTENT_TYPE: CONTENT_TYPE_JSON,
|
|
}
|
|
|
|
def send_message(self, message="", **kwargs):
|
|
"""Send a message to a user."""
|
|
title = kwargs.get(ATTR_TITLE, ATTR_TITLE_DEFAULT)
|
|
data = {
|
|
'event': self._event,
|
|
'trackers': {self._tracker: '{} : {}'.format(title, message)}
|
|
}
|
|
|
|
response = requests.post(
|
|
'{}{}'.format(_RESOURCE, 'post'), data=json.dumps(data),
|
|
headers=self._headers, timeout=DEFAULT_TIMEOUT)
|
|
|
|
if response.json()['status'] == 401:
|
|
_LOGGER.error(response.json()['msg'],
|
|
"Please check your Instapush settings")
|