mirror of
https://github.com/home-assistant/core.git
synced 2025-11-09 02:49:40 +00:00
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
"""Repairs for ntfy integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
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 entity_registry as er
|
|
|
|
from .const import CONF_TOPIC
|
|
|
|
|
|
class TopicProtectedRepairFlow(RepairsFlow):
|
|
"""Handler for protected topic issue fixing flow."""
|
|
|
|
def __init__(self, data: dict[str, str]) -> None:
|
|
"""Initialize."""
|
|
self.entity_id = data["entity_id"]
|
|
self.topic = data["topic"]
|
|
|
|
async def async_step_init(
|
|
self, user_input: dict[str, str] | None = None
|
|
) -> data_entry_flow.FlowResult:
|
|
"""Init repair flow."""
|
|
|
|
return await self.async_step_confirm()
|
|
|
|
async def async_step_confirm(
|
|
self, user_input: dict[str, str] | None = None
|
|
) -> data_entry_flow.FlowResult:
|
|
"""Confirm repair flow."""
|
|
if user_input is not None:
|
|
er.async_get(self.hass).async_update_entity(
|
|
self.entity_id,
|
|
disabled_by=er.RegistryEntryDisabler.USER,
|
|
)
|
|
return self.async_create_entry(data={})
|
|
|
|
return self.async_show_form(
|
|
step_id="confirm",
|
|
data_schema=vol.Schema({}),
|
|
description_placeholders={CONF_TOPIC: self.topic},
|
|
)
|
|
|
|
|
|
async def async_create_fix_flow(
|
|
hass: HomeAssistant,
|
|
issue_id: str,
|
|
data: dict[str, str],
|
|
) -> RepairsFlow:
|
|
"""Create flow."""
|
|
if issue_id.startswith("topic_protected"):
|
|
return TopicProtectedRepairFlow(data)
|
|
return ConfirmRepairFlow()
|