Add sensor entities for the MultistateInput cluster

zha/application/platforms/sensor/__init__.py

@@ -65,6 +65,7 @@
     CLUSTER_HANDLER_HUMIDITY,
     CLUSTER_HANDLER_ILLUMINANCE,
     CLUSTER_HANDLER_LEAF_WETNESS,
+    CLUSTER_HANDLER_MULTISTATE_INPUT,
     CLUSTER_HANDLER_POWER_CONFIGURATION,
     CLUSTER_HANDLER_PRESSURE,
     CLUSTER_HANDLER_SMARTENERGY_METERING,
@@ -502,6 +503,38 @@ def formatter(self, value: int) -> str | None:
         return self._enum(value).name
 
 
+@CONFIG_DIAGNOSTIC_MATCH(cluster_handler_names=CLUSTER_HANDLER_MULTISTATE_INPUT)
+class MultiStateInputSensor(EnumSensor):
+    """Sensor that displays enumerated values."""
+
+    _attribute_name = "present_value"
+    _unique_id_suffix = "present_value"
+    _attr_translation_key: str = "multistate_input"
+    _attr_entity_registry_enabled_default = False
+    _attr_extra_state_attribute_names: set[str] = {
+        "state_text",
+        "description",
+        "number_of_states",
+        "out_of_service",
+        "reliability",
+        "status_flags",
+        "application_type",
+    }
+    _enum = enum.Enum("Empty", names=())
+
+    def __init__(
+        self,
+        unique_id: str,
+        cluster_handlers: list[ClusterHandler],
+        endpoint: Endpoint,
+        device: Device,
+        **kwargs: Any,
+    ) -> None:
+        """Init this sensor."""
+        super().__init__(unique_id, cluster_handlers, endpoint, device, **kwargs)
+        self._enum = enum.Enum("state_text", self._cluster_handler.state_text.value)
+
+
 @MULTI_MATCH(
     cluster_handler_names=CLUSTER_HANDLER_ANALOG_INPUT,
     manufacturers="Digi",

zha/zigbee/cluster_handlers/general.py

@@ -390,6 +390,61 @@ class MultistateInputClusterHandler(ClusterHandler):
             config=REPORT_CONFIG_DEFAULT,
         ),
     )
+    ZCL_INIT_ATTRS = {
+        MultistateInput.AttributeDefs.state_text.name,
+        MultistateInput.AttributeDefs.description.name,
+        MultistateInput.AttributeDefs.number_of_states.name,
+        MultistateInput.AttributeDefs.out_of_service.name,
+        MultistateInput.AttributeDefs.reliability.name,
+        MultistateInput.AttributeDefs.status_flags.name,
+        MultistateInput.AttributeDefs.application_type.name,
+    }
+
+    @property
+    def present_value(self) -> int | None:
+        """Return cached value of present_value."""
+        return self.cluster.get(MultistateInput.AttributeDefs.present_value.name)
+
+    @property
+    def state_text(self) -> t.LVList | None:
+        """Return cached value of state_text."""
+        return self.cluster.get(MultistateInput.AttributeDefs.state_text.name)
+
+    @property
+    def description(self) -> str | None:
+        """Return cached value of description."""
+        return self.cluster.get(MultistateInput.AttributeDefs.description.name)
+
+    @property
+    def number_of_states(self) -> int | None:
+        """Return cached value of number_of_states."""
+        return self.cluster.get(MultistateInput.AttributeDefs.number_of_states.name)
+
+    @property
+    def out_of_service(self) -> bool | None:
+        """Return cached value of out_of_service."""
+        return self.cluster.get(MultistateInput.AttributeDefs.out_of_service.name)
+
+    @property
+    def reliability(self) -> int | None:
+        """Return cached value of reliability."""
+        return self.cluster.get(MultistateInput.AttributeDefs.reliability.name)
+
+    @property
+    def status_flags(self) -> int | None:
+        """Return cached value of status_flags."""
+        return self.cluster.get(MultistateInput.AttributeDefs.status_flags.name)
+
+    @property
+    def application_type(self) -> int | None:
+        """Return cached value of application_type."""
+        return self.cluster.get(MultistateInput.AttributeDefs.application_type.name)
+
+    async def async_update(self):
+        """Update cluster value attribute."""
+        await self.get_attribute_value(
+            MultistateInput.AttributeDefs.present_value.name, from_cache=False
+        )
 
 
 @registries.CLUSTER_HANDLER_REGISTRY.register(MultistateOutput.cluster_id)