59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
|
"""Repairs platform for HACS."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from homeassistant import data_entry_flow
|
||
|
|
from homeassistant.components.repairs import RepairsFlow
|
||
|
|
from homeassistant.core import HomeAssistant
|
||
|
|
import voluptuous as vol
|
||
|
|
|
||
|
|
from custom_components.hacs.base import HacsBase
|
||
|
|
|
||
|
|
from .const import DOMAIN
|
||
|
|
|
||
|
|
|
||
|
|
class RestartRequiredFixFlow(RepairsFlow):
|
||
|
|
"""Handler for an issue fixing flow."""
|
||
|
|
|
||
|
|
def __init__(self, issue_id: str) -> None:
|
||
|
|
self.issue_id = issue_id
|
||
|
|
|
||
|
|
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_restart()
|
||
|
|
|
||
|
|
async def async_step_confirm_restart(
|
||
|
|
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:
|
||
|
|
await self.hass.services.async_call("homeassistant", "restart")
|
||
|
|
return self.async_create_entry(title="", data={})
|
||
|
|
|
||
|
|
hacs: HacsBase = self.hass.data[DOMAIN]
|
||
|
|
integration = hacs.repositories.get_by_id(self.issue_id.split("_")[2])
|
||
|
|
|
||
|
|
return self.async_show_form(
|
||
|
|
step_id="confirm_restart",
|
||
|
|
data_schema=vol.Schema({}),
|
||
|
|
description_placeholders={"name": integration.display_name},
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
async def async_create_fix_flow(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
issue_id: str,
|
||
|
|
data: dict[str, str | int | float | None] | None = None,
|
||
|
|
*args: Any,
|
||
|
|
**kwargs: Any,
|
||
|
|
) -> RepairsFlow | None:
|
||
|
|
"""Create flow."""
|
||
|
|
if issue_id.startswith("restart_required"):
|
||
|
|
return RestartRequiredFixFlow(issue_id)
|
||
|
|
return None
|