71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
"""Discord platform for notify component."""
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import CONF_TOKEN
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
from homeassistant.components.notify import (ATTR_DATA, ATTR_TARGET,
|
|
PLATFORM_SCHEMA,
|
|
BaseNotificationService)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
REQUIREMENTS = ['discord.py==0.16.12']
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
|
vol.Required(CONF_TOKEN): cv.string
|
|
})
|
|
|
|
ATTR_IMAGES = 'images'
|
|
|
|
|
|
def get_service(hass, config, discovery_info=None):
|
|
"""Get the Discord notification service."""
|
|
token = config.get(CONF_TOKEN)
|
|
return DiscordNotificationService(hass, token)
|
|
|
|
|
|
class DiscordNotificationService(BaseNotificationService):
|
|
"""Implement the notification service for Discord."""
|
|
|
|
def __init__(self, hass, token):
|
|
"""Initialize the service."""
|
|
self.token = token
|
|
self.hass = hass
|
|
|
|
async def async_send_message(self, message, **kwargs):
|
|
"""Login to Discord, send message to channel(s) and log out."""
|
|
import discord
|
|
|
|
discord.VoiceClient.warn_nacl = False
|
|
discord_bot = discord.Client(loop=self.hass.loop)
|
|
|
|
if ATTR_TARGET not in kwargs:
|
|
_LOGGER.error("No target specified")
|
|
return None
|
|
|
|
# pylint: disable=unused-variable
|
|
@discord_bot.event
|
|
async def on_ready():
|
|
"""Send the messages when the bot is ready."""
|
|
try:
|
|
data = kwargs.get(ATTR_DATA)
|
|
images = None
|
|
if data:
|
|
images = data.get(ATTR_IMAGES)
|
|
for channelid in kwargs[ATTR_TARGET]:
|
|
channel = discord.Object(id=channelid)
|
|
await discord_bot.send_message(channel, message)
|
|
if images:
|
|
for anum, f_name in enumerate(images):
|
|
await discord_bot.send_file(channel, f_name)
|
|
except (discord.errors.HTTPException,
|
|
discord.errors.NotFound) as error:
|
|
_LOGGER.warning("Communication error: %s", error)
|
|
await discord_bot.logout()
|
|
await discord_bot.close()
|
|
|
|
await discord_bot.start(self.token)
|