Catch exceptions (#9085)
* Catch exceptions * Fix pylint disable * Move check for emtpy target listpull/9027/merge
parent
55a44b0a1c
commit
2355216f61
|
@ -4,9 +4,11 @@ Discord platform for notify component.
|
|||
For more details about this platform, please refer to the documentation at
|
||||
https://home-assistant.io/components/notify.discord/
|
||||
"""
|
||||
import logging
|
||||
import asyncio
|
||||
import logging
|
||||
|
||||
import voluptuous as vol
|
||||
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
from homeassistant.components.notify import (
|
||||
PLATFORM_SCHEMA, BaseNotificationService, ATTR_TARGET)
|
||||
|
@ -42,13 +44,22 @@ class DiscordNotificationService(BaseNotificationService):
|
|||
import discord
|
||||
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
|
||||
@asyncio.coroutine
|
||||
def on_ready(): # pylint: disable=unused-variable
|
||||
def on_ready():
|
||||
"""Send the messages when the bot is ready."""
|
||||
for channelid in kwargs[ATTR_TARGET]:
|
||||
channel = discord.Object(id=channelid)
|
||||
yield from discord_bot.send_message(channel, message)
|
||||
try:
|
||||
for channelid in kwargs[ATTR_TARGET]:
|
||||
channel = discord.Object(id=channelid)
|
||||
yield from discord_bot.send_message(channel, message)
|
||||
except (discord.errors.HTTPException,
|
||||
discord.errors.NotFound) as error:
|
||||
_LOGGER.warning("Communication error: %s", error)
|
||||
yield from discord_bot.logout()
|
||||
yield from discord_bot.close()
|
||||
|
||||
|
|
Loading…
Reference in New Issue