Add well-known attributes methods ofr airConditionerMode, airConditionerFanMode, and airFlowDirection

Author: andrewsayre

pysmartthings/device.py

@@ -541,6 +541,32 @@ def air_conditioner_mode(self) -> Optional[str]:
         """Get the air conditioner mode attribute."""
         return self._attributes[Attribute.air_conditioner_mode].value
 
+    @property
+    def supported_ac_modes(self) -> Sequence[str]:
+        """Get the supported AC modes attribute."""
+        value = self._attributes[Attribute.supported_ac_modes].value
+        if isinstance(value, Sequence):
+            return sorted(value)
+        return []
+
+    @property
+    def fan_mode(self) -> Optional[str]:
+        """Get the fan mode attribute."""
+        return self._attributes[Attribute.fan_mode].value
+
+    @property
+    def supported_ac_fan_modes(self) -> Sequence[str]:
+        """Get the supported AC fan modes attribute."""
+        value = self._attributes[Attribute.supported_ac_fan_modes].value
+        if isinstance(value, Sequence):
+            return sorted(value)
+        return []
+
+    @property
+    def air_flow_direction(self) -> str:
+        """Get the airFlowDirection attribute."""
+        return self._attributes[Attribute.air_flow_direction].value
+
     @property
     def three_axis(self) -> Optional[Tuple[int, int, int]]:
         """Get the three axis attribute."""
@@ -935,6 +961,28 @@ async def set_air_conditioner_mode(
                 Attribute.air_conditioner_mode, mode)
         return result
 
+    async def set_fan_mode(self, mode: str, *, set_status: bool = False,
+                           component_id: str = 'main'):
+        """Call the setFanMode command."""
+        result = await self.command(
+            component_id, Capability.air_conditioner_fan_mode,
+            Command.set_fan_mode, [mode])
+        if result and set_status:
+            self.status.update_attribute_value(Attribute.fan_mode, mode)
+        return result
+
+    async def set_air_flow_direction(
+            self, direction: str, *, set_status: bool = False,
+            component_id: str = 'main'):
+        """Call the setAirFlowDirection command."""
+        result = await self.command(
+            component_id, Capability.air_flow_direction,
+            Command.set_air_flow_direction, [direction])
+        if result and set_status:
+            self.status.update_attribute_value(
+                Attribute.air_flow_direction, direction)
+        return result
+
     @property
     def status(self):
         """Get the status entity of the device."""

tests/json/device_command_post_set_air_flow_direction.json

@@ -0,0 +1,10 @@
+{
+  "commands": [
+    {
+      "component": "main",
+      "capability": "airFlowDirection",
+      "command": "setAirFlowDirection",
+      "arguments": ["fixed"]
+    }
+  ]
+}

tests/json/device_command_post_set_fan_mode.json

@@ -0,0 +1,10 @@
+{
+  "commands": [
+    {
+      "component": "main",
+      "capability": "airConditionerFanMode",
+      "command": "setFanMode",
+      "arguments": ["auto"]
+    }
+  ]
+}

tests/test_device.py

@@ -771,6 +771,30 @@ async def test_set_air_conditioner_mode(api):
         assert await device.set_air_conditioner_mode('auto', set_status=True)
         assert device.status.air_conditioner_mode == 'auto'
 
+    @staticmethod
+    @pytest.mark.asyncio
+    async def test_set_fan_mode(api):
+        """Tests the set_fan_mode method."""
+        # Arrange
+        device = DeviceEntity(api, device_id=DEVICE_ID)
+        # Act/Assert
+        assert await device.set_fan_mode('auto')
+        assert device.status.fan_mode is None
+        assert await device.set_fan_mode('auto', set_status=True)
+        assert device.status.fan_mode == 'auto'
+
+    @staticmethod
+    @pytest.mark.asyncio
+    async def test_set_air_flow_direction(api):
+        """Tests the set_air_flow_direction method."""
+        # Arrange
+        device = DeviceEntity(api, device_id=DEVICE_ID)
+        # Act/Assert
+        assert await device.set_air_flow_direction('fixed')
+        assert device.status.air_flow_direction is None
+        assert await device.set_air_flow_direction('fixed', set_status=True)
+        assert device.status.air_flow_direction == 'fixed'
+
 
 class TestDeviceStatus:
     """Tests for the DeviceStatus class."""
@@ -1015,6 +1039,10 @@ def test_well_known_attributes():
         """Tests the humidity property."""
         # Arrange
         status = DeviceStatus(None, device_id=DEVICE_ID)
+
+        assert status.supported_ac_modes == []
+        assert status.supported_ac_fan_modes == []
+
         status.update_attribute_value(Attribute.humidity, 50)
         status.update_attribute_value(Attribute.temperature, 55)
         status.update_attribute_value(
@@ -1028,6 +1056,11 @@ def test_well_known_attributes():
         status.update_attribute_value(Attribute.window_shade, 'closed')
         status.update_attribute_value(Attribute.data, {'test': 'test'})
         status.update_attribute_value(Attribute.three_axis, [0, 0, 0])
+        status.update_attribute_value(
+            Attribute.supported_ac_modes, ['auto', 'cool'])
+        status.update_attribute_value(Attribute.fan_mode, 'low')
+        status.update_attribute_value(
+            Attribute.supported_ac_fan_modes, ['auto', 'low'])
         # Act/Assert
         assert status.humidity == 50
         assert status.temperature == 55
@@ -1039,6 +1072,8 @@ def test_well_known_attributes():
         assert status.window_shade == 'closed'
         assert status.data == {'test': 'test'}
         assert status.three_axis == [0, 0, 0]
+        assert status.supported_ac_modes == ['auto', 'cool']
+        assert status.supported_ac_fan_modes == ['auto', 'low']
 
     @staticmethod
     def test_well_known_ocf_attributes():