wizata-dsapi 2.2.0.dev13__tar.gz → 2.2.0.dev14__tar.gz

{wizata_dsapi-2.2.0.dev13 → wizata_dsapi-2.2.0.dev14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.2.0.dev13
+Version: 2.2.0.dev14
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

wizata_dsapi-2.2.0.dev14/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "2.2.0.dev14"

{wizata_dsapi-2.2.0.dev13 → wizata_dsapi-2.2.0.dev14}/wizata_dsapi/wizata_dsapi_client.py

@@ -56,6 +56,7 @@ import ast
 from .deployment import Deployment, DeploymentStatus
 from .product import ProductKind, Product, ProductComponent
 from .inventory import InventoryBalance, InventoryStorage, InventoryStock
+from .connection import EdgeDeviceKey
 from .api_config import _registry
 
 from .streamlit_utils import get_streamlit_token, get_streamlit_domain
@@ -2989,6 +2990,43 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
         self._planner_request("DELETE", f"planner/inventory/balances/{self._balance_id(balance)}/",
                               expected=(204,))
 
+    # ------------------------------------------------------------ edge device keys
+
+    @staticmethod
+    def _edge_device_keys_route(edge_device_id) -> str:
+        if not edge_device_id:
+            raise ValueError("please provide the edge device id")
+        return f"edgedevices/{urllib.parse.quote(str(edge_device_id), safe='')}/keys"
+
+    def register_edge_device_key(self, edge_device_id: str, public_key: str) -> EdgeDeviceKey:
+        """
+        Register the public half of the key pair an edge generated for itself (DEV-6013).
+        Connection secrets are sealed for the edge's active key, and only that edge can
+        open them: the private half never leaves the device.
+
+        Registering the same key again is a no-op, so an edge can register on every start.
+        A different key retires the current one, and every secret sealed for that one must
+        be entered again.
+
+        :param edge_device_id: the device id the edge runs as, not its UUID.
+        :param public_key: PEM of the edge's X25519 public key.
+        :return: the edge's active key.
+        """
+        response = self._process_request(method="POST",
+                                         route=self._edge_device_keys_route(edge_device_id),
+                                         data=json.dumps({"publicKey": public_key}))
+        return EdgeDeviceKey.from_dict(response.json())
+
+    def get_edge_device_keys(self, edge_device_id: str) -> list:
+        """
+        :param edge_device_id: the device id the edge runs as, not its UUID.
+        :return: every EdgeDeviceKey the edge registered, newest first. With no active
+            one, the edge has not registered yet and no secret can be entered for it.
+        """
+        return self._process_request(method="GET",
+                                     route=self._edge_device_keys_route(edge_device_id),
+                                     dto_class=EdgeDeviceKey, result_type="list")
+
 
 def api() -> WizataDSAPIClient:
     """

{wizata_dsapi-2.2.0.dev13 → wizata_dsapi-2.2.0.dev14}/wizata_dsapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.2.0.dev13
+Version: 2.2.0.dev14
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

wizata_dsapi-2.2.0.dev13/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "2.2.0.dev13"