Add homeassistant.add_area_to_floor service

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

@@ -0,0 +1,45 @@
+"""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 (
+    area_registry as ar,
+    config_validation as cv,
+    floor_registry as fr,
+)
+
+from ....services import AbstractSpookAdminService
+
+if TYPE_CHECKING:
+    from homeassistant.core import ServiceCall
+
+
+class SpookService(AbstractSpookAdminService):
+    """Home Assistant service to add an area to a floor."""
+
+    domain = DOMAIN
+    service = "add_area_to_floor"
+    schema = {
+        vol.Required("floor_id"): cv.string,
+        vol.Required("entity_id"): vol.All(cv.ensure_list, [cv.string]),
+    }
+
+    async def async_handle_service(self, call: ServiceCall) -> None:
+        """Handle the service call."""
+        floor_registry = fr.async_get(self.hass)
+        if not floor_registry.async_get_floor(call.data["floor_id"]):
+            msg = f"Floor {call.data['floor_id']} not found"
+            raise HomeAssistantError(msg)
+
+        area_registry = ar.async_get(self.hass)
+        for area_id in call.data["area_id"]:
+            area_registry.async_update(
+                area_id=area_id,
+                floor_id=call.data["floor_id"],
+            )

custom_components/spook/services.yaml

@@ -390,6 +390,26 @@ homeassistant_set_floor_aliases:
       selector:
         object:
 
+homeassistant_add_area_to_floor:
+  name: Add an area to a floor 👻
+  description: >-
+    Adds an area to a floor. Please note, if the area is already on a floor,
+    it will be removed from the previous floor.
+  fields:
+    floor_id:
+      name: Floor ID
+      description: The ID of the floor to add the area on.
+      required: true
+      selector:
+        text:
+    area_id:
+      name: Area ID
+      description: The ID of the area(s) to add to the floor.
+      required: true
+      selector:
+        area:
+          multiple: true
+
 homeassistant_ignore_all_discovered:
   name: Ignore all currently discovered devices 👻
   description: >-