mirror of
https://github.com/home-assistant/core.git
synced 2025-12-25 05:26:47 +00:00
Great migration notify (#22406)
* Move notify platforms into components * Move notify tests * Fix notify tests * More fixes * Update requirements * Update .coveragerc * Run gen reqs
This commit is contained in:
committed by
Paulus Schoutsen
parent
92457ca5ca
commit
7741ec4d5a
119
homeassistant/components/facebook/notify.py
Normal file
119
homeassistant/components/facebook/notify.py
Normal file
@@ -0,0 +1,119 @@
|
||||
"""
|
||||
Facebook platform for notify component.
|
||||
|
||||
For more details about this platform, please refer to the documentation at
|
||||
https://home-assistant.io/components/notify.facebook/
|
||||
"""
|
||||
import json
|
||||
import logging
|
||||
|
||||
from aiohttp.hdrs import CONTENT_TYPE
|
||||
import requests
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant.const import CONTENT_TYPE_JSON
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
|
||||
from homeassistant.components.notify import (ATTR_DATA, ATTR_TARGET,
|
||||
PLATFORM_SCHEMA,
|
||||
BaseNotificationService)
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
CONF_PAGE_ACCESS_TOKEN = 'page_access_token'
|
||||
BASE_URL = 'https://graph.facebook.com/v2.6/me/messages'
|
||||
CREATE_BROADCAST_URL = 'https://graph.facebook.com/v2.11/me/message_creatives'
|
||||
SEND_BROADCAST_URL = 'https://graph.facebook.com/v2.11/me/broadcast_messages'
|
||||
|
||||
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
||||
vol.Required(CONF_PAGE_ACCESS_TOKEN): cv.string,
|
||||
})
|
||||
|
||||
|
||||
def get_service(hass, config, discovery_info=None):
|
||||
"""Get the Facebook notification service."""
|
||||
return FacebookNotificationService(config[CONF_PAGE_ACCESS_TOKEN])
|
||||
|
||||
|
||||
class FacebookNotificationService(BaseNotificationService):
|
||||
"""Implementation of a notification service for the Facebook service."""
|
||||
|
||||
def __init__(self, access_token):
|
||||
"""Initialize the service."""
|
||||
self.page_access_token = access_token
|
||||
|
||||
def send_message(self, message="", **kwargs):
|
||||
"""Send some message."""
|
||||
payload = {'access_token': self.page_access_token}
|
||||
targets = kwargs.get(ATTR_TARGET)
|
||||
data = kwargs.get(ATTR_DATA)
|
||||
|
||||
body_message = {"text": message}
|
||||
|
||||
if data is not None:
|
||||
body_message.update(data)
|
||||
# Only one of text or attachment can be specified
|
||||
if 'attachment' in body_message:
|
||||
body_message.pop('text')
|
||||
|
||||
if not targets:
|
||||
_LOGGER.error("At least 1 target is required")
|
||||
return
|
||||
|
||||
# broadcast message
|
||||
if targets[0].lower() == 'broadcast':
|
||||
broadcast_create_body = {"messages": [body_message]}
|
||||
_LOGGER.debug("Broadcast body %s : ", broadcast_create_body)
|
||||
|
||||
resp = requests.post(CREATE_BROADCAST_URL,
|
||||
data=json.dumps(broadcast_create_body),
|
||||
params=payload,
|
||||
headers={CONTENT_TYPE: CONTENT_TYPE_JSON},
|
||||
timeout=10)
|
||||
_LOGGER.debug("FB Messager broadcast id %s : ", resp.json())
|
||||
|
||||
# at this point we get broadcast id
|
||||
broadcast_body = {
|
||||
"message_creative_id": resp.json().get('message_creative_id'),
|
||||
"notification_type": "REGULAR",
|
||||
}
|
||||
|
||||
resp = requests.post(SEND_BROADCAST_URL,
|
||||
data=json.dumps(broadcast_body),
|
||||
params=payload,
|
||||
headers={CONTENT_TYPE: CONTENT_TYPE_JSON},
|
||||
timeout=10)
|
||||
if resp.status_code != 200:
|
||||
log_error(resp)
|
||||
|
||||
# non-broadcast message
|
||||
else:
|
||||
for target in targets:
|
||||
# If the target starts with a "+", it's a phone number,
|
||||
# otherwise it's a user id.
|
||||
if target.startswith('+'):
|
||||
recipient = {"phone_number": target}
|
||||
else:
|
||||
recipient = {"id": target}
|
||||
|
||||
body = {
|
||||
"recipient": recipient,
|
||||
"message": body_message
|
||||
}
|
||||
resp = requests.post(BASE_URL, data=json.dumps(body),
|
||||
params=payload,
|
||||
headers={CONTENT_TYPE: CONTENT_TYPE_JSON},
|
||||
timeout=10)
|
||||
if resp.status_code != 200:
|
||||
log_error(resp)
|
||||
|
||||
|
||||
def log_error(response):
|
||||
"""Log error message."""
|
||||
obj = response.json()
|
||||
error_message = obj['error']['message']
|
||||
error_code = obj['error']['code']
|
||||
|
||||
_LOGGER.error(
|
||||
"Error %s : %s (Code %s)", response.status_code, error_message,
|
||||
error_code)
|
||||
Reference in New Issue
Block a user