async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
"""Set up kmtronic from a config entry."""
session = aiohttp_client.async_get_clientsession(hass)
auth = Auth(
session,
f"http://{entry.data[CONF_HOSTNAME]}",
entry.data[CONF_USERNAME],
entry.data[CONF_PASSWORD],
)
hub = KMTronicHubAPI(auth)
async def async_update_data():
try:
async with async_timeout.timeout(10):
await hub.async_update_relays()
except aiohttp.client_exceptions.ClientResponseError as err:
raise UpdateFailed(f"Wrong credentials: {err}") from err
except (
asyncio.TimeoutError,
aiohttp.client_exceptions.ClientConnectorError,
) as err:
raise UpdateFailed(f"Error communicating with API: {err}") from err
coordinator = DataUpdateCoordinator(
hass,
_LOGGER,
name=f"{MANUFACTURER} {hub.name}",
update_method=async_update_data,
update_interval=timedelta(seconds=30),
)
await coordinator.async_refresh()
if not coordinator.last_update_success:
raise ConfigEntryNotReady
if DOMAIN not in hass.data:
hass.data[DOMAIN] = {}
hass.data[DOMAIN][entry.entry_id] = {
DATA_HUB: hub,
DATA_HOST: entry.data[DATA_HOST],
DATA_COORDINATOR: coordinator,
}
device_registry = await dr.async_get_registry(hass)
device_registry.async_get_or_create(
identifiers={(DOMAIN, hub.name)},
config_entry_id=entry.entry_id,
manufacturer=MANUFACTURER,
name=hub.name,
)
for component in PLATFORMS:
hass.async_create_task(
hass.config_entries.async_forward_entry_setup(entry, component))
return True
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
"""Set up kmtronic from a config entry."""
session = aiohttp_client.async_get_clientsession(hass)
auth = Auth(
session,
f"http://{entry.data[CONF_HOST]}",
entry.data[CONF_USERNAME],
entry.data[CONF_PASSWORD],
)
hub = KMTronicHubAPI(auth)
async def async_update_data():
try:
async with async_timeout.timeout(10):
await hub.async_update_relays()
except aiohttp.client_exceptions.ClientResponseError as err:
raise UpdateFailed(f"Wrong credentials: {err}") from err
except (
asyncio.TimeoutError,
aiohttp.client_exceptions.ClientConnectorError,
) as err:
raise UpdateFailed(f"Error communicating with API: {err}") from err
coordinator = DataUpdateCoordinator(
hass,
_LOGGER,
name=f"{MANUFACTURER} {hub.name}",
update_method=async_update_data,
update_interval=timedelta(seconds=30),
)
await coordinator.async_refresh()
if not coordinator.last_update_success:
raise ConfigEntryNotReady
hass.data[DOMAIN][entry.entry_id] = {
DATA_HUB: hub,
DATA_COORDINATOR: coordinator,
}
for platform in PLATFORMS:
hass.async_create_task(
hass.config_entries.async_forward_entry_setup(entry, platform)
)
update_listener = entry.add_update_listener(async_update_options)
hass.data[DOMAIN][entry.entry_id][UPDATE_LISTENER] = update_listener
return True
async def validate_input(hass: core.HomeAssistant, data):
"""Validate the user input allows us to connect."""
session = aiohttp_client.async_get_clientsession(hass)
auth = Auth(
session,
f"http://{data[CONF_HOST]}",
data[CONF_USERNAME],
data[CONF_PASSWORD],
)
hub = KMTronicHubAPI(auth)
try:
await hub.async_get_status()
except aiohttp.client_exceptions.ClientResponseError as err:
raise InvalidAuth from err
except aiohttp.client_exceptions.ClientConnectorError as err:
raise CannotConnect from err
return data
async def async_setup_entry(opp: OpenPeerPower, entry: ConfigEntry):
"""Set up kmtronic from a config entry."""
session = aiohttp_client.async_get_clientsession(opp)
auth = Auth(
session,
f"http://{entry.data[CONF_HOST]}",
entry.data[CONF_USERNAME],
entry.data[CONF_PASSWORD],
)
hub = KMTronicHubAPI(auth)
async def async_update_data():
try:
async with async_timeout.timeout(10):
await hub.async_update_relays()
except aiohttp.client_exceptions.ClientResponseError as err:
raise UpdateFailed(f"Wrong credentials: {err}") from err
except aiohttp.client_exceptions.ClientConnectorError as err:
raise UpdateFailed(f"Error communicating with API: {err}") from err
coordinator = DataUpdateCoordinator(
opp,
_LOGGER,
name=f"{MANUFACTURER} {hub.name}",
update_method=async_update_data,
update_interval=timedelta(seconds=30),
)
await coordinator.async_config_entry_first_refresh()
opp.data.setdefault(DOMAIN, {})
opp.data[DOMAIN][entry.entry_id] = {
DATA_HUB: hub,
DATA_COORDINATOR: coordinator,
}
opp.config_entries.async_setup_platforms(entry, PLATFORMS)
update_listener = entry.add_update_listener(async_update_options)
opp.data[DOMAIN][entry.entry_id][UPDATE_LISTENER] = update_listener
return True