diaspora-event-sdk 0.1.5__tar.gz → 0.1.7__tar.gz

{diaspora-event-sdk-0.1.5/diaspora_event_sdk.egg-info → diaspora-event-sdk-0.1.7}/PKG-INFO

@@ -1,22 +1,14 @@
 Metadata-Version: 2.1
 Name: diaspora-event-sdk
-Version: 0.1.5
+Version: 0.1.7
 Summary: SDK of Diaspora Event Fabric: Resilience-enabling services for science from HPC to edge
 Home-page: https://github.com/globus-labs/diaspora-event-sdk
 License: LICENSE
+Platform: UNKNOWN
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: globus-sdk<4,>=3.20.1
 Provides-Extra: kafka-python
-Requires-Dist: kafka-python; extra == "kafka-python"
-Requires-Dist: aws-msk-iam-sasl-signer-python; extra == "kafka-python"
 Provides-Extra: test
-Requires-Dist: pytest; extra == "test"
-Requires-Dist: pytest-cov; extra == "test"
-Requires-Dist: coverage; extra == "test"
-Requires-Dist: mypy; extra == "test"
-Requires-Dist: tox; extra == "test"
-Requires-Dist: check-manifest; extra == "test"
+License-File: LICENSE
 
 # Diaspora: Resilience-enabling services for science from HPC to edge
 
@@ -47,3 +39,4 @@ Note: This method does not include dependencies for `KafkaProducer` and `KafkaCo
 **Advanced Consumers with Faust**: Explore the [Faust Streaming Guide](docs/faust_weather_app.md) for advanced event streaming with Faust.
 
 **Advanced Consumer Functions**: See our [Colab example](https://colab.research.google.com/drive/1tPKfxU2qPsLvNTreF6nKINU62k7pQWxa?usp=sharing) for demonstration.
+

{diaspora-event-sdk-0.1.5 → diaspora-event-sdk-0.1.7}/diaspora_event_sdk/sdk/_environments.py

@@ -1,7 +1,7 @@
 import os
 
 
-TOKEN_EXCHANGE = "http://3.220.110.101"
+TOKEN_EXCHANGE = "http://3.220.110.101/"
 DIASPORA_RESOURCE_SERVER = "2b9d2f5c-fa32-45b5-875b-b24cd343b917"
 
 

{diaspora-event-sdk-0.1.5 → diaspora-event-sdk-0.1.7}/diaspora_event_sdk/sdk/client.py

@@ -89,16 +89,6 @@ class Client:
                 "endpoint": tokens["endpoint"],
             }
 
-    # @requires_login  # TODO
-    # def get_secret_key(self):
-    #     tokens = self.login_manager._token_storage.get_token_data(
-    #         DIASPORA_RESOURCE_SERVER
-    #     )
-    #     if tokens is None or "secret_key" not in tokens:
-    #         return None
-    #     else:
-    #         return tokens["secret_key"]
-
     @requires_login
     def put_secret_key(self, access_key, secret_key, endpoint):
         tokens = self.login_manager._token_storage.get_token_data(
@@ -133,54 +123,74 @@ class Client:
         """
         Registers a new topic under the user's OpenID.
         """
-        return self.web_client.register_topic(self.subject_openid, topic)
+        return self.web_client.register_topic(self.subject_openid, topic, "register")
 
     @requires_login
     def unregister_topic(self, topic):
         """
         Unregisters a topic from the user's OpenID.
         """
-        return self.web_client.unregister_topic(self.subject_openid, topic)
+        return self.web_client.register_topic(self.subject_openid, topic, "unregister")
+
+    @requires_login
+    def get_topic_configs(self, topic):
+        """
+        Get topic configurations.
+        """
+        return self.web_client.get_topic_configs(self.subject_openid, topic)
 
     @requires_login
-    def register_topic_for_user(self, topic, user):
+    def update_topic_configs(self, topic, configs):
+        """
+        Set topic configurations.
+        """
+        return self.web_client.update_topic_configs(self.subject_openid, topic, configs)
+
+    @requires_login
+    def update_topic_partitions(self, topic, new_partitions):
+        """
+        Adjust topic number of partitions
+        """
+        return self.web_client.update_topic_partitions(self.subject_openid, topic, new_partitions)
+
+    @requires_login
+    def grant_user_access(self, topic, user):
         """
         Registers a new topic under the user's OpenID.
         """
-        return self.web_client.register_topic_for_user(self.subject_openid, topic, user)
+        return self.web_client.grant_user_access(self.subject_openid, topic, user, "grant")
 
     @requires_login
-    def unregister_topic_for_user(self, topic, user):
+    def revoke_user_access(self, topic, user):
         """
         Unregisters a topic from the user's OpenID.
         """
-        return self.web_client.unregister_topic_for_user(self.subject_openid, topic, user)
-
+        return self.web_client.grant_user_access(self.subject_openid, topic, user, "revoke")
 
     @requires_login
-    def list_functions(self):
+    def list_triggers(self):
         """
         Retrieves the list of functions associated with the user's OpenID.
         """
-        return self.web_client.list_functions(self.subject_openid)
+        return self.web_client.list_triggers(self.subject_openid)
 
     @requires_login
-    def register_function(self, topic, function, function_configs):
+    def create_trigger(self, topic, function, function_configs):
         """
         Registers a new functions under the user's OpenID.
         """
-        return self.web_client.register_function(self.subject_openid, topic, function, function_configs)
+        return self.web_client.create_trigger(self.subject_openid, topic, function, "create", function_configs)
 
     @requires_login
-    def unregister_function(self, topic, function):
+    def delete_trigger(self, topic, function):
         """
         Unregisters a functions from the user's OpenID.
         """
-        return self.web_client.unregister_function(self.subject_openid, topic, function)
-    
+        return self.web_client.create_trigger(self.subject_openid, topic, function, "delete", {})
+
     @requires_login
-    def update_trigger_config(self, trigger_uuid, trigger_configs):
+    def update_trigger(self, trigger_uuid, trigger_configs):
         """
         Update a functions's trigger'.
         """
-        return self.web_client.update_function_trigger(self.subject_openid, trigger_uuid, trigger_configs)
+        return self.web_client.update_trigger(self.subject_openid, trigger_uuid, trigger_configs)

diaspora-event-sdk-0.1.7/diaspora_event_sdk/sdk/web_client.py

@@ -0,0 +1,99 @@
+from typing import Optional
+import json
+import globus_sdk
+from diaspora_event_sdk.sdk.utils.uuid_like import UUID_LIKE_T
+
+from ._environments import TOKEN_EXCHANGE
+
+
+class WebClient(globus_sdk.BaseClient):
+    def __init__(
+        self,
+        *,
+        environment: Optional[str] = None,
+        base_url: Optional[str] = None,
+        app_name: Optional[str] = None,
+        **kwargs,
+    ):
+        if base_url is None:
+            base_url = TOKEN_EXCHANGE
+
+        super().__init__(environment=environment, base_url=base_url, **kwargs)
+
+        self._user_app_name = None
+        self.user_app_name = app_name
+
+    def create_key(self, subject: UUID_LIKE_T) -> globus_sdk.GlobusHTTPResponse:
+        return self.get("/api/v2/create_key", headers={"Subject": str(subject)})
+
+    def list_topics(self, subject: UUID_LIKE_T) -> globus_sdk.GlobusHTTPResponse:
+        return self.get("/api/v2/topics", headers={"Subject": str(subject)})
+
+    def register_topic(
+        self, subject: UUID_LIKE_T, topic: str, action: str
+    ) -> globus_sdk.GlobusHTTPResponse:
+        return self.put(
+            f"/api/v2/topic/{topic}", headers={"Subject": str(subject), "Action": action}
+        )
+
+    def get_topic_configs(
+        self, subject: UUID_LIKE_T, topic: str
+    ) -> globus_sdk.GlobusHTTPResponse:
+        return self.get(
+            f"/api/v2/topic/{topic}",
+            headers={"Subject": str(subject), "Topic": topic},
+            # data=json.dumps(configs).encode("utf-8")
+        )
+
+    def update_topic_configs(
+        self, subject: UUID_LIKE_T, topic: str, configs: dict
+    ) -> globus_sdk.GlobusHTTPResponse:
+        return self.post(
+            f"/api/v2/topic/{topic}",
+            headers={"Subject": str(subject), "Topic": topic,
+                     "Content-Type": "text/plain"},
+            data=json.dumps(configs).encode("utf-8")
+        )
+
+    def update_topic_partitions(
+        self, subject: UUID_LIKE_T, topic: str, new_partitions: int
+    ) -> globus_sdk.GlobusHTTPResponse:
+        return self.post(
+            f"/api/v2/topic/{topic}/partitions",
+            headers={"Subject": str(subject), "Topic": topic,
+                     "NewPartitions": str(new_partitions)},
+        )
+
+    def grant_user_access(
+        self, subject: UUID_LIKE_T, topic: str, user: UUID_LIKE_T, action: str
+    ) -> globus_sdk.GlobusHTTPResponse:
+        return self.post(
+            f"/api/v2/topic/{topic}/user",
+            headers={"Subject": str(subject), "Action": action,
+                     "Topic": topic, "User": str(user)}
+        )
+
+    def list_triggers(self, subject: UUID_LIKE_T) -> globus_sdk.GlobusHTTPResponse:
+        return self.get("/api/v2/triggers", headers={"Subject": str(subject)})
+
+    def create_trigger(
+        self, subject: UUID_LIKE_T, topic: str, function: str, action: str,
+        function_configs: dict
+    ) -> globus_sdk.GlobusHTTPResponse:
+        return self.put(
+            "/api/v2/trigger",
+            headers={"Subject": str(subject), "Topic": topic,
+                     "Trigger": function, "Action": action,
+                     "Content-Type": "text/plain"},
+            data=json.dumps(function_configs).encode("utf-8")
+        )
+
+    def update_trigger(
+        self, subject: UUID_LIKE_T, trigger_uuid: UUID_LIKE_T, trigger_configs: dict
+    ) -> globus_sdk.GlobusHTTPResponse:
+        return self.post(
+            f"/api/v2/triggers/{trigger_uuid}",
+            headers={"Subject": str(subject), "Trigger_id": str(trigger_uuid),
+                     "Content-Type": "text/plain"},
+            data=json.dumps(trigger_configs).encode("utf-8")
+        )

diaspora-event-sdk-0.1.7/diaspora_event_sdk/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.7"

{diaspora-event-sdk-0.1.5 → diaspora-event-sdk-0.1.7/diaspora_event_sdk.egg-info}/PKG-INFO

@@ -1,22 +1,14 @@
 Metadata-Version: 2.1
 Name: diaspora-event-sdk
-Version: 0.1.5
+Version: 0.1.7
 Summary: SDK of Diaspora Event Fabric: Resilience-enabling services for science from HPC to edge
 Home-page: https://github.com/globus-labs/diaspora-event-sdk
 License: LICENSE
+Platform: UNKNOWN
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: globus-sdk<4,>=3.20.1
 Provides-Extra: kafka-python
-Requires-Dist: kafka-python; extra == "kafka-python"
-Requires-Dist: aws-msk-iam-sasl-signer-python; extra == "kafka-python"
 Provides-Extra: test
-Requires-Dist: pytest; extra == "test"
-Requires-Dist: pytest-cov; extra == "test"
-Requires-Dist: coverage; extra == "test"
-Requires-Dist: mypy; extra == "test"
-Requires-Dist: tox; extra == "test"
-Requires-Dist: check-manifest; extra == "test"
+License-File: LICENSE
 
 # Diaspora: Resilience-enabling services for science from HPC to edge
 
@@ -47,3 +39,4 @@ Note: This method does not include dependencies for `KafkaProducer` and `KafkaCo
 **Advanced Consumers with Faust**: Explore the [Faust Streaming Guide](docs/faust_weather_app.md) for advanced event streaming with Faust.
 
 **Advanced Consumer Functions**: See our [Colab example](https://colab.research.google.com/drive/1tPKfxU2qPsLvNTreF6nKINU62k7pQWxa?usp=sharing) for demonstration.
+

diaspora-event-sdk-0.1.5/diaspora_event_sdk/sdk/web_client.py

@@ -1,92 +0,0 @@
-from typing import Optional
-
-import globus_sdk
-from diaspora_event_sdk.sdk.utils.uuid_like import UUID_LIKE_T
-
-from ._environments import TOKEN_EXCHANGE
-
-
-class WebClient(globus_sdk.BaseClient):
-    def __init__(
-        self,
-        *,
-        environment: Optional[str] = None,
-        base_url: Optional[str] = None,
-        app_name: Optional[str] = None,
-        **kwargs,
-    ):
-        if base_url is None:
-            base_url = TOKEN_EXCHANGE
-
-        super().__init__(environment=environment, base_url=base_url, **kwargs)
-
-        self._user_app_name = None
-        self.user_app_name = app_name
-
-    def create_key(self, subject: UUID_LIKE_T) -> globus_sdk.GlobusHTTPResponse:
-        return self.post("/v1/create_key", headers={"Subject": str(subject)})
-
-    def list_topics(self, subject: UUID_LIKE_T) -> globus_sdk.GlobusHTTPResponse:
-        return self.get("/v1/list_topics", headers={"Subject": str(subject)})
-
-    def register_topic(
-        self, subject: UUID_LIKE_T, topic: str
-    ) -> globus_sdk.GlobusHTTPResponse:
-        return self.post(
-            "/v1/register_topic", headers={"Subject": str(subject), "Topic": topic}
-        )
-
-    def unregister_topic(
-        self, subject: UUID_LIKE_T, topic: str
-    ) -> globus_sdk.GlobusHTTPResponse:
-        return self.post(
-            "/v1/unregister_topic", headers={"Subject": str(subject), "Topic": topic}
-        )
-
-    def register_topic_for_user(
-        self, subject: UUID_LIKE_T, topic: str, user: UUID_LIKE_T
-    ) -> globus_sdk.GlobusHTTPResponse:
-        return self.post(
-            "/v1/register_topic_for_user",
-            headers={"Subject": str(
-                subject), "Topic": topic, "User": str(user)}
-        )
-
-    def unregister_topic_for_user(
-        self, subject: UUID_LIKE_T, topic: str, user: UUID_LIKE_T
-    ) -> globus_sdk.GlobusHTTPResponse:
-        return self.post(
-            "/v1/unregister_topic_for_user",
-            headers={"Subject": str(
-                subject), "Topic": topic, "User": str(user)}
-        )
-
-    def list_functions(self, subject: UUID_LIKE_T) -> globus_sdk.GlobusHTTPResponse:
-        return self.get("/v1/list_functions", headers={"Subject": str(subject)})
-
-    def register_function(
-        self, subject: UUID_LIKE_T, topic: str, function: str,
-        function_configs: dict
-    ) -> globus_sdk.GlobusHTTPResponse:
-        return self.post(
-            "/v1/register_function",
-            headers={"Subject": str(subject), "Topic": topic,
-                     "Function": function},
-            data=function_configs
-        )
-
-    def unregister_function(
-        self, subject: UUID_LIKE_T, topic: str, function: str
-    ) -> globus_sdk.GlobusHTTPResponse:
-        return self.post(
-            "/v1/unregister_function", headers={"Subject": str(subject), "Topic": topic, "Function": function}
-        )
-
-    def update_function_trigger(
-        self, subject: UUID_LIKE_T, trigger_uuid: UUID_LIKE_T, trigger_configs: dict
-    ) -> globus_sdk.GlobusHTTPResponse:
-        return self.post(
-            "/v1/update_function_trigger",
-            headers={"Subject": str(subject), "Trigger": str(trigger_uuid)},
-            data=trigger_configs
-        )

diaspora-event-sdk-0.1.5/diaspora_event_sdk/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.5"