diaspora-event-sdk 0.2.9__py3-none-any.whl → 0.3.1__py3-none-any.whl

diaspora_event_sdk/__init__.py

@@ -1,6 +1,5 @@
-""" Diaspora Event Fabric: Resilience-enabling services for science from HPC to edge.
+"""Diaspora Event Fabric: Resilience-enabling services for science from HPC to edge."""
 
-"""
 from diaspora_event_sdk.version import __version__ as _version
 
 __author__ = "The Diaspora Event Team"

diaspora_event_sdk/sdk/_environments.py

@@ -1,9 +1,22 @@
 import os
+from typing import Union
 
 
-TOKEN_EXCHANGE = "http://3.220.110.101/"
+# TOKEN_EXCHANGE = "http://3.220.110.101/"
 DIASPORA_RESOURCE_SERVER = "2b9d2f5c-fa32-45b5-875b-b24cd343b917"
 
 
 def _get_envname():
     return os.getenv("DIASPORA_SDK_ENVIRONMENT", "production")
+
+
+def get_web_service_url(envname: Union[str, None] = None) -> str:
+    env = envname or _get_envname()
+    urls = {
+        "production": "https://diaspora-web-service.ml22sevubfnks.us-east-1.cs.amazonlightsail.com",
+        "dev": "https://diaspora-web-service-dev.ml22sevubfnks.us-east-1.cs.amazonlightsail.com",
+        "local": "http://localhost:8000",
+        "legacy": "http://3.220.110.101/",
+    }
+
+    return urls.get(env, urls["production"])

diaspora_event_sdk/sdk/aws_iam_msk.py

@@ -2,6 +2,7 @@
 #  SPDX-License-Identifier: Apache-2.0
 
 import base64
+
 # import logging
 from datetime import datetime, timezone
 from urllib.parse import parse_qs, urlparse
@@ -11,6 +12,7 @@ from urllib.parse import parse_qs, urlparse
 # import pkg_resources
 from .botocore.auth import SigV4QueryAuth
 from .botocore.awsrequest import AWSRequest
+
 # from botocore.config import Config
 from .botocore.credentials import Credentials
 
@@ -25,7 +27,7 @@ LIB_NAME = "aws-msk-iam-sasl-signer-python"
 
 
 def __get_user_agent__():
-    return (f"{LIB_NAME}/1.0.1")
+    return f"{LIB_NAME}/1.0.1"
 
 
 def __get_expiration_time_ms(request):
@@ -37,15 +39,17 @@ def __get_expiration_time_ms(request):
     # Parse the signed request
     parsed_url = urlparse(request.url)
     parsed_ul_params = parse_qs(parsed_url.query)
-    parsed_signing_time = datetime.strptime(parsed_ul_params['X-Amz-Date'][0],
-                                            "%Y%m%dT%H%M%SZ")
+    parsed_signing_time = datetime.strptime(
+        parsed_ul_params["X-Amz-Date"][0], "%Y%m%dT%H%M%SZ"
+    )
 
     # Make the datetime object timezone-aware
     signing_time = parsed_signing_time.replace(tzinfo=timezone.utc)
 
     # Convert the Unix timestamp to milliseconds
-    expiration_timestamp_seconds = int(
-        signing_time.timestamp()) + DEFAULT_TOKEN_EXPIRY_SECONDS
+    expiration_timestamp_seconds = (
+        int(signing_time.timestamp()) + DEFAULT_TOKEN_EXPIRY_SECONDS
+    )
 
     # Get lifetime of token
     expiration_timestamp_ms = expiration_timestamp_seconds * 1000
@@ -74,8 +78,7 @@ def __construct_auth_token(region, aws_credentials):
 
     # Create SigV4 instance
     sig_v4 = SigV4QueryAuth(
-        aws_credentials, SIGNING_NAME, region,
-        expires=DEFAULT_TOKEN_EXPIRY_SECONDS
+        aws_credentials, SIGNING_NAME, region, expires=DEFAULT_TOKEN_EXPIRY_SECONDS
     )
 
     # Create request with url and parameters
@@ -110,10 +113,12 @@ def generate_auth_token(region, aws_debug_creds=False):
 
     # Load credentials
     import os
+
     assert os.environ["AWS_ACCESS_KEY_ID"]
     assert os.environ["AWS_SECRET_ACCESS_KEY"]
 
-    aws_credentials = Credentials(os.environ["AWS_ACCESS_KEY_ID"],
-                                  os.environ["AWS_SECRET_ACCESS_KEY"])
+    aws_credentials = Credentials(
+        os.environ["AWS_ACCESS_KEY_ID"], os.environ["AWS_SECRET_ACCESS_KEY"]
+    )
 
     return __construct_auth_token(region, aws_credentials)

diaspora_event_sdk/sdk/client.py

@@ -7,7 +7,7 @@ from diaspora_event_sdk.sdk.login_manager import (
     requires_login,
 )
 
-from ._environments import DIASPORA_RESOURCE_SERVER, TOKEN_EXCHANGE
+from ._environments import DIASPORA_RESOURCE_SERVER, get_web_service_url
 
 
 class Client:
@@ -16,6 +16,8 @@ class Client:
         environment: Optional[str] = None,
         login_manager: Optional[LoginManagerProtocol] = None,
     ):
+        self.web_service_address = get_web_service_url(environment)
+
         # if a login manager was passed, no login flow is triggered
         if login_manager is not None:
             self.login_manager: LoginManagerProtocol = login_manager
@@ -26,7 +28,8 @@ class Client:
             self.login_manager.ensure_logged_in()
 
         self.web_client = self.login_manager.get_web_client(
-            base_url=TOKEN_EXCHANGE)
+            base_url=self.web_service_address
+        )
         self.auth_client = self.login_manager.get_auth_client()
         self.subject_openid = self.auth_client.oauth2_userinfo()["sub"]
 
@@ -151,7 +154,9 @@ class Client:
         """
         Increases the number of partitions for a given topic to the specified new partition count.
         """
-        return self.web_client.update_topic_partitions(self.subject_openid, topic, new_partitions)
+        return self.web_client.update_topic_partitions(
+            self.subject_openid, topic, new_partitions
+        )
 
     @requires_login
     def reset_topic(self, topic):
@@ -165,14 +170,18 @@ class Client:
         """
         Authorizes another user to access a registered topic under the invoker's account.
         """
-        return self.web_client.grant_user_access(self.subject_openid, topic, user, "grant")
+        return self.web_client.grant_user_access(
+            self.subject_openid, topic, user, "grant"
+        )
 
     @requires_login
     def revoke_user_access(self, topic, user):
         """
         Removes access permissions for another user from a registered topic under the invoker's account.
         """
-        return self.web_client.grant_user_access(self.subject_openid, topic, user, "revoke")
+        return self.web_client.grant_user_access(
+            self.subject_openid, topic, user, "revoke"
+        )
 
     @requires_login
     def list_topic_users(self, topic):
@@ -191,10 +200,16 @@ class Client:
     @requires_login
     def create_trigger(self, topic, function, function_configs, trigger_configs):
         """
-        Creates a new trigger under the user's account with specific function and invocation configurations. 
+        Creates a new trigger under the user's account with specific function and invocation configurations.
         """
         return self.web_client.create_trigger(
-            self.subject_openid, topic, function, "create", function_configs, trigger_configs)
+            self.subject_openid,
+            topic,
+            function,
+            "create",
+            function_configs,
+            trigger_configs,
+        )
 
     @requires_login
     def delete_trigger(self, topic, function):
@@ -202,27 +217,28 @@ class Client:
         Deletes a trigger and related AWS resources, while the associated topic remains unaffected.
         """
         return self.web_client.create_trigger(
-            self.subject_openid, topic, function, "delete", {}, {})
+            self.subject_openid, topic, function, "delete", {}, {}
+        )
 
     @requires_login
     def update_trigger(self, trigger_uuid, trigger_configs):
         """
         Updates invocation configurations of an existing trigger, identified by its unique trigger UUID.
         """
-        return self.web_client.update_trigger(self.subject_openid, trigger_uuid, trigger_configs)
+        return self.web_client.update_trigger(
+            self.subject_openid, trigger_uuid, trigger_configs
+        )
 
     @requires_login
     def list_log_streams(self, trigger):
         """
         List log streams of a trigger under the user's account
         """
-        return self.web_client.list_log_streams(
-            self.subject_openid, trigger)
+        return self.web_client.list_log_streams(self.subject_openid, trigger)
 
     @requires_login
     def get_log_events(self, trigger, stream):
         """
         Get events in a particular log stream of a trigger under the user's account
         """
-        return self.web_client.get_log_events(
-            self.subject_openid, trigger, stream)
+        return self.web_client.get_log_events(self.subject_openid, trigger, stream)

diaspora_event_sdk/sdk/kafka_client.py

@@ -17,7 +17,7 @@ try:
         def token(self):
             token, _ = generate_auth_token("us-east-1")
             return token
-except Exception as e:
+except Exception:
     kafka_available = False
 
 
@@ -90,7 +90,7 @@ def block_until_ready(max_minutes=5):
                 value={"message": "Synchronous message from Diaspora SDK"},
             )
             result["producer_connection_test"] = future.get(timeout=10)
-        except Exception as e:
+        except Exception:
             pass
 
     def consumer_connection_test(result):
@@ -103,7 +103,7 @@ def block_until_ready(max_minutes=5):
             for msg in consumer:
                 result["consumer_connection_test"] = msg
                 break
-        except Exception as e:
+        except Exception:
             pass
 
     result, retry_count = {}, 0

diaspora_event_sdk/sdk/login_manager/client_login.py

@@ -4,6 +4,7 @@ Logic for using client identities with the Diaspora SDK
 The design is based on the Globus CLI client login:
 https://github.com/globus/globus-cli/blob/main/src/globus_cli/login_manager/client_login.py
 """
+
 from __future__ import annotations
 
 import logging

diaspora_event_sdk/sdk/login_manager/decorators.py

@@ -3,8 +3,6 @@ import logging
 
 import globus_sdk
 
-from .._environments import TOKEN_EXCHANGE
-
 log = logging.getLogger(__name__)
 
 
@@ -27,7 +25,7 @@ def requires_login(func):
                 self.login_manager.run_login_flow()
                 # Initiate a new web client with updated authorizer
                 self.web_client = self.login_manager.get_web_client(
-                    base_url=TOKEN_EXCHANGE
+                    base_url=self.web_service_address
                 )
                 return func(self, *args, **kwargs)
 

diaspora_event_sdk/sdk/login_manager/login_flow.py

@@ -32,4 +32,4 @@ def do_link_auth_flow(scopes: List[str]):
     auth_code = input("Enter the resulting Authorization Code here: ").strip()
 
     # finish auth flow
-    return auth_client.oauth2_exchange_code_for_tokens(auth_code)
+    return auth_client.oauth2_exchange_code_for_tokens(auth_code)

diaspora_event_sdk/sdk/login_manager/protocol.py

@@ -16,14 +16,10 @@ else:
 
 @runtime_checkable
 class LoginManagerProtocol(Protocol):
-    def ensure_logged_in(self) -> None:
-        ...
+    def ensure_logged_in(self) -> None: ...
 
-    def logout(self) -> bool:
-        ...
+    def logout(self) -> bool: ...
 
-    def get_auth_client(self) -> globus_sdk.AuthClient:
-        ...
+    def get_auth_client(self) -> globus_sdk.AuthClient: ...
 
-    def get_web_client(self, *, base_url: str | None = None) -> WebClient:
-        ...
+    def get_web_client(self, *, base_url: str | None = None) -> WebClient: ...

diaspora_event_sdk/sdk/login_manager/tokenstore.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import json
 import os
 import pathlib
 
@@ -8,7 +7,6 @@ from globus_sdk.tokenstorage import SQLiteAdapter
 
 from .._environments import _get_envname
 from .client_login import get_client_login, is_client_login
-from .globus_auth import internal_auth_client
 
 
 def _home() -> pathlib.Path:

diaspora_event_sdk/sdk/web_client.py

@@ -5,7 +5,7 @@ import globus_sdk
 
 from diaspora_event_sdk.sdk.utils.uuid_like import UUID_LIKE_T
 
-from ._environments import TOKEN_EXCHANGE
+from ._environments import get_web_service_url
 
 
 class WebClient(globus_sdk.BaseClient):
@@ -18,7 +18,7 @@ class WebClient(globus_sdk.BaseClient):
         **kwargs,
     ):
         if base_url is None:
-            base_url = TOKEN_EXCHANGE
+            base_url = get_web_service_url(environment)
 
         super().__init__(environment=environment, base_url=base_url, **kwargs)
 
@@ -36,15 +36,14 @@ class WebClient(globus_sdk.BaseClient):
     ) -> globus_sdk.GlobusHTTPResponse:
         return self.put(
             f"/api/v2/topic/{topic}",
-            headers={"Subject": str(subject), "Action": action}
+            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}
+            f"/api/v2/topic/{topic}", headers={"Subject": str(subject), "Topic": topic}
         )
 
     def update_topic_configs(
@@ -52,9 +51,12 @@ class WebClient(globus_sdk.BaseClient):
     ) -> 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")
+            headers={
+                "Subject": str(subject),
+                "Topic": topic,
+                "Content-Type": "text/plain",
+            },
+            data=json.dumps(configs).encode("utf-8"),
         )
 
     def update_topic_partitions(
@@ -62,8 +64,11 @@ class WebClient(globus_sdk.BaseClient):
     ) -> globus_sdk.GlobusHTTPResponse:
         return self.post(
             f"/api/v2/topic/{topic}/partitions",
-            headers={"Subject": str(subject), "Topic": topic,
-                     "NewPartitions": str(new_partitions)}
+            headers={
+                "Subject": str(subject),
+                "Topic": topic,
+                "NewPartitions": str(new_partitions),
+            },
         )
 
     def reset_topic(
@@ -71,8 +76,7 @@ class WebClient(globus_sdk.BaseClient):
     ) -> globus_sdk.GlobusHTTPResponse:
         return self.post(
             f"/api/v2/topic/{topic}/reset",
-            headers={"Subject": str(subject),
-                     "Topic": topic}
+            headers={"Subject": str(subject), "Topic": topic},
         )
 
     def grant_user_access(
@@ -80,8 +84,12 @@ class WebClient(globus_sdk.BaseClient):
     ) -> globus_sdk.GlobusHTTPResponse:
         return self.post(
             f"/api/v2/topic/{topic}/user",
-            headers={"Subject": str(subject), "Action": action,
-                     "Topic": topic, "User": str(user)}
+            headers={
+                "Subject": str(subject),
+                "Action": action,
+                "Topic": topic,
+                "User": str(user),
+            },
         )
 
     def list_topic_users(
@@ -89,24 +97,33 @@ class WebClient(globus_sdk.BaseClient):
     ) -> globus_sdk.GlobusHTTPResponse:
         return self.get(
             f"/api/v2/topic/{topic}/users",
-            headers={"Subject": str(subject),
-                     "Topic": topic}
+            headers={"Subject": str(subject), "Topic": topic},
         )
 
     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, trigger_configs: dict
+        self,
+        subject: UUID_LIKE_T,
+        topic: str,
+        function: str,
+        action: str,
+        function_configs: dict,
+        trigger_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": function_configs,
-                             "trigger": trigger_configs}).encode("utf-8")
+            headers={
+                "Subject": str(subject),
+                "Topic": topic,
+                "Trigger": function,
+                "Action": action,
+                "Content-Type": "text/plain",
+            },
+            data=json.dumps(
+                {"function": function_configs, "trigger": trigger_configs}
+            ).encode("utf-8"),
         )
 
     def update_trigger(
@@ -114,24 +131,25 @@ class WebClient(globus_sdk.BaseClient):
     ) -> 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")
+            headers={
+                "Subject": str(subject),
+                "Trigger_id": str(trigger_uuid),
+                "Content-Type": "text/plain",
+            },
+            data=json.dumps(trigger_configs).encode("utf-8"),
         )
 
     def list_log_streams(
         self, subject: UUID_LIKE_T, trigger: str
     ) -> globus_sdk.GlobusHTTPResponse:
         return self.get(
-            f"/api/v2/logs",
-            headers={"Subject": str(subject), "Trigger": trigger}
+            "/api/v2/logs", headers={"Subject": str(subject), "Trigger": trigger}
         )
 
     def get_log_events(
         self, subject: UUID_LIKE_T, trigger: str, stream: str
     ) -> globus_sdk.GlobusHTTPResponse:
         return self.get(
-            f"/api/v2/log",
-            headers={"Subject": str(subject), "Trigger": trigger,
-                     "Stream": stream}
+            "/api/v2/log",
+            headers={"Subject": str(subject), "Trigger": trigger, "Stream": stream},
         )

diaspora_event_sdk/version.py

@@ -1 +1 @@
-__version__ = "0.2.9"
+__version__ = "0.3.1"

{diaspora_event_sdk-0.2.9.dist-info → diaspora_event_sdk-0.3.1.dist-info}/LICENSE

@@ -199,4 +199,4 @@
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
-   limitations under the License.
+   limitations under the License.

diaspora_event_sdk-0.3.1.dist-info/METADATA

@@ -0,0 +1,46 @@
+Metadata-Version: 2.1
+Name: diaspora-event-sdk
+Version: 0.3.1
+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: Apache 2.0
+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'
+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'
+Requires-Dist: pre-commit ; extra == 'test'
+
+# Diaspora Event Fabric SDK
+
+## Installation Guide
+### Recommended Method: With `kafka-python`
+To integrate with Diaspora Event Fabric using `KafkaProducer` and `KafkaConsumer`, install the SDK with `kafka-python`:
+```bash
+pip install "diaspora-event-sdk[kafka-python]"
+```
+
+### Alternative Installation: Without Kafka Client Library
+For other Kafka client libraries (e.g., `confluent-kafka-python`, `aiokafka`), install the SDK without `kafka-python`:
+```bash
+pip install diaspora-event-sdk
+```
+Note: This does not include `KafkaProducer` and `KafkaConsumer` dependencies.
+
+## Using Diaspora Event Fabric SDK
+Check our [Notebook](DiasporaDemo.ipynb) for a quickstart and demonstration.
+
+<!-- **Getting Started**: Visit our [QuickStart Guide](docs/quickstart.md) for details on using the SDK with the kafka-python library and instructions for other Kafka clients.
+
+**Troubleshooting and Credential Management**: Consult our [TrobleShooting Guide](docs/troubleshooting.md) for solving common issues and tips on managing keys effectively. -->
+
+<!-- **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.3.1.dist-info/RECORD

@@ -0,0 +1,34 @@
+diaspora_event_sdk/__init__.py,sha256=x5A69JeA_iNi0i1XK3T2wuSTiOLxRhNdGYN4G39wtFw,455
+diaspora_event_sdk/version.py,sha256=r4xAFihOf72W9TD-lpMi6ntWSTKTP2SlzKP1ytkjRbI,22
+diaspora_event_sdk/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diaspora_event_sdk/sdk/_environments.py,sha256=lrdUaBCmlv6kESLi3O9EN6cMGXEylO8zWQX7NQ8OKCQ,684
+diaspora_event_sdk/sdk/aws_iam_msk.py,sha256=VMQpA-20fFWaAzzv9TyBpDK8dcMHS4tijjjsYhTJF74,3892
+diaspora_event_sdk/sdk/client.py,sha256=Mzsd8WDyaAGMkNDDjpzrWMLPwyC6JoeFVYb0YERrBrk,8774
+diaspora_event_sdk/sdk/decorators.py,sha256=Gel8AyhIjbf4-FNintTNcOqvC9hHH_YwbOH257Nfmf0,884
+diaspora_event_sdk/sdk/kafka_client.py,sha256=ZwNf83gahsLVRHert8YeiruioWAoEIOSTSKZOY-A2PY,4271
+diaspora_event_sdk/sdk/web_client.py,sha256=mc6OvYgq8Cl-irjOW5FfhU_1fl1nRF8OKlPPmaQc0zo,4992
+diaspora_event_sdk/sdk/botocore/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diaspora_event_sdk/sdk/botocore/auth.py,sha256=QXwCNMzM0wAqD64RlFd-K27atRpixti73VKConpb1kk,18699
+diaspora_event_sdk/sdk/botocore/awsrequest.py,sha256=fcR28YxbsCk2ZT1rGtgWxhje-sYt-SxIqixjOWAFLcM,9344
+diaspora_event_sdk/sdk/botocore/compat.py,sha256=5A-15L8yJhJLQaWvx3xvLfyf5-oO_Ejc-lK55aF0TjI,11197
+diaspora_event_sdk/sdk/botocore/credentials.py,sha256=upSYlgIcp8piSFtGay4ryiCMqNJD5fWFq802p0DWIfI,2455
+diaspora_event_sdk/sdk/botocore/exceptions.py,sha256=Wcu505IzZuyxOl4aE3iuXL-ByL6GDebWFrby7nACgqY,2002
+diaspora_event_sdk/sdk/botocore/utils.py,sha256=TCSvM3QmwVFTDWmU55c1FtAZS0zMWjTkyywmaL1qfCk,5370
+diaspora_event_sdk/sdk/login_manager/__init__.py,sha256=yeqVgjeHLMX0WZJu2feJmq-fbeXvSxWghVV81ygfY-w,239
+diaspora_event_sdk/sdk/login_manager/client_login.py,sha256=e8o6Tf40AW61auHNhQ_5HV7-LDEuNxTWcsKHVUfgxsg,1836
+diaspora_event_sdk/sdk/login_manager/decorators.py,sha256=JlCEwoceSMmFy6PLQ5kz5mpkKfitJgCVAA6I4I_k1nw,1013
+diaspora_event_sdk/sdk/login_manager/globus_auth.py,sha256=pbDy67Mdf4SKFf0IlXD3ZFGj4lG-xpzFTD1ygW-0H1k,430
+diaspora_event_sdk/sdk/login_manager/login_flow.py,sha256=NjmDtr9QWCG1SIPt9TL3pxcUgLfz2qzzWVHBBMJWvE0,978
+diaspora_event_sdk/sdk/login_manager/manager.py,sha256=AD3f8rx154oesgkpqG0K-eZsEGjntr08Vgrxr1m6uao,7040
+diaspora_event_sdk/sdk/login_manager/protocol.py,sha256=ipAOUi7GYF5YfU-az1LWUbU_rNXHarDwUkKn6TUyjVY,678
+diaspora_event_sdk/sdk/login_manager/tokenstore.py,sha256=ImncC8EIxoAuGtDiZIwdtUgOD2fWo8oBP22G-fiZ5L4,2036
+diaspora_event_sdk/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diaspora_event_sdk/sdk/utils/uuid_like.py,sha256=xbxf0YXpDhdii16lwPLWRN21qFekHrNrqODSToMPtCg,470
+tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/unit/apis_test.py,sha256=S_6JjHU-YAcz6nITveN5qzkINkWrrgDcIPR8Ik8JNf4,4365
+tests/unit/client_test.py,sha256=sJUtPmnNGnohnP38RQrwcJ4D5j3-g1WFQ6gaKf520AQ,3019
+diaspora_event_sdk-0.3.1.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+diaspora_event_sdk-0.3.1.dist-info/METADATA,sha256=WolDkvsp9WLY5fn9miVBx8I5cxdHDjRTjozPj3ZT4LE,2101
+diaspora_event_sdk-0.3.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+diaspora_event_sdk-0.3.1.dist-info/top_level.txt,sha256=OVun-67t3fkLFEIwvJuNINgFFvAc--bClYhXjLhMmvs,25
+diaspora_event_sdk-0.3.1.dist-info/RECORD,,

tests/unit/apis_test.py

@@ -0,0 +1,132 @@
+import pytest
+import os
+import logging
+from globus_sdk import ConfidentialAppAuthClient
+from diaspora_event_sdk import Client
+from diaspora_event_sdk.sdk.login_manager import tokenstore
+
+# Configure module-level logger
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+)
+logger = logging.getLogger(__name__)
+
+
+@pytest.fixture(scope="module")
+def setup():
+    client_id = os.environ["DIASPORA_SDK_CLIENT_ID"]
+    client_secret = os.environ["DIASPORA_SDK_CLIENT_SECRET"]
+    requested_scopes = os.environ["CLIENT_SCOPE"]
+
+    ca = ConfidentialAppAuthClient(
+        client_id=client_id,
+        client_secret=client_secret,
+    )
+    token_response = ca.oauth2_client_credentials_tokens(
+        requested_scopes=requested_scopes,
+    )
+    token = token_response.by_resource_server[client_id]
+
+    storage = tokenstore.get_token_storage_adapter()
+    storage.store(token_response)
+    storage.get_by_resource_server()
+
+    return {
+        "client_id": client_id,
+        "client_secret": client_secret,
+        "requested_scopes": requested_scopes,
+        "token_response": token_response,
+        "token": token,
+    }
+
+
+@pytest.fixture(scope="module")
+def client():
+    return Client()
+
+
+def test_create_key(setup, client):
+    key_response = client.create_key()
+    assert isinstance(key_response, dict)
+    assert "access_key" in key_response
+    assert "secret_key" in key_response
+    assert "endpoint" in key_response
+
+
+def test_register_topic(setup, client):
+    topic = "topic" + client.subject_openid[-12:]
+    register_response = client.register_topic(topic)
+    assert register_response["status"] in ["success", "no-op"]
+    assert "message" in register_response
+
+
+def test_list_topics(setup, client):
+    topics = client.list_topics()
+    assert topics["status"] == "success"
+    assert isinstance(topics["topics"], list)
+    assert len(topics["topics"]) > 0
+    expected_topics = ["diaspora-cicd"]
+    assert set(expected_topics).issubset(set(topics["topics"]))
+
+
+def test_get_topic_configs(setup, client):
+    topic = "topic" + client.subject_openid[-12:]
+    client.register_topic(topic)
+    configs_response = client.get_topic_configs(topic)
+    assert configs_response["status"] == "success"
+    assert "configs" in configs_response
+    assert isinstance(configs_response["configs"], dict)
+
+
+def test_update_topic_configs(setup, client):
+    topic = "topic" + client.subject_openid[-12:]
+    client.register_topic(topic)
+    configs = {"min.insync.replicas": 1}
+    update_response = client.update_topic_configs(topic, configs)
+    assert update_response["status"] == "success"
+    assert "before" in update_response
+    assert "after" in update_response
+    assert isinstance(update_response["before"], dict)
+    assert isinstance(update_response["after"], dict)
+
+
+def test_update_topic_partitions(setup, client):
+    topic = "topic" + client.subject_openid[-12:]
+    client.register_topic(topic)
+    new_partitions = 2
+    partitions_response = client.update_topic_partitions(topic, new_partitions)
+    assert partitions_response["status"] in ["success", "error"]
+    if partitions_response["status"] == "error":
+        assert "message" in partitions_response
+
+
+def test_reset_topic(setup, client):
+    topic = "topic" + client.subject_openid[-12:]
+    client.register_topic(topic)
+    reset_response = client.reset_topic(topic)
+    assert reset_response["status"] in ["success", "error"]
+    if reset_response["status"] == "error":
+        assert "message" in reset_response
+
+
+def test_user_access_management(setup, client):
+    topic = "topic" + client.subject_openid[-12:]
+    client.register_topic(topic)
+    user_id = "e2a8169b-feef-4d56-8eba-ab12747bee04"
+    grant_response = client.grant_user_access(topic, user_id)
+    assert grant_response["status"] in ["success", "no-op"]
+    assert "message" in grant_response
+
+    list_users_response = client.list_topic_users(topic)
+    assert list_users_response["status"] == "success"
+    assert "users" in list_users_response
+    assert isinstance(list_users_response["users"], list)
+
+    revoke_response = client.revoke_user_access(topic, user_id)
+    assert revoke_response["status"] in ["success", "no-op"]
+    assert "message" in revoke_response
+
+
+if __name__ == "__main__":
+    pytest.main(["-s", "tests/unit/test_apis.py"])

diaspora_event_sdk-0.2.9.dist-info/METADATA

@@ -1,51 +0,0 @@
-Metadata-Version: 2.1
-Name: diaspora-event-sdk
-Version: 0.2.9
-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: Apache 2.0
-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'
-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'
-
-# Diaspora: Hybrid Event-Driven Architecture for Distributed Scientific Computing
-
-## Event Fabric SDK Installation Guide
-### Recommended Method: Use with `kafka-python`
-For easy integration with Diaspora Event Fabric, use the `KafkaProducer` and `KafkaConsumer` classes from our SDK. This requires the `kafka-python` library.
-
-To install the Event Fabric SDK and `kafka-python,` with the following command:
-```bash
-pip install "diaspora-event-sdk[kafka-python]"
-```
-
-### Alternative Installation: Without Kafka Client Library
-To use alternative Kafka client libraries (e.g., `confluent-kafka-python`, `aiokafka`, and libraries for other programming laguages), you can install the SDK without the `kafka-python` dependency. This option still provides topic-level access control (authorization) and login credential management features.
-
-To install the SDK without `kafka-python`, use:
-```bash
-pip install diaspora-event-sdk
-```
-Note: This method does not include dependencies for `KafkaProducer` and `KafkaConsumer` classes mentioned in the QuickStart
-
-## Use Diaspora Event Fabric SDK
-
-**Getting Started**: Visit our [QuickStart Guide](docs/quickstart.md) for details on using the SDK with the kafka-python library and instructions for other Kafka clients.
-
-**Troubleshooting and Credential Management**: Consult our [TrobleShooting Guide](docs/troubleshooting.md) for solving common issues and tips on managing keys effectively.
-
-**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.2.9.dist-info/RECORD

@@ -1,33 +0,0 @@
-diaspora_event_sdk/__init__.py,sha256=v8IN3-WFpliakQKru8TAcmQ4IRdvRe_m9-abSDnGIFM,457
-diaspora_event_sdk/version.py,sha256=F8OVhAhMXSkvvXYgZtbPn2SG1AQC3joK4yu-FrHt81Y,22
-diaspora_event_sdk/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-diaspora_event_sdk/sdk/_environments.py,sha256=QyQA7dV4goUKUoxAoaes8OGv0xKxz2qiFBHNGr-vQNA,204
-diaspora_event_sdk/sdk/aws_iam_msk.py,sha256=rT6RipRVt-Zf7pTsMf9UhH2HDNC0FW28LhciVH_45bc,3942
-diaspora_event_sdk/sdk/client.py,sha256=fbjP_YRmGF2eItnMxNkdz5RIRvur0_P8DhE6AlYk4tM,8541
-diaspora_event_sdk/sdk/decorators.py,sha256=Gel8AyhIjbf4-FNintTNcOqvC9hHH_YwbOH257Nfmf0,884
-diaspora_event_sdk/sdk/kafka_client.py,sha256=xAxuPmMXIv0i_K_PFqc_BEG3wdGsFdYYvd7gUzeoLV8,4286
-diaspora_event_sdk/sdk/web_client.py,sha256=Rkp0ZUUXCeqbfVFUqX2oxvuLwqqW5_jsJJNN-v2L4FI,4770
-diaspora_event_sdk/sdk/botocore/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-diaspora_event_sdk/sdk/botocore/auth.py,sha256=QXwCNMzM0wAqD64RlFd-K27atRpixti73VKConpb1kk,18699
-diaspora_event_sdk/sdk/botocore/awsrequest.py,sha256=fcR28YxbsCk2ZT1rGtgWxhje-sYt-SxIqixjOWAFLcM,9344
-diaspora_event_sdk/sdk/botocore/compat.py,sha256=5A-15L8yJhJLQaWvx3xvLfyf5-oO_Ejc-lK55aF0TjI,11197
-diaspora_event_sdk/sdk/botocore/credentials.py,sha256=upSYlgIcp8piSFtGay4ryiCMqNJD5fWFq802p0DWIfI,2455
-diaspora_event_sdk/sdk/botocore/exceptions.py,sha256=Wcu505IzZuyxOl4aE3iuXL-ByL6GDebWFrby7nACgqY,2002
-diaspora_event_sdk/sdk/botocore/utils.py,sha256=TCSvM3QmwVFTDWmU55c1FtAZS0zMWjTkyywmaL1qfCk,5370
-diaspora_event_sdk/sdk/login_manager/__init__.py,sha256=yeqVgjeHLMX0WZJu2feJmq-fbeXvSxWghVV81ygfY-w,239
-diaspora_event_sdk/sdk/login_manager/client_login.py,sha256=8ild28_cgPSrLtg3Jhmzpg3EymAvH2UdAhKY52oOB_c,1835
-diaspora_event_sdk/sdk/login_manager/decorators.py,sha256=EFEp71d0oJ7vo2H8W7DJ2gPrDfGzeNXUNxri1C0l8h0,1047
-diaspora_event_sdk/sdk/login_manager/globus_auth.py,sha256=pbDy67Mdf4SKFf0IlXD3ZFGj4lG-xpzFTD1ygW-0H1k,430
-diaspora_event_sdk/sdk/login_manager/login_flow.py,sha256=6Vi_e3rACYdv4TRXPAvUv5xivCRkCo4DS841Ixu_r24,977
-diaspora_event_sdk/sdk/login_manager/manager.py,sha256=AD3f8rx154oesgkpqG0K-eZsEGjntr08Vgrxr1m6uao,7040
-diaspora_event_sdk/sdk/login_manager/protocol.py,sha256=RCuo2jy_XkpZvbxnKlDfTKs-L6b9_8_JR-Kq9wHwhoM,710
-diaspora_event_sdk/sdk/login_manager/tokenstore.py,sha256=Kq0IZGf9G9dE44yoyUZInod5xL_8caN9OugeTK6GBGg,2094
-diaspora_event_sdk/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-diaspora_event_sdk/sdk/utils/uuid_like.py,sha256=xbxf0YXpDhdii16lwPLWRN21qFekHrNrqODSToMPtCg,470
-tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/test_client.py,sha256=sJUtPmnNGnohnP38RQrwcJ4D5j3-g1WFQ6gaKf520AQ,3019
-diaspora_event_sdk-0.2.9.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-diaspora_event_sdk-0.2.9.dist-info/METADATA,sha256=L3RH4LPf5XaJnONbmOfFTK9awU5LT3agzb6KkaBHZQs,2449
-diaspora_event_sdk-0.2.9.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-diaspora_event_sdk-0.2.9.dist-info/top_level.txt,sha256=OVun-67t3fkLFEIwvJuNINgFFvAc--bClYhXjLhMmvs,25
-diaspora_event_sdk-0.2.9.dist-info/RECORD,,