✨ Add GET/PUT device

Author: MateoGreil

README.md

@@ -14,7 +14,8 @@ The client currently supports the following methods:
 - `get_site_consumption_breakdown_time_ago(self, site_id, aggregation_level = "HOUR", time_ago_unit = "DAY", time_ago_value = 1)` Retrieves the consumption breakdown data for a specific site, based on the provided parameters.
 - `get_devices(self, site_id)`: Retrieves a list of devices for the specified site.
 - `get_device_ts_time_ago(self, device_id, measure_kind = "FLOW", aggregation_level = "HOUR", aggregation_type = "MAX", time_ago_unit = "DAY", time_ago_value = "1")`: Retrieves the time series data for a specific device, based on the provided parameters.
-
+- `get_device(self, device_id)`: Retrieves information about a specific device.
+- `put_device(self, device_id, payload)`: Updates a specific device with the provided payload.
 
 ## Installation
 You can install the Comwatt Python Client using pip. Run the following command:
@@ -58,6 +59,11 @@ print(devices)
 # Get time series data for a specific device
 time_series_data = client.get_device_ts_time_ago(devices[0]['id'])
 print(time_series_data)
+
+# Set the control mode of a specific device to MANUL
+device = client.get_device(devices[0]['id'])
+device['configuration']['controlMode'] = 'MANUAL'
+client.put_device(device['id'], device)
 ```
 
 Make sure to replace `'username'`, `'password'` with the actual values for your Comwatt account.

comwatt_client/client.py

@@ -187,6 +187,46 @@ def get_devices(self, site_id):
         else:
             raise Exception(f'Error retrieving sites: {response.status_code}')
 
+    def get_device(self, device_id):
+        """
+        Retrieves information about a specific device.
+
+        Args:
+            device_id (str): The ID of the device.
+
+        Returns:
+            dict: A dictionary containing the device information.
+
+        """
+        url = f'{self.base_url}/devices/{device_id}'
+
+        response = self.session.get(url)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception(f'Error retrieving device {device_id}: {response.status_code}')
+
+    def put_device(self, device_id, payload):
+        """
+        Updates a specific device with the provided payload.
+
+        Args:
+            device_id (str): The ID of the device.
+            payload (dict): The payload to update the device.
+
+        Returns:
+            dict: A dictionary containing the response from the API.
+
+        """
+        url = f'{self.base_url}/devices/{device_id}'
+
+        response = self.session.put(url, json=payload)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception(f'Error retrieving sites: {response.status_code}')
+
+
     def get_device_ts_time_ago(self, device_id,
             measure_kind = "FLOW",
             aggregation_level = "HOUR",