72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
|
"""
|
||
|
|
Mastodon platform for notify component.
|
||
|
|
|
||
|
|
For more details about this platform, please refer to the documentation at
|
||
|
|
https://home-assistant.io/components/notify.mastodon/
|
||
|
|
"""
|
||
|
|
import logging
|
||
|
|
|
||
|
|
import voluptuous as vol
|
||
|
|
|
||
|
|
from homeassistant.const import CONF_ACCESS_TOKEN
|
||
|
|
import homeassistant.helpers.config_validation as cv
|
||
|
|
|
||
|
from homeassistant.components.notify import (PLATFORM_SCHEMA,
|
||
|
|
BaseNotificationService)
|
||
|
|
||
|
|
REQUIREMENTS = ['Mastodon.py==1.3.1']
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
CONF_BASE_URL = 'base_url'
|
||
|
|
CONF_CLIENT_ID = 'client_id'
|
||
|
|
CONF_CLIENT_SECRET = 'client_secret'
|
||
|
|
|
||
|
|
DEFAULT_URL = 'https://mastodon.social'
|
||
|
|
|
||
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
||
|
|
vol.Required(CONF_ACCESS_TOKEN): cv.string,
|
||
|
|
vol.Required(CONF_CLIENT_ID): cv.string,
|
||
|
|
vol.Required(CONF_CLIENT_SECRET): cv.string,
|
||
|
|
vol.Optional(CONF_BASE_URL, default=DEFAULT_URL): cv.string,
|
||
|
|
})
|
||
|
|
|
||
|
|
|
||
|
|
def get_service(hass, config, discovery_info=None):
|
||
|
|
"""Get the Mastodon notification service."""
|
||
|
|
from mastodon import Mastodon
|
||
|
|
from mastodon.Mastodon import MastodonUnauthorizedError
|
||
|
|
|
||
|
|
client_id = config.get(CONF_CLIENT_ID)
|
||
|
|
client_secret = config.get(CONF_CLIENT_SECRET)
|
||
|
|
access_token = config.get(CONF_ACCESS_TOKEN)
|
||
|
|
base_url = config.get(CONF_BASE_URL)
|
||
|
|
|
||
|
|
try:
|
||
|
|
mastodon = Mastodon(
|
||
|
|
client_id=client_id, client_secret=client_secret,
|
||
|
|
access_token=access_token, api_base_url=base_url)
|
||
|
|
mastodon.account_verify_credentials()
|
||
|
|
except MastodonUnauthorizedError:
|
||
|
|
_LOGGER.warning("Authentication failed")
|
||
|
|
return None
|
||
|
|
|
||
|
|
return MastodonNotificationService(mastodon)
|
||
|
|
|
||
|
|
|
||
|
|
class MastodonNotificationService(BaseNotificationService):
|
||
|
|
"""Implement the notification service for Mastodon."""
|
||
|
|
|
||
|
|
def __init__(self, api):
|
||
|
|
"""Initialize the service."""
|
||
|
|
self._api = api
|
||
|
|
|
||
|
|
def send_message(self, message="", **kwargs):
|
||
|
|
"""Send a message to a user."""
|
||
|
|
from mastodon.Mastodon import MastodonAPIError
|
||
|
|
|
||
|
|
try:
|
||
|
|
self._api.toot(message)
|
||
|
|
except MastodonAPIError:
|
||
|
|
_LOGGER.error("Unable to send message")
|