From dc4dd5a622ef1fe746f4c8d52703d949d964fc19 Mon Sep 17 00:00:00 2001
From: andy blair <andy.blair@iotics.com>
Date: Wed, 11 Dec 2024 15:25:31 +0000
Subject: [PATCH] feat: ei-3552 - Manage connections

---
 nyx_client/nyx_client/client.py     | 54 +++++++++++++++++++++++++++++
 nyx_client/nyx_client/connection.py | 13 +++++--
 2 files changed, 65 insertions(+), 2 deletions(-)

diff --git a/nyx_client/nyx_client/client.py b/nyx_client/nyx_client/client.py
index ce02f06..6fdb3e8 100644
--- a/nyx_client/nyx_client/client.py
+++ b/nyx_client/nyx_client/client.py
@@ -953,3 +953,57 @@ def get_connections(self, allow_upload: bool | None = None) -> list[Connection]:
         connections = self._nyx_get(NYX_CONNECTIONS_ENDPOINT, params=params)
 
         return [Connection.from_dict(c) for c in connections]
+
+    def delete_connection(self, connection: Connection):
+        """Deletes a connection.
+
+        Args:
+            connection: The connection to be deleted.
+
+        Raises:
+            requests.HTTPError: If the API request fails.
+        """
+        self._nyx_delete(f"{NYX_CIRCLE_ENDPOINT}/{connection.id}")
+
+    def create_connection(
+        self, name: str, description: str, json_blob: dict[str, Any], secret_json_blob: dict[str, Any]
+    ) -> Connection:
+        """Create a connection.
+
+        Args:
+            name: the connection name
+            description: a description of the connection
+            json_blob: a dictionary of attributes that are non-sensitive
+            secret_json_blob: a dictionary of sensitive attributes, such as authorization keys
+
+        Returns:
+            An `Connection` object
+
+        Raises:
+            requests.HTTPError: If the API request fails.
+        """
+        connection_json = {
+            "name": name,
+            "description": description,
+            "json_blob": json_blob,
+            "secret_json_blob": secret_json_blob,
+        }
+        resp = self._nyx_post(NYX_CONNECTIONS_ENDPOINT, data=connection_json)
+
+        connection_json["id"] = resp["id"]
+        return Connection.from_dict(connection_json)
+
+    def update_connection(self, connection: Connection, secret_json_blob: dict[str, Any] | None = None):
+        """Updates a connection.
+
+        Args:
+            connection: The connection to be updated.
+            secret_json_blob: optional, if provided it will update entries in the secrets
+
+        Raises:
+            requests.HTTPError: If the API request fails.
+        """
+        connection_json = connection.as_dict()
+        if secret_json_blob:
+            connection_json["secret_json_blob"] = secret_json_blob
+        self._nyx_patch(f"{NYX_CONNECTIONS_ENDPOINT}/{connection.id}", data=connection_json)
diff --git a/nyx_client/nyx_client/connection.py b/nyx_client/nyx_client/connection.py
index b5d21f3..1d4a7b7 100644
--- a/nyx_client/nyx_client/connection.py
+++ b/nyx_client/nyx_client/connection.py
@@ -14,11 +14,11 @@
 
 """Module for connection (3rd party data integrations) definitions in Nyx."""
 
-from dataclasses import dataclass
+from dataclasses import asdict, dataclass
 from typing import Any
 
 
-@dataclass(frozen=True)
+@dataclass
 class Connection:
     """Represents a connection to a 3rd party data integration."""
 
@@ -50,3 +50,12 @@ def from_dict(cls, value: dict[str, Any]) -> "Connection":
             description=value.get("description", ""),
             allow_update=value.get("allow_upload", False),
         )
+
+    def as_dict(self) -> dict[str, Any]:
+        """Returns the object as a dictionary.
+
+        Returns:
+            A dictionary of the connection, that matches POST/PUT requests in the API.
+
+        """
+        return asdict(self)