132 lines
3.8 KiB
Python
132 lines
3.8 KiB
Python
"""Support for IKEA Tradfri."""
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.util.json import load_json
|
|
|
|
from .const import (
|
|
CONF_IMPORT_GROUPS, CONF_IDENTITY, CONF_HOST, CONF_KEY, CONF_GATEWAY_ID)
|
|
|
|
from . import config_flow # noqa pylint_disable=unused-import
|
|
|
|
REQUIREMENTS = ['pytradfri[async]==6.0.1']
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
DOMAIN = 'tradfri'
|
|
CONFIG_FILE = '.tradfri_psk.conf'
|
|
KEY_GATEWAY = 'tradfri_gateway'
|
|
KEY_API = 'tradfri_api'
|
|
CONF_ALLOW_TRADFRI_GROUPS = 'allow_tradfri_groups'
|
|
DEFAULT_ALLOW_TRADFRI_GROUPS = False
|
|
|
|
CONFIG_SCHEMA = vol.Schema({
|
|
DOMAIN: vol.Schema({
|
|
vol.Optional(CONF_HOST): cv.string,
|
|
vol.Optional(CONF_ALLOW_TRADFRI_GROUPS,
|
|
default=DEFAULT_ALLOW_TRADFRI_GROUPS): cv.boolean,
|
|
})
|
|
}, extra=vol.ALLOW_EXTRA)
|
|
|
|
|
|
async def async_setup(hass, config):
|
|
"""Set up the Tradfri component."""
|
|
conf = config.get(DOMAIN)
|
|
|
|
if conf is None:
|
|
return True
|
|
|
|
configured_hosts = [entry.data['host'] for entry in
|
|
hass.config_entries.async_entries(DOMAIN)]
|
|
|
|
legacy_hosts = await hass.async_add_executor_job(
|
|
load_json, hass.config.path(CONFIG_FILE))
|
|
|
|
for host, info in legacy_hosts.items():
|
|
if host in configured_hosts:
|
|
continue
|
|
|
|
info[CONF_HOST] = host
|
|
info[CONF_IMPORT_GROUPS] = conf[CONF_ALLOW_TRADFRI_GROUPS]
|
|
|
|
hass.async_create_task(hass.config_entries.flow.async_init(
|
|
DOMAIN, context={'source': config_entries.SOURCE_IMPORT},
|
|
data=info
|
|
))
|
|
|
|
host = conf.get(CONF_HOST)
|
|
import_groups = conf[CONF_ALLOW_TRADFRI_GROUPS]
|
|
|
|
if host is None or host in configured_hosts or host in legacy_hosts:
|
|
return True
|
|
|
|
hass.async_create_task(hass.config_entries.flow.async_init(
|
|
DOMAIN, context={'source': config_entries.SOURCE_IMPORT},
|
|
data={CONF_HOST: host, CONF_IMPORT_GROUPS: import_groups}
|
|
))
|
|
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass, entry):
|
|
"""Create a gateway."""
|
|
# host, identity, key, allow_tradfri_groups
|
|
from pytradfri import Gateway, RequestError # pylint: disable=import-error
|
|
from pytradfri.api.aiocoap_api import APIFactory
|
|
|
|
factory = APIFactory(
|
|
entry.data[CONF_HOST],
|
|
psk_id=entry.data[CONF_IDENTITY],
|
|
psk=entry.data[CONF_KEY],
|
|
loop=hass.loop
|
|
)
|
|
|
|
async def on_hass_stop(event):
|
|
"""Close connection when hass stops."""
|
|
await factory.shutdown()
|
|
|
|
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, on_hass_stop)
|
|
|
|
api = factory.request
|
|
gateway = Gateway()
|
|
|
|
try:
|
|
gateway_info = await api(gateway.get_gateway_info())
|
|
except RequestError:
|
|
_LOGGER.error("Tradfri setup failed.")
|
|
return False
|
|
|
|
hass.data.setdefault(KEY_API, {})[entry.entry_id] = api
|
|
hass.data.setdefault(KEY_GATEWAY, {})[entry.entry_id] = gateway
|
|
|
|
dev_reg = await hass.helpers.device_registry.async_get_registry()
|
|
dev_reg.async_get_or_create(
|
|
config_entry_id=entry.entry_id,
|
|
connections=set(),
|
|
identifiers={
|
|
(DOMAIN, entry.data[CONF_GATEWAY_ID])
|
|
},
|
|
manufacturer='IKEA',
|
|
name='Gateway',
|
|
# They just have 1 gateway model. Type is not exposed yet.
|
|
model='E1526',
|
|
sw_version=gateway_info.firmware_version,
|
|
)
|
|
|
|
hass.async_create_task(hass.config_entries.async_forward_entry_setup(
|
|
entry, 'light'
|
|
))
|
|
hass.async_create_task(hass.config_entries.async_forward_entry_setup(
|
|
entry, 'sensor'
|
|
))
|
|
hass.async_create_task(hass.config_entries.async_forward_entry_setup(
|
|
entry, 'switch'
|
|
))
|
|
|
|
return True
|