40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.homeassistant import DOMAIN
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import config_validation as cv
|
|
|
|
from ....services import AbstractSpookAdminService
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import ServiceCall
|
|
|
|
|
|
class SpookService(AbstractSpookAdminService):
|
|
"""Home Assistant Core integration service to disable polling."""
|
|
|
|
domain = DOMAIN
|
|
service = "disable_polling"
|
|
schema = {vol.Required("config_entry_id"): cv.string}
|
|
|
|
async def async_handle_service(self, call: ServiceCall) -> None:
|
|
"""Handle the service call."""
|
|
if not (
|
|
entry := self.hass.config_entries.async_get_entry(
|
|
call.data["config_entry_id"],
|
|
)
|
|
):
|
|
msg = f"Config entry not found: {call.data['config_entry_id']}"
|
|
raise HomeAssistantError(msg)
|
|
|
|
self.hass.config_entries.async_update_entry(
|
|
entry,
|
|
pref_disable_polling=True,
|
|
)
|