Improve config flow dialogs

zhbjsh · Nov 1, 2024 · b0d37af · b0d37af
1 parent bd7f39a
commit b0d37af
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 12 deletions.
diff --git a/custom_components/ssh/config_flow.py b/custom_components/ssh/config_flow.py
@@ -65,6 +65,7 @@
 from homeassistant.data_entry_flow import FlowResult
 from homeassistant.helpers.device_registry import format_mac
 from homeassistant.helpers.selector import (
+    BooleanSelector,
     ObjectSelector,
     SelectOptionDict,
     SelectSelector,
@@ -361,15 +362,11 @@ async def async_step_init(
                     vol.Required(
                         CONF_ALLOW_TURN_OFF,
                         default=self._data[CONF_ALLOW_TURN_OFF],
-                    ): bool,
+                    ): BooleanSelector(),
                     vol.Required(
                         CONF_DISCONNECT_MODE,
                         default=self._data[CONF_DISCONNECT_MODE],
-                    ): bool,
-                    vol.Required(
-                        CONF_RESET_COMMANDS,
-                        default=False,
-                    ): bool,
+                    ): BooleanSelector(),
                     vol.Required(
                         CONF_UPDATE_INTERVAL,
                         default=self._data[CONF_UPDATE_INTERVAL],
@@ -386,6 +383,10 @@ async def async_step_init(
                         CONF_SENSOR_COMMANDS,
                         default=self._data[CONF_SENSOR_COMMANDS],
                     ): ListSelector(SENSOR_COMMAND_SCHEMA),
+                    vol.Required(
+                        CONF_RESET_COMMANDS,
+                        default=False,
+                    ): BooleanSelector(),
                 }
             ),
         )
@@ -411,11 +412,11 @@ async def async_step_reset_commands(
                     vol.Required(
                         CONF_RESET_DEFAULT_COMMANDS,
                         default=True,
-                    ): bool,
+                    ): BooleanSelector(),
                     vol.Required(
                         CONF_REMOVE_CUSTOM_COMMANDS,
                         default=False,
-                    ): bool,
+                    ): BooleanSelector(),
                 }
             ),
         )
@@ -633,13 +634,13 @@ async def async_step_user(
                         default=self._data.get(
                             CONF_ADD_HOST_KEYS, DEFAULT_ADD_HOST_KEYS
                         ),
-                    ): bool,
+                    ): BooleanSelector(),
                     vol.Required(
                         CONF_LOAD_SYSTEM_HOST_KEYS,
                         default=self._data.get(
                             CONF_LOAD_SYSTEM_HOST_KEYS, DEFAULT_LOAD_SYSTEM_HOST_KEYS
                         ),
-                    ): bool,
+                    ): BooleanSelector(),
                 }
             ),
         )

diff --git a/custom_components/ssh/strings.json b/custom_components/ssh/strings.json
@@ -55,11 +55,11 @@
         "data": {
           "allow_turn_off": "Allow to turn the device off",
           "disconnect_mode": "Disconnect between commands (experimental)",
-          "reset_commands": "Reset commands",
           "update_interval": "Update interval",
           "command_timeout": "Command timeout",
           "action_commands": "Action commands",
-          "sensor_commands": "Sensor commands"
+          "sensor_commands": "Sensor commands",
+          "reset_commands": "Reset commands"
         }
       },
       "reset_commands": {