99 lines
3.3 KiB
Python
99 lines
3.3 KiB
Python
"""ntfy notification entity."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from aiontfy import Message
|
|
from aiontfy.exceptions import (
|
|
NtfyException,
|
|
NtfyHTTPError,
|
|
NtfyUnauthorizedAuthenticationError,
|
|
)
|
|
from yarl import URL
|
|
|
|
from homeassistant.components.notify import (
|
|
NotifyEntity,
|
|
NotifyEntityDescription,
|
|
NotifyEntityFeature,
|
|
)
|
|
from homeassistant.config_entries import ConfigSubentry
|
|
from homeassistant.const import CONF_NAME, CONF_URL
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from .const import CONF_TOPIC, DOMAIN
|
|
from .coordinator import NtfyConfigEntry
|
|
|
|
PARALLEL_UPDATES = 0
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: NtfyConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up the ntfy notification entity platform."""
|
|
|
|
for subentry_id, subentry in config_entry.subentries.items():
|
|
async_add_entities(
|
|
[NtfyNotifyEntity(config_entry, subentry)], config_subentry_id=subentry_id
|
|
)
|
|
|
|
|
|
class NtfyNotifyEntity(NotifyEntity):
|
|
"""Representation of a ntfy notification entity."""
|
|
|
|
entity_description = NotifyEntityDescription(
|
|
key="publish",
|
|
translation_key="publish",
|
|
name=None,
|
|
has_entity_name=True,
|
|
)
|
|
_attr_supported_features = NotifyEntityFeature.TITLE
|
|
|
|
def __init__(
|
|
self,
|
|
config_entry: NtfyConfigEntry,
|
|
subentry: ConfigSubentry,
|
|
) -> None:
|
|
"""Initialize a notification entity."""
|
|
|
|
self._attr_unique_id = f"{config_entry.entry_id}_{subentry.subentry_id}_{self.entity_description.key}"
|
|
self.topic = subentry.data[CONF_TOPIC]
|
|
|
|
self._attr_device_info = DeviceInfo(
|
|
entry_type=DeviceEntryType.SERVICE,
|
|
manufacturer="ntfy LLC",
|
|
model="ntfy",
|
|
name=subentry.data.get(CONF_NAME, self.topic),
|
|
configuration_url=URL(config_entry.data[CONF_URL]) / self.topic,
|
|
identifiers={(DOMAIN, f"{config_entry.entry_id}_{subentry.subentry_id}")},
|
|
via_device=(DOMAIN, config_entry.entry_id),
|
|
)
|
|
self.config_entry = config_entry
|
|
self.ntfy = config_entry.runtime_data.ntfy
|
|
|
|
async def async_send_message(self, message: str, title: str | None = None) -> None:
|
|
"""Publish a message to a topic."""
|
|
msg = Message(topic=self.topic, message=message, title=title)
|
|
try:
|
|
await self.ntfy.publish(msg)
|
|
except NtfyUnauthorizedAuthenticationError as e:
|
|
self.config_entry.async_start_reauth(self.hass)
|
|
raise HomeAssistantError(
|
|
translation_domain=DOMAIN,
|
|
translation_key="authentication_error",
|
|
) from e
|
|
except NtfyHTTPError as e:
|
|
raise HomeAssistantError(
|
|
translation_key="publish_failed_request_error",
|
|
translation_domain=DOMAIN,
|
|
translation_placeholders={"error_msg": e.error},
|
|
) from e
|
|
except NtfyException as e:
|
|
raise HomeAssistantError(
|
|
translation_key="publish_failed_exception",
|
|
translation_domain=DOMAIN,
|
|
) from e
|