208 files

custom_components/spook/ectoplasms/homeassistant/services/enable_config_entry.py

@@ -4,12 +4,10 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING
 
-import voluptuous as vol
-
 from homeassistant.components.homeassistant import DOMAIN
-from homeassistant.helpers import config_validation as cv
 
 from ....services import AbstractSpookAdminService
+from . import CONFIG_ENTRY_SERVICE_SCHEMA, async_config_entry_ids
 
 if TYPE_CHECKING:
     from homeassistant.core import ServiceCall
@@ -20,11 +18,11 @@ class SpookService(AbstractSpookAdminService):
 
     domain = DOMAIN
     service = "enable_config_entry"
-    schema = {vol.Required("config_entry_id"): vol.All(cv.ensure_list, [cv.string])}
+    schema = CONFIG_ENTRY_SERVICE_SCHEMA
 
     async def async_handle_service(self, call: ServiceCall) -> None:
         """Handle the service call."""
-        for config_entry_id in call.data["config_entry_id"]:
+        for config_entry_id in async_config_entry_ids(self.hass, call):
             await self.hass.config_entries.async_set_disabled_by(
                 config_entry_id,
                 disabled_by=None,