mirror of
https://github.com/mx42/home-assistant-ecocito.git
synced 2026-01-14 05:49:51 +01:00
67016339b3
If the selected type is "Skip synchro", the ecocito data won't contain the related coordinator(s) and any sync of that type will be skipped.
114 lines
3.9 KiB
Python
114 lines
3.9 KiB
Python
"""Options flow for ecocito configuration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
import voluptuous as vol
|
|
from homeassistant import config_entries
|
|
from homeassistant.const import CONF_DOMAIN, CONF_PASSWORD, CONF_USERNAME
|
|
from homeassistant.helpers import selector
|
|
|
|
from .client import EcocitoClient
|
|
from .const import (
|
|
ECOCITO_GARBAGE_TYPE,
|
|
ECOCITO_RECYCLE_TYPE,
|
|
ECOCITO_REFRESH_MIN_KEY,
|
|
)
|
|
|
|
# Hardcode a value for "skip synchro"
|
|
# -> it seems empty value would fill in previous known value
|
|
SKIP_SYNC_VALUE = "-42"
|
|
|
|
def build_schema(type_mapping: dict[int, str], current: dict[str, Any]) -> vol.Schema:
|
|
"""Build the schema."""
|
|
types_options = [
|
|
{"value": str(type_id), "label": type_label}
|
|
for type_id, type_label in type_mapping.items()
|
|
]
|
|
types_options.append({
|
|
"value": SKIP_SYNC_VALUE,
|
|
"label": "Skip synchro"
|
|
})
|
|
return vol.Schema(
|
|
{
|
|
vol.Optional(ECOCITO_GARBAGE_TYPE, default=current[ECOCITO_GARBAGE_TYPE]):
|
|
selector.SelectSelector(selector.SelectSelectorConfig(
|
|
options=types_options,
|
|
mode=selector.SelectSelectorMode.DROPDOWN,
|
|
multiple=False
|
|
)),
|
|
vol.Optional(ECOCITO_RECYCLE_TYPE, default=current[ECOCITO_RECYCLE_TYPE]):
|
|
selector.SelectSelector(selector.SelectSelectorConfig(
|
|
options=types_options,
|
|
mode=selector.SelectSelectorMode.DROPDOWN,
|
|
multiple=False
|
|
)),
|
|
vol.Required(
|
|
ECOCITO_REFRESH_MIN_KEY, default=current[ECOCITO_REFRESH_MIN_KEY]
|
|
): int
|
|
}
|
|
)
|
|
|
|
class EcocitoOptionsFlowHandler(config_entries.OptionsFlow):
|
|
"""Handle an option flow for ecocito."""
|
|
|
|
def __init__(self, entry: config_entries.ConfigEntry) -> None:
|
|
"""Init."""
|
|
self._entry = entry
|
|
|
|
|
|
async def get_type_mapping(self) -> dict[int, str]:
|
|
"""Build client from config."""
|
|
client = EcocitoClient(
|
|
self._entry.data[CONF_DOMAIN],
|
|
self._entry.data[CONF_USERNAME],
|
|
self._entry.data[CONF_PASSWORD],
|
|
)
|
|
await client.authenticate()
|
|
return await client.get_collection_types()
|
|
|
|
async def update_config(self, user_input: dict[str, Any]) -> None:
|
|
"""Update configuration with new user input."""
|
|
new_data = dict(self._entry.data)
|
|
for key in [
|
|
ECOCITO_GARBAGE_TYPE,
|
|
ECOCITO_RECYCLE_TYPE,
|
|
ECOCITO_REFRESH_MIN_KEY,
|
|
]:
|
|
if key not in user_input:
|
|
continue
|
|
if user_input[key] == SKIP_SYNC_VALUE:
|
|
new_data.pop(key, None) # = None
|
|
continue
|
|
int_val = int(user_input[key])
|
|
if (key not in new_data or int_val != new_data.get(key)) and int_val > 0:
|
|
new_data[key] = int_val
|
|
self.hass.config_entries.async_update_entry(
|
|
self._entry, data=new_data
|
|
)
|
|
await self.hass.config_entries.async_reload(self._entry.entry_id)
|
|
|
|
async def async_step_init(
|
|
self,
|
|
user_input: dict[str, Any] | None = None
|
|
) -> config_entries.ConfigFlowResult:
|
|
"""Display configuration menu."""
|
|
errors: dict[str, str] = {}
|
|
if user_input is not None:
|
|
await self.update_config(user_input)
|
|
return self.async_abort(reason="Changes saved.")
|
|
placeholders = {
|
|
ECOCITO_GARBAGE_TYPE: str(self._entry.data.get(ECOCITO_GARBAGE_TYPE, SKIP_SYNC_VALUE)),
|
|
ECOCITO_RECYCLE_TYPE: str(self._entry.data.get(ECOCITO_RECYCLE_TYPE, SKIP_SYNC_VALUE)),
|
|
ECOCITO_REFRESH_MIN_KEY: int(
|
|
self._entry.data.get(ECOCITO_REFRESH_MIN_KEY, 60)
|
|
),
|
|
}
|
|
schema = build_schema(await self.get_type_mapping(), placeholders)
|
|
return self.async_show_form(
|
|
step_id="init",
|
|
data_schema=schema,
|
|
errors=errors
|
|
)
|