76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
"""SynologyChat platform for notify component."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from http import HTTPStatus
|
|
import json
|
|
import logging
|
|
|
|
import requests
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.notify import (
|
|
ATTR_DATA,
|
|
PLATFORM_SCHEMA as NOTIFY_PLATFORM_SCHEMA,
|
|
BaseNotificationService,
|
|
)
|
|
from homeassistant.const import CONF_RESOURCE, CONF_VERIFY_SSL
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import config_validation as cv
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
ATTR_FILE_URL = "file_url"
|
|
|
|
PLATFORM_SCHEMA = NOTIFY_PLATFORM_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_RESOURCE): cv.url,
|
|
vol.Optional(CONF_VERIFY_SSL, default=True): cv.boolean,
|
|
}
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def get_service(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> SynologyChatNotificationService:
|
|
"""Get the Synology Chat notification service."""
|
|
resource = config.get(CONF_RESOURCE)
|
|
verify_ssl = config.get(CONF_VERIFY_SSL)
|
|
|
|
return SynologyChatNotificationService(resource, verify_ssl)
|
|
|
|
|
|
class SynologyChatNotificationService(BaseNotificationService):
|
|
"""Implementation of a notification service for Synology Chat."""
|
|
|
|
def __init__(self, resource, verify_ssl):
|
|
"""Initialize the service."""
|
|
self._resource = resource
|
|
self._verify_ssl = verify_ssl
|
|
|
|
def send_message(self, message="", **kwargs):
|
|
"""Send a message to a user."""
|
|
data = {"text": message}
|
|
|
|
extended_data = kwargs.get(ATTR_DATA)
|
|
file_url = extended_data.get(ATTR_FILE_URL) if extended_data else None
|
|
|
|
if file_url:
|
|
data["file_url"] = file_url
|
|
|
|
to_send = f"payload={json.dumps(data)}"
|
|
|
|
response = requests.post(
|
|
self._resource, data=to_send, timeout=10, verify=self._verify_ssl
|
|
)
|
|
|
|
if response.status_code not in (HTTPStatus.OK, HTTPStatus.CREATED):
|
|
_LOGGER.exception(
|
|
"Error sending message. Response %d: %s:",
|
|
response.status_code,
|
|
response.reason,
|
|
)
|