core/homeassistant/components/free_mobile/notify.py

55 lines
1.9 KiB
Python

"""Support for Free Mobile SMS platform."""
from __future__ import annotations
from http import HTTPStatus
import logging
from freesms import FreeClient
import voluptuous as vol
from homeassistant.components.notify import (
PLATFORM_SCHEMA as NOTIFY_PLATFORM_SCHEMA,
BaseNotificationService,
)
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_USERNAME
from homeassistant.core import HomeAssistant
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
_LOGGER = logging.getLogger(__name__)
PLATFORM_SCHEMA = NOTIFY_PLATFORM_SCHEMA.extend(
{vol.Required(CONF_USERNAME): cv.string, vol.Required(CONF_ACCESS_TOKEN): cv.string}
)
def get_service(
hass: HomeAssistant,
config: ConfigType,
discovery_info: DiscoveryInfoType | None = None,
) -> FreeSMSNotificationService:
"""Get the Free Mobile SMS notification service."""
return FreeSMSNotificationService(config[CONF_USERNAME], config[CONF_ACCESS_TOKEN])
class FreeSMSNotificationService(BaseNotificationService):
"""Implement a notification service for the Free Mobile SMS service."""
def __init__(self, username, access_token):
"""Initialize the service."""
self.free_client = FreeClient(username, access_token)
def send_message(self, message="", **kwargs):
"""Send a message to the Free Mobile user cell."""
resp = self.free_client.send_sms(message)
if resp.status_code == HTTPStatus.BAD_REQUEST:
_LOGGER.error("At least one parameter is missing")
elif resp.status_code == HTTPStatus.PAYMENT_REQUIRED:
_LOGGER.error("Too much SMS send in a few time")
elif resp.status_code == HTTPStatus.FORBIDDEN:
_LOGGER.error("Wrong Username/Password")
elif resp.status_code == HTTPStatus.INTERNAL_SERVER_ERROR:
_LOGGER.error("Server error, try later")