154 lines
4.7 KiB
Python
154 lines
4.7 KiB
Python
"""The sms component."""
|
|
from datetime import timedelta
|
|
import logging
|
|
|
|
import async_timeout
|
|
import gammu # pylint: disable=import-error
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_DEVICE, CONF_NAME, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
from homeassistant.helpers import config_validation as cv, discovery
|
|
from homeassistant.helpers.typing import ConfigType
|
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
|
|
|
from .const import (
|
|
CONF_BAUD_SPEED,
|
|
DEFAULT_BAUD_SPEED,
|
|
DEFAULT_SCAN_INTERVAL,
|
|
DOMAIN,
|
|
GATEWAY,
|
|
HASS_CONFIG,
|
|
NETWORK_COORDINATOR,
|
|
SIGNAL_COORDINATOR,
|
|
SMS_GATEWAY,
|
|
)
|
|
from .gateway import create_sms_gateway
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
PLATFORMS = [Platform.SENSOR]
|
|
|
|
SMS_CONFIG_SCHEMA = {vol.Required(CONF_DEVICE): cv.isdevice}
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Schema(
|
|
vol.All(
|
|
cv.deprecated(CONF_DEVICE),
|
|
SMS_CONFIG_SCHEMA,
|
|
),
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
|
|
"""Configure Gammu state machine."""
|
|
hass.data.setdefault(DOMAIN, {})
|
|
hass.data[HASS_CONFIG] = config
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Configure Gammu state machine."""
|
|
|
|
device = entry.data[CONF_DEVICE]
|
|
connection_mode = "at"
|
|
baud_speed = entry.data.get(CONF_BAUD_SPEED, DEFAULT_BAUD_SPEED)
|
|
if baud_speed != DEFAULT_BAUD_SPEED:
|
|
connection_mode += baud_speed
|
|
config = {"Device": device, "Connection": connection_mode}
|
|
_LOGGER.debug("Connecting mode:%s", connection_mode)
|
|
gateway = await create_sms_gateway(config, hass)
|
|
if not gateway:
|
|
raise ConfigEntryNotReady(f"Cannot find device {device}")
|
|
|
|
signal_coordinator = SignalCoordinator(hass, gateway)
|
|
network_coordinator = NetworkCoordinator(hass, gateway)
|
|
|
|
# Fetch initial data so we have data when entities subscribe
|
|
await signal_coordinator.async_config_entry_first_refresh()
|
|
await network_coordinator.async_config_entry_first_refresh()
|
|
|
|
hass.data.setdefault(DOMAIN, {})
|
|
hass.data[DOMAIN][SMS_GATEWAY] = {
|
|
SIGNAL_COORDINATOR: signal_coordinator,
|
|
NETWORK_COORDINATOR: network_coordinator,
|
|
GATEWAY: gateway,
|
|
}
|
|
|
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
|
|
|
# set up notify platform, no entry support for notify component yet,
|
|
# have to use discovery to load platform.
|
|
hass.async_create_task(
|
|
discovery.async_load_platform(
|
|
hass,
|
|
Platform.NOTIFY,
|
|
DOMAIN,
|
|
{CONF_NAME: DOMAIN},
|
|
hass.data[HASS_CONFIG],
|
|
)
|
|
)
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
|
if unload_ok:
|
|
gateway = hass.data[DOMAIN].pop(SMS_GATEWAY)[GATEWAY]
|
|
await gateway.terminate_async()
|
|
|
|
return unload_ok
|
|
|
|
|
|
class SignalCoordinator(DataUpdateCoordinator):
|
|
"""Signal strength coordinator."""
|
|
|
|
def __init__(self, hass, gateway):
|
|
"""Initialize signal strength coordinator."""
|
|
super().__init__(
|
|
hass,
|
|
_LOGGER,
|
|
name="Device signal state",
|
|
update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL),
|
|
)
|
|
self._gateway = gateway
|
|
|
|
async def _async_update_data(self):
|
|
"""Fetch device signal quality."""
|
|
try:
|
|
async with async_timeout.timeout(10):
|
|
return await self._gateway.get_signal_quality_async()
|
|
except gammu.GSMError as exc:
|
|
raise UpdateFailed(f"Error communicating with device: {exc}") from exc
|
|
|
|
|
|
class NetworkCoordinator(DataUpdateCoordinator):
|
|
"""Network info coordinator."""
|
|
|
|
def __init__(self, hass, gateway):
|
|
"""Initialize network info coordinator."""
|
|
super().__init__(
|
|
hass,
|
|
_LOGGER,
|
|
name="Device network state",
|
|
update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL),
|
|
)
|
|
self._gateway = gateway
|
|
|
|
async def _async_update_data(self):
|
|
"""Fetch device network info."""
|
|
try:
|
|
async with async_timeout.timeout(10):
|
|
return await self._gateway.get_network_info_async()
|
|
except gammu.GSMError as exc:
|
|
raise UpdateFailed(f"Error communicating with device: {exc}") from exc
|