Add vehicle select to Ohme (#139795)

* Add vehicle select to Ohme

* mypy fixes

* Update homeassistant/components/ohme/select.py

Co-authored-by: Josef Zweck <josef@zweck.dev>

---------

Co-authored-by: Josef Zweck <josef@zweck.dev>

homeassistant/components/ohme/icons.json

@@ -16,6 +16,9 @@
     "select": {
       "charge_mode": {
         "default": "mdi:play-box"
+      },
+      "vehicle": {
+        "default": "mdi:car"
       }
     },
     "sensor": {

homeassistant/components/ohme/select.py

@@ -25,10 +25,12 @@ class OhmeSelectDescription(OhmeEntityDescription, SelectEntityDescription):
     """Class to describe an Ohme select entity."""
 
     select_fn: Callable[[OhmeApiClient, Any], Awaitable[None]]
+    options: list[str] | None = None
+    options_fn: Callable[[OhmeApiClient], list[str]] | None = None
     current_option_fn: Callable[[OhmeApiClient], str | None]
 
 
-SELECT_DESCRIPTION: Final[OhmeSelectDescription] = OhmeSelectDescription(
+MODE_SELECT_DESCRIPTION: Final[OhmeSelectDescription] = OhmeSelectDescription(
     key="charge_mode",
     translation_key="charge_mode",
     select_fn=lambda client, mode: client.async_set_mode(mode),
@@ -37,6 +39,14 @@ SELECT_DESCRIPTION: Final[OhmeSelectDescription] = OhmeSelectDescription(
     available_fn=lambda client: client.mode is not None,
 )
 
+VEHICLE_SELECT_DESCRIPTION: Final[OhmeSelectDescription] = OhmeSelectDescription(
+    key="vehicle",
+    translation_key="vehicle",
+    select_fn=lambda client, selection: client.async_set_vehicle(selection),
+    options_fn=lambda client: client.vehicles,
+    current_option_fn=lambda client: client.current_vehicle or None,
+)
+
 
 async def async_setup_entry(
     hass: HomeAssistant,
@@ -44,9 +54,15 @@ async def async_setup_entry(
     async_add_entities: AddConfigEntryEntitiesCallback,
 ) -> None:
     """Set up Ohme selects."""
-    coordinator = config_entry.runtime_data.charge_session_coordinator
+    charge_sessions_coordinator = config_entry.runtime_data.charge_session_coordinator
+    device_info_coordinator = config_entry.runtime_data.device_info_coordinator
 
-    async_add_entities([OhmeSelect(coordinator, SELECT_DESCRIPTION)])
+    async_add_entities(
+        [
+            OhmeSelect(charge_sessions_coordinator, MODE_SELECT_DESCRIPTION),
+            OhmeSelect(device_info_coordinator, VEHICLE_SELECT_DESCRIPTION),
+        ]
+    )
 
 
 class OhmeSelect(OhmeEntity, SelectEntity):
@@ -64,6 +80,14 @@ class OhmeSelect(OhmeEntity, SelectEntity):
             ) from e
         await self.coordinator.async_request_refresh()
 
+    @property
+    def options(self) -> list[str]:
+        """Return a set of selectable options."""
+        if self.entity_description.options_fn:
+            return self.entity_description.options_fn(self.coordinator.client)
+        assert self.entity_description.options
+        return self.entity_description.options
+
     @property
     def current_option(self) -> str | None:
         """Return the current selected option."""

homeassistant/components/ohme/strings.json

@@ -66,6 +66,9 @@
           "max_charge": "Max charge",
           "paused": "Paused"
         }
+      },
+      "vehicle": {
+        "name": "Vehicle"
       }
     },
     "sensor": {

tests/components/ohme/conftest.py

@@ -66,4 +66,6 @@ def mock_client():
             "model": "Home Pro",
             "sw_version": "v2.65",
         }
+        client.vehicles = ["Nissan Leaf", "Tesla Model 3"]
+        client.current_vehicle = "Nissan Leaf"
         yield client

tests/components/ohme/snapshots/test_select.ambr

@@ -57,3 +57,59 @@
     'state': 'unknown',
   })
 # ---
+# name: test_selects[select.ohme_home_pro_vehicle-entry]
+  EntityRegistryEntrySnapshot({
+    'aliases': set({
+    }),
+    'area_id': None,
+    'capabilities': dict({
+      'options': list([
+        'Nissan Leaf',
+        'Tesla Model 3',
+      ]),
+    }),
+    'config_entry_id': <ANY>,
+    'config_subentry_id': <ANY>,
+    'device_class': None,
+    'device_id': <ANY>,
+    'disabled_by': None,
+    'domain': 'select',
+    'entity_category': None,
+    'entity_id': 'select.ohme_home_pro_vehicle',
+    'has_entity_name': True,
+    'hidden_by': None,
+    'icon': None,
+    'id': <ANY>,
+    'labels': set({
+    }),
+    'name': None,
+    'options': dict({
+    }),
+    'original_device_class': None,
+    'original_icon': None,
+    'original_name': 'Vehicle',
+    'platform': 'ohme',
+    'previous_unique_id': None,
+    'supported_features': 0,
+    'translation_key': 'vehicle',
+    'unique_id': 'chargerid_vehicle',
+    'unit_of_measurement': None,
+  })
+# ---
+# name: test_selects[select.ohme_home_pro_vehicle-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'Ohme Home Pro Vehicle',
+      'options': list([
+        'Nissan Leaf',
+        'Tesla Model 3',
+      ]),
+    }),
+    'context': <ANY>,
+    'entity_id': 'select.ohme_home_pro_vehicle',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'Nissan Leaf',
+  })
+# ---