core/homeassistant/components/huawei_lte/notify.py

55 lines
1.6 KiB
Python

"""Support for Huawei LTE router notifications."""
import logging
from typing import Any, List
import attr
from huawei_lte_api.exceptions import ResponseErrorException
from homeassistant.components.notify import BaseNotificationService, ATTR_TARGET
from homeassistant.const import CONF_RECIPIENT, CONF_URL
from . import Router
from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
async def async_get_service(hass, config, discovery_info=None):
"""Get the notification service."""
if discovery_info is None:
_LOGGER.warning(
"Loading as a platform is no longer supported, convert to use "
"config entries or the huawei_lte component"
)
return None
router = hass.data[DOMAIN].routers[discovery_info[CONF_URL]]
default_targets = discovery_info[CONF_RECIPIENT] or []
return HuaweiLteSmsNotificationService(router, default_targets)
@attr.s
class HuaweiLteSmsNotificationService(BaseNotificationService):
"""Huawei LTE router SMS notification service."""
router: Router = attr.ib()
default_targets: List[str] = attr.ib()
def send_message(self, message: str = "", **kwargs: Any) -> None:
"""Send message to target numbers."""
targets = kwargs.get(ATTR_TARGET, self.default_targets)
if not targets or not message:
return
try:
resp = self.router.client.sms.send_sms(
phone_numbers=targets, message=message
)
_LOGGER.debug("Sent to %s: %s", targets, resp)
except ResponseErrorException as ex:
_LOGGER.error("Could not send to %s: %s", targets, ex)