61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
"""Repairs flows for Home Connect."""
|
|
|
|
from typing import cast
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import data_entry_flow
|
|
from homeassistant.components.repairs import ConfirmRepairFlow, RepairsFlow
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import issue_registry as ir
|
|
|
|
from .coordinator import HomeConnectConfigEntry
|
|
|
|
|
|
class EnableApplianceUpdatesFlow(RepairsFlow):
|
|
"""Handler for enabling appliance's updates after being refreshed too many times."""
|
|
|
|
async def async_step_init(
|
|
self, user_input: dict[str, str] | None = None
|
|
) -> data_entry_flow.FlowResult:
|
|
"""Handle the first step of a fix flow."""
|
|
return await self.async_step_confirm()
|
|
|
|
async def async_step_confirm(
|
|
self, user_input: dict[str, str] | None = None
|
|
) -> data_entry_flow.FlowResult:
|
|
"""Handle the confirm step of a fix flow."""
|
|
if user_input is not None:
|
|
assert self.data
|
|
entry = self.hass.config_entries.async_get_entry(
|
|
cast(str, self.data["entry_id"])
|
|
)
|
|
assert entry
|
|
entry = cast(HomeConnectConfigEntry, entry)
|
|
await entry.runtime_data.reset_execution_tracker(
|
|
cast(str, self.data["appliance_ha_id"])
|
|
)
|
|
return self.async_create_entry(data={})
|
|
|
|
issue_registry = ir.async_get(self.hass)
|
|
description_placeholders = None
|
|
if issue := issue_registry.async_get_issue(self.handler, self.issue_id):
|
|
description_placeholders = issue.translation_placeholders
|
|
|
|
return self.async_show_form(
|
|
step_id="confirm",
|
|
data_schema=vol.Schema({}),
|
|
description_placeholders=description_placeholders,
|
|
)
|
|
|
|
|
|
async def async_create_fix_flow(
|
|
hass: HomeAssistant,
|
|
issue_id: str,
|
|
data: dict[str, str | int | float | None] | None,
|
|
) -> RepairsFlow:
|
|
"""Create flow."""
|
|
if issue_id.startswith("home_connect_too_many_connected_paired_events"):
|
|
return EnableApplianceUpdatesFlow()
|
|
return ConfirmRepairFlow()
|