55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.repairs import RepairsFlow
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant import data_entry_flow
|
|
|
|
from .core import RESTART_REQUIRED_ISSUE_ID
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class BCSRestartRequiredFlow(RepairsFlow):
|
|
"""Repairs flow to restart Home Assistant after BCS install/update."""
|
|
|
|
def __init__(self, hass: HomeAssistant) -> None:
|
|
self.hass = hass
|
|
|
|
async def async_step_init(
|
|
self, user_input: dict[str, str] | None = None
|
|
) -> data_entry_flow.FlowResult:
|
|
return await self.async_step_confirm(user_input)
|
|
|
|
async def async_step_confirm(
|
|
self, user_input: dict[str, str] | None = None
|
|
) -> data_entry_flow.FlowResult:
|
|
if user_input is not None:
|
|
_LOGGER.info("BCS repairs flow: restarting Home Assistant (user confirmed)")
|
|
await self.hass.services.async_call(
|
|
"homeassistant",
|
|
"restart",
|
|
{},
|
|
blocking=False,
|
|
)
|
|
return self.async_create_entry(title="", data={})
|
|
|
|
return self.async_show_form(
|
|
step_id="confirm",
|
|
data_schema=vol.Schema({}),
|
|
)
|
|
|
|
|
|
async def async_create_fix_flow(
|
|
hass: HomeAssistant,
|
|
issue_id: str,
|
|
data: dict[str, str | int | float | None] | None,
|
|
) -> RepairsFlow:
|
|
"""Create a repairs flow for BCS fixable issues."""
|
|
if issue_id == RESTART_REQUIRED_ISSUE_ID:
|
|
return BCSRestartRequiredFlow(hass)
|
|
|
|
raise data_entry_flow.UnknownHandler |