Add attribute support to state selector (#77071)

homeassistant/helpers/selector.py

@@ -741,10 +741,11 @@ class TargetSelectorConfig(TypedDict, total=False):
     device: SingleDeviceSelectorConfig
 
 
-class StateSelectorConfig(TypedDict):
+class StateSelectorConfig(TypedDict, total=False):
     """Class to represent an state selector config."""
 
     entity_id: str
+    attribute: str
 
 
 @SELECTORS.register("state")
@@ -753,7 +754,12 @@ class StateSelector(Selector):
 
     selector_type = "state"
 
-    CONFIG_SCHEMA = vol.Schema({vol.Required("entity_id"): cv.entity_id})
+    CONFIG_SCHEMA = vol.Schema(
+        {
+            vol.Required("entity_id"): cv.entity_id,
+            vol.Optional("attribute"): str,
+        }
+    )
 
     def __init__(self, config: StateSelectorConfig) -> None:
         """Instantiate a selector."""

tests/helpers/test_selector.py

@@ -302,6 +302,11 @@ def test_time_selector_schema(schema, valid_selections, invalid_selections):
             ("on", "armed"),
             (None, True, 1),
         ),
+        (
+            {"entity_id": "sensor.abc", "attribute": "device_class"},
+            ("temperature", "humidity"),
+            (None,),
+        ),
     ),
 )
 def test_state_selector_schema(schema, valid_selections, invalid_selections):