064cc52ad6
* Add config flow to HLK-SW16
* Use entry_id for unique_id
* Add options update capability
* Refactor entry_id under domain
* Remove name from config
* Set options
* Remove options flow
* remove unneccesary else block from validate_input and move domain cleanup to async_unload_entry
* Add tests and config import
* Add back config schema
* Remove config import
* Refactor unload
* Add back config import
* Update coveragerc
* Don't mock validate_input
* Test duplicate configs
* Add import test
* Use patch for timeout test
* Use mock for testing timeout
* Use MockSW16Client for tests
* Check mock_calls count
* Remove unused NameExists exception
* Remove title from strings.json
* Mock setup for import test
* Set PARALLEL_UPDATES for switch
* Move hass.data.setdefault(DOMAIN, {}) to async_setup_entry
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
"""Config flow for HLK-SW16."""
|
|
import asyncio
|
|
|
|
from hlk_sw16 import create_hlk_sw16_connection
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.const import CONF_HOST, CONF_PORT
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .const import (
|
|
CONNECTION_TIMEOUT,
|
|
DEFAULT_KEEP_ALIVE_INTERVAL,
|
|
DEFAULT_PORT,
|
|
DEFAULT_RECONNECT_INTERVAL,
|
|
DOMAIN,
|
|
)
|
|
from .errors import AlreadyConfigured, CannotConnect
|
|
|
|
DATA_SCHEMA = vol.Schema(
|
|
{
|
|
vol.Required(CONF_HOST): str,
|
|
vol.Optional(CONF_PORT, default=DEFAULT_PORT): vol.Coerce(int),
|
|
}
|
|
)
|
|
|
|
|
|
async def connect_client(hass, user_input):
|
|
"""Connect the HLK-SW16 client."""
|
|
client_aw = create_hlk_sw16_connection(
|
|
host=user_input[CONF_HOST],
|
|
port=user_input[CONF_PORT],
|
|
loop=hass.loop,
|
|
timeout=CONNECTION_TIMEOUT,
|
|
reconnect_interval=DEFAULT_RECONNECT_INTERVAL,
|
|
keep_alive_interval=DEFAULT_KEEP_ALIVE_INTERVAL,
|
|
)
|
|
return await asyncio.wait_for(client_aw, timeout=CONNECTION_TIMEOUT)
|
|
|
|
|
|
async def validate_input(hass: HomeAssistant, user_input):
|
|
"""Validate the user input allows us to connect."""
|
|
for entry in hass.config_entries.async_entries(DOMAIN):
|
|
if (
|
|
entry.data[CONF_HOST] == user_input[CONF_HOST]
|
|
and entry.data[CONF_PORT] == user_input[CONF_PORT]
|
|
):
|
|
raise AlreadyConfigured
|
|
|
|
try:
|
|
client = await connect_client(hass, user_input)
|
|
except asyncio.TimeoutError:
|
|
raise CannotConnect
|
|
try:
|
|
|
|
def disconnect_callback():
|
|
if client.in_transaction:
|
|
client.active_transaction.set_exception(CannotConnect)
|
|
|
|
client.disconnect_callback = disconnect_callback
|
|
await client.status()
|
|
except CannotConnect:
|
|
client.disconnect_callback = None
|
|
client.stop()
|
|
raise CannotConnect
|
|
else:
|
|
client.disconnect_callback = None
|
|
client.stop()
|
|
|
|
|
|
class SW16FlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
|
|
"""Handle a HLK-SW16 config flow."""
|
|
|
|
VERSION = 1
|
|
CONNECTION_CLASS = config_entries.CONN_CLASS_LOCAL_PUSH
|
|
|
|
async def async_step_import(self, user_input):
|
|
"""Handle import."""
|
|
return await self.async_step_user(user_input)
|
|
|
|
async def async_step_user(self, user_input=None):
|
|
"""Handle the initial step."""
|
|
errors = {}
|
|
if user_input is not None:
|
|
try:
|
|
await validate_input(self.hass, user_input)
|
|
address = f"{user_input[CONF_HOST]}:{user_input[CONF_PORT]}"
|
|
return self.async_create_entry(title=address, data=user_input)
|
|
except AlreadyConfigured:
|
|
errors["base"] = "already_configured"
|
|
except CannotConnect:
|
|
errors["base"] = "cannot_connect"
|
|
|
|
return self.async_show_form(
|
|
step_id="user", data_schema=DATA_SCHEMA, errors=errors
|
|
)
|