Chris Talkington
07ae3f9ee9
* use discovery resolved host rather than mdns host. * Update __init__.py * Update test_config_flow.py * Update __init__.py * Update test_init.py * Update test_config_flow.py * Update test_config_flow.py * Update __init__.py * Update __init__.py * Update __init__.py * Update test_init.py * Update test_config_flow.py
145 lines
4.9 KiB
Python
145 lines
4.9 KiB
Python
"""Config flow to configure the IPP integration."""
|
|
import logging
|
|
from typing import Any, Dict, Optional
|
|
|
|
from pyipp import IPP, IPPConnectionError, IPPConnectionUpgradeRequired
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.config_entries import CONN_CLASS_LOCAL_POLL, ConfigFlow
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_NAME,
|
|
CONF_PORT,
|
|
CONF_SSL,
|
|
CONF_VERIFY_SSL,
|
|
)
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|
from homeassistant.helpers.typing import ConfigType, HomeAssistantType
|
|
|
|
from .const import CONF_BASE_PATH, CONF_UUID
|
|
from .const import DOMAIN # pylint: disable=unused-import
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def validate_input(hass: HomeAssistantType, data: dict) -> Dict[str, Any]:
|
|
"""Validate the user input allows us to connect.
|
|
|
|
Data has the keys from DATA_SCHEMA with values provided by the user.
|
|
"""
|
|
session = async_get_clientsession(hass)
|
|
ipp = IPP(
|
|
host=data[CONF_HOST],
|
|
port=data[CONF_PORT],
|
|
base_path=data[CONF_BASE_PATH],
|
|
tls=data[CONF_SSL],
|
|
verify_ssl=data[CONF_VERIFY_SSL],
|
|
session=session,
|
|
)
|
|
|
|
printer = await ipp.printer()
|
|
|
|
return {CONF_UUID: printer.info.uuid}
|
|
|
|
|
|
class IPPFlowHandler(ConfigFlow, domain=DOMAIN):
|
|
"""Handle an IPP config flow."""
|
|
|
|
VERSION = 1
|
|
CONNECTION_CLASS = CONN_CLASS_LOCAL_POLL
|
|
|
|
def __init__(self):
|
|
"""Set up the instance."""
|
|
self.discovery_info = {}
|
|
|
|
async def async_step_user(
|
|
self, user_input: Optional[ConfigType] = None
|
|
) -> Dict[str, Any]:
|
|
"""Handle a flow initiated by the user."""
|
|
if user_input is None:
|
|
return self._show_setup_form()
|
|
|
|
try:
|
|
info = await validate_input(self.hass, user_input)
|
|
except IPPConnectionUpgradeRequired:
|
|
return self._show_setup_form({"base": "connection_upgrade"})
|
|
except IPPConnectionError:
|
|
return self._show_setup_form({"base": "connection_error"})
|
|
user_input[CONF_UUID] = info[CONF_UUID]
|
|
|
|
await self.async_set_unique_id(user_input[CONF_UUID])
|
|
self._abort_if_unique_id_configured(updates={CONF_HOST: user_input[CONF_HOST]})
|
|
|
|
return self.async_create_entry(title=user_input[CONF_HOST], data=user_input)
|
|
|
|
async def async_step_zeroconf(self, discovery_info: ConfigType) -> Dict[str, Any]:
|
|
"""Handle zeroconf discovery."""
|
|
# Hostname is format: EPSON123456.local.
|
|
host = discovery_info["hostname"].rstrip(".")
|
|
port = discovery_info["port"]
|
|
name, _ = host.rsplit(".")
|
|
tls = discovery_info["type"] == "_ipps._tcp.local."
|
|
|
|
# pylint: disable=no-member # https://github.com/PyCQA/pylint/issues/3167
|
|
self.context.update({"title_placeholders": {"name": name}})
|
|
|
|
self.discovery_info.update(
|
|
{
|
|
CONF_HOST: discovery_info[CONF_HOST],
|
|
CONF_PORT: port,
|
|
CONF_SSL: tls,
|
|
CONF_VERIFY_SSL: False,
|
|
CONF_BASE_PATH: "/"
|
|
+ discovery_info["properties"].get("rp", "ipp/print"),
|
|
CONF_NAME: name,
|
|
CONF_UUID: discovery_info["properties"].get("UUID"),
|
|
}
|
|
)
|
|
|
|
try:
|
|
info = await validate_input(self.hass, self.discovery_info)
|
|
except IPPConnectionUpgradeRequired:
|
|
return self.async_abort(reason="connection_upgrade")
|
|
except IPPConnectionError:
|
|
return self.async_abort(reason="connection_error")
|
|
|
|
self.discovery_info[CONF_UUID] = info[CONF_UUID]
|
|
|
|
await self.async_set_unique_id(self.discovery_info[CONF_UUID])
|
|
self._abort_if_unique_id_configured(
|
|
updates={CONF_HOST: self.discovery_info[CONF_HOST]}
|
|
)
|
|
|
|
return await self.async_step_zeroconf_confirm()
|
|
|
|
async def async_step_zeroconf_confirm(
|
|
self, user_input: ConfigType = None
|
|
) -> Dict[str, Any]:
|
|
"""Handle a confirmation flow initiated by zeroconf."""
|
|
if user_input is None:
|
|
return self.async_show_form(
|
|
step_id="zeroconf_confirm",
|
|
description_placeholders={"name": self.discovery_info[CONF_NAME]},
|
|
errors={},
|
|
)
|
|
|
|
return self.async_create_entry(
|
|
title=self.discovery_info[CONF_NAME], data=self.discovery_info,
|
|
)
|
|
|
|
def _show_setup_form(self, errors: Optional[Dict] = None) -> Dict[str, Any]:
|
|
"""Show the setup form to the user."""
|
|
return self.async_show_form(
|
|
step_id="user",
|
|
data_schema=vol.Schema(
|
|
{
|
|
vol.Required(CONF_HOST): str,
|
|
vol.Required(CONF_PORT, default=631): int,
|
|
vol.Required(CONF_BASE_PATH, default="/ipp/print"): str,
|
|
vol.Required(CONF_SSL, default=False): bool,
|
|
vol.Required(CONF_VERIFY_SSL, default=False): bool,
|
|
}
|
|
),
|
|
errors=errors or {},
|
|
)
|