diff --git a/.coveragerc b/.coveragerc index d8dd47ad7de..96bd0be736f 100644 --- a/.coveragerc +++ b/.coveragerc @@ -430,6 +430,7 @@ omit = homeassistant/components/notify/pushover.py homeassistant/components/notify/pushsafer.py homeassistant/components/notify/rest.py + homeassistant/components/notify/rocketchat.py homeassistant/components/notify/sendgrid.py homeassistant/components/notify/simplepush.py homeassistant/components/notify/slack.py diff --git a/homeassistant/components/notify/rocketchat.py b/homeassistant/components/notify/rocketchat.py new file mode 100644 index 00000000000..f2898c8b998 --- /dev/null +++ b/homeassistant/components/notify/rocketchat.py @@ -0,0 +1,76 @@ +""" +Rocket.Chat notification service. + +For more details about this platform, please refer to the documentation at +https://home-assistant.io/components/notify.rocketchat/ +""" +import logging + +import voluptuous as vol + +from homeassistant.const import ( + CONF_URL, CONF_USERNAME, CONF_PASSWORD) +from homeassistant.components.notify import ( + ATTR_DATA, PLATFORM_SCHEMA, + BaseNotificationService) +import homeassistant.helpers.config_validation as cv + +REQUIREMENTS = ['rocketchat-API==0.6.1'] + +CONF_ROOM = 'room' + +_LOGGER = logging.getLogger(__name__) + +# pylint: disable=no-value-for-parameter +PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ + vol.Required(CONF_URL): vol.Url(), + vol.Required(CONF_USERNAME): cv.string, + vol.Required(CONF_PASSWORD): cv.string, + vol.Required(CONF_ROOM): cv.string, +}) + + +def get_service(hass, config, discovery_info=None): + """Return the notify service.""" + from rocketchat_API.APIExceptions.RocketExceptions import ( + RocketConnectionException, RocketAuthenticationException) + username = config.get(CONF_USERNAME) + password = config.get(CONF_PASSWORD) + + url = config.get(CONF_URL) + room = config.get(CONF_ROOM) + + try: + return RocketChatNotificationService(url, username, password, room) + except RocketConnectionException: + _LOGGER.warning( + "Unable to connect to Rocket.Chat server at %s.", url) + except RocketAuthenticationException: + _LOGGER.warning( + "Rocket.Chat authentication failed for user %s.", username) + _LOGGER.info("Please check your username/password.") + + return None + + +class RocketChatNotificationService(BaseNotificationService): + """Implement the notification service for Rocket.Chat.""" + + def __init__(self, url, username, password, room): + """Initialize the service.""" + from rocketchat_API.rocketchat import RocketChat + self._room = room + self._server = RocketChat(username, password, server_url=url) + + def send_message(self, message="", **kwargs): + """Send a message to Rocket.Chat.""" + data = kwargs.get(ATTR_DATA) or {} + resp = self._server.chat_post_message(message, channel=self._room, + **data) + if resp.status_code == 200: + success = resp.json()["success"] + if not success: + _LOGGER.error("Unable to post Rocket.Chat message") + else: + _LOGGER.error("Incorrect status code when posting message: %d", + resp.status_code) diff --git a/requirements_all.txt b/requirements_all.txt index 3b344120da3..4b196e2f2fe 100644 --- a/requirements_all.txt +++ b/requirements_all.txt @@ -886,6 +886,9 @@ rflink==0.0.34 # homeassistant.components.ring ring_doorbell==0.1.4 +# homeassistant.components.notify.rocketchat +rocketchat-API==0.6.1 + # homeassistant.components.vacuum.roomba roombapy==1.3.1