Merge pull request #190 from ocrease/TimerHold-182-v2

Fix for timer hold service targeting all switch entities on device

custom_components/heatmiserneo/switch.py

@@ -7,6 +7,7 @@
 """
 
 from datetime import time
+from enum import IntFlag
 import logging
 
 import voluptuous as vol
@@ -76,9 +77,16 @@ async def async_setup_entry(hass, entry, async_add_entities):
             vol.Required(ATTR_HOLD_DURATION, default=1): cv.positive_time_period,
         },
         "async_turn_on",
+        [HeatmiseerNeoSwitchEntityFeature.HOLD],
     )
 
 
+class HeatmiseerNeoSwitchEntityFeature(IntFlag):
+    """Supported features of the heatmiser neo switches entity."""
+
+    HOLD = 1
+
+
 class HeatmiserNeoPlugPowerSwitch(CoordinatorEntity, SwitchEntity):
     """Represents a Heatmiser Neo Plug"""
 
@@ -411,6 +419,8 @@ async def async_turn_off(self, **kwargs):
 class NeoTimerEntity(CoordinatorEntity, SwitchEntity):
     """Represents a Heatmiser neoStat thermostat acting in TimeClock mode."""
 
+    _attr_supported_features = HeatmiseerNeoSwitchEntityFeature.HOLD
+
     def __init__(
         self, neostat: NeoStat, coordinator: DataUpdateCoordinator, hub: NeoHub
     ):