Add icon selection to room config flow

- Add IconSelector to config flow with dropdown of all HA icons
- Set default icon to mdi:texture-box
- Update translations for icon field
- Update stubs with IconSelector and IconSelectorConfig

Author: DefinitelyADev

custom_components/rooms/config_flow.py

@@ -47,6 +47,10 @@ async def async_step_user(self, user_input: Optional[Dict[str, Any]] = None) ->
             await self.async_set_unique_id(user_input[CONF_ROOM_NAME])
             self._abort_if_unique_id_configured()
 
+            # Ensure icon has a default value if not provided
+            if CONF_ICON not in user_input or user_input[CONF_ICON] is None:
+                user_input[CONF_ICON] = DEFAULT_ICON
+
             return self.async_create_entry(
                 title=user_input[CONF_ROOM_NAME],
                 data=user_input,
@@ -57,8 +61,8 @@ async def async_step_user(self, user_input: Optional[Dict[str, Any]] = None) ->
             data_schema=vol.Schema(
                 {
                     vol.Required(CONF_ROOM_NAME): str,
-                    vol.Optional(CONF_ICON, default=DEFAULT_ICON): getattr(selector, "IconSelector")(
-                        getattr(selector, "IconSelectorConfig")()
+                    vol.Optional(CONF_ICON): selector.IconSelector(
+                        selector.IconSelectorConfig(placeholder="mdi:texture-box")
                     ),
                     vol.Optional(CONF_POWER_ENTITY): selector.EntitySelector(
                         selector.EntitySelectorConfig(domain="sensor")

custom_components/rooms/strings.json

@@ -13,7 +13,8 @@
           "motion_entity": "Motion Sensor (optional)",
           "window_entity": "Window Sensor (optional)",
           "climate_entity": "Climate Entity (optional)",
-          "active_threshold": "Active Power Threshold (W)"
+          "active_threshold": "Active Power Threshold (W)",
+          "icon": "Icon (optional)"
         }
       }
     },
@@ -39,7 +40,8 @@
           "motion_entity": "Motion Sensor (optional)",
           "window_entity": "Window Sensor (optional)",
           "climate_entity": "Climate Entity (optional)",
-          "active_threshold": "Active Power Threshold (W)"
+          "active_threshold": "Active Power Threshold (W)",
+          "icon": "Icon (optional)"
         }
       }
     }

custom_components/rooms/translations/en.json

@@ -13,7 +13,8 @@
           "motion_entity": "Motion Sensor (optional)",
           "window_entity": "Window Sensor (optional)",
           "climate_entity": "Climate Entity (optional)",
-          "active_threshold": "Active Power Threshold (W)"
+          "active_threshold": "Active Power Threshold (W)",
+          "icon": "Icon (optional)"
         }
       }
     },
@@ -39,7 +40,8 @@
           "motion_entity": "Motion Sensor (optional)",
           "window_entity": "Window Sensor (optional)",
           "climate_entity": "Climate Entity (optional)",
-          "active_threshold": "Active Power Threshold (W)"
+          "active_threshold": "Active Power Threshold (W)",
+          "icon": "Icon (optional)"
         }
       }
     }

stubs/homeassistant/helpers/selector.py

@@ -8,7 +8,7 @@ class Selector:
 
     def __init__(
         self,
-        config: Union[Dict[str, Any], "EntitySelectorConfig", "TextSelectorConfig"],
+        config: Union[Dict[str, Any], "EntitySelectorConfig", "TextSelectorConfig", "IconSelectorConfig"],
     ) -> None:
         """Initialize selector."""
         ...
@@ -41,6 +41,18 @@ def __init__(
         ...
 
 
+class IconSelectorConfig:
+    """Configuration for icon selector."""
+
+    def __init__(
+        self,
+        *,
+        placeholder: Optional[str] = None,
+    ) -> None:
+        """Initialize icon selector config."""
+        ...
+
+
 class EntitySelector(Selector):
     """Entity selector."""
 
@@ -68,3 +80,16 @@ def __init__(
     ) -> None:
         """Initialize text selector."""
         ...
+
+
+class IconSelector(Selector):
+    """Icon selector."""
+
+    def __init__(
+        self,
+        config: Optional[Union[Dict[str, Any], IconSelectorConfig]] = None,
+        *,
+        placeholder: Optional[str] = None,
+    ) -> None:
+        """Initialize icon selector."""
+        ...