✨ Add switch_capacity

Author: MateoGreil

README.md

@@ -64,6 +64,15 @@ print(time_series_data)
 device = client.get_device(devices[0]['id'])
 device['configuration']['controlMode'] = 'MANUAL'
 client.put_device(device['id'], device)
+
+# Switch the POWER_SWITCH capacity
+for feature in device['features']:
+    for capacity in feature['capacities']:
+        if capacity.get('capacity', {}).get('nature') == "POWER_SWITCH":
+            capacity_id = capacity['capacity']['id']
+client.switch_capacity(capacity_id, False)
+client.switch_capacity(capacity_id, True)
+
 ```
 
 Make sure to replace `'username'`, `'password'` with the actual values for your Comwatt account.

comwatt_client/client.py

@@ -265,3 +265,23 @@ def get_device_ts_time_ago(self, device_id,
             return response.json()
         else:
             raise Exception(f'Error retrieving aggregations: {response.status_code}')
+
+    def switch_capacity(self, capacity_id, enable):
+        """
+        Switch a specific capcaity to the enable value.
+
+        Args:
+            capacity_id (str): The ID of the capacity.
+            enable (bool): The target state.
+
+        Returns:
+            dict: A dictionary containing the response from the API.
+
+        """
+        url = f'{self.base_url}/capacities/{capacity_id}/switch?enable={enable}'
+
+        response = self.session.put(url)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception(f'Error retrieving sites: {response.status_code}')