mirror of https://github.com/home-assistant/core
105 lines
3.1 KiB
Python
105 lines
3.1 KiB
Python
"""The sms component."""
|
|
|
|
import logging
|
|
|
|
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 .const import (
|
|
CONF_BAUD_SPEED,
|
|
DEFAULT_BAUD_SPEED,
|
|
DOMAIN,
|
|
GATEWAY,
|
|
HASS_CONFIG,
|
|
NETWORK_COORDINATOR,
|
|
SIGNAL_COORDINATOR,
|
|
SMS_GATEWAY,
|
|
)
|
|
from .coordinator import NetworkCoordinator, SignalCoordinator
|
|
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,
|
|
)
|
|
|
|
|
|
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
|