Fix missing Ecovacs station actions (home-assistant#155237)

Author: edenhaus

homeassistant/components/ecovacs/button.py

@@ -16,7 +16,7 @@
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
 from . import EcovacsConfigEntry
-from .const import SUPPORTED_LIFESPANS
+from .const import SUPPORTED_LIFESPANS, SUPPORTED_STATION_ACTIONS
 from .entity import (
     EcovacsCapabilityEntityDescription,
     EcovacsDescriptionEntity,
@@ -62,7 +62,7 @@ class EcovacsStationActionButtonEntityDescription(ButtonEntityDescription):
         key=f"station_action_{action.name.lower()}",
         translation_key=f"station_action_{action.name.lower()}",
     )
-    for action in StationAction
+    for action in SUPPORTED_STATION_ACTIONS
 )
 
 

homeassistant/components/ecovacs/const.py

@@ -23,7 +23,11 @@
     LifeSpan.STATION_FILTER,
 )
 
-SUPPORTED_STATION_ACTIONS = (StationAction.EMPTY_DUSTBIN,)
+SUPPORTED_STATION_ACTIONS = (
+    StationAction.CLEAN_BASE,
+    StationAction.DRY_MOP,
+    StationAction.EMPTY_DUSTBIN,
+)
 
 LEGACY_SUPPORTED_LIFESPANS = (
     "main_brush",

homeassistant/components/ecovacs/icons.json

@@ -36,6 +36,12 @@
       "reset_lifespan_round_mop": {
         "default": "mdi:broom"
       },
+      "station_action_clean_base": {
+        "default": "mdi:home"
+      },
+      "station_action_dry_mop": {
+        "default": "mdi:broom"
+      },
       "station_action_empty_dustbin": {
         "default": "mdi:delete-restore"
       }

homeassistant/components/ecovacs/strings.json

@@ -70,6 +70,12 @@
       "reset_lifespan_side_brush": {
         "name": "Reset side brush lifespan"
       },
+      "station_action_clean_base": {
+        "name": "Clean base"
+      },
+      "station_action_dry_mop": {
+        "name": "Dry mop"
+      },
       "station_action_empty_dustbin": {
         "name": "Empty dustbin"
       }