atlan-application-sdk 0.1.1rc60__py3-none-any.whl → 0.1.1rc62__py3-none-any.whl

application_sdk/clients/temporal.py

@@ -1,4 +1,5 @@
 import asyncio
+import time
 import uuid
 from concurrent.futures import ThreadPoolExecutor
 from typing import Any, Dict, Optional, Sequence, Type
@@ -26,10 +27,17 @@ from application_sdk.constants import (
     WORKFLOW_PORT,
     WORKFLOW_TLS_ENABLED_KEY,
 )
+from application_sdk.events.models import (
+    ApplicationEventNames,
+    Event,
+    EventTypes,
+    WorkerTokenRefreshEventData,
+)
 from application_sdk.interceptors.cleanup import CleanupInterceptor, cleanup
 from application_sdk.interceptors.events import EventInterceptor, publish_event
 from application_sdk.interceptors.lock import RedisLockInterceptor
 from application_sdk.observability.logger_adaptor import get_logger
+from application_sdk.services.eventstore import EventStore
 from application_sdk.services.secretstore import SecretStore
 from application_sdk.services.statestore import StateStore, StateType
 from application_sdk.workflows import WorkflowInterface
@@ -142,6 +150,39 @@ class TemporalWorkflowClient(WorkflowClient):
         """
         return self.namespace
 
+    async def _publish_token_refresh_event(self) -> None:
+        """Publish a token refresh event to the event store.
+
+        This method creates and publishes an event containing token refresh information,
+        including application name, deployment name, token expiry times, and refresh timestamp.
+        If event publishing fails, it logs a warning but does not raise an exception to avoid
+        disrupting the token refresh loop.
+
+        Note:
+            This method handles exceptions internally and will not propagate errors,
+            ensuring the token refresh loop continues even if event publishing fails.
+        """
+        try:
+            current_time = time.time()
+            worker_token_refresh_data = WorkerTokenRefreshEventData(
+                application_name=self.application_name,
+                deployment_name=DEPLOYMENT_NAME,
+                force_refresh=True,
+                token_expiry_time=self.auth_manager.get_token_expiry_time() or 0,
+                time_until_expiry=self.auth_manager.get_time_until_expiry() or 0,
+                refresh_timestamp=current_time,
+            )
+
+            event = Event(
+                event_type=EventTypes.APPLICATION_EVENT.value,
+                event_name=ApplicationEventNames.TOKEN_REFRESH.value,
+                data=worker_token_refresh_data.model_dump(),
+            )
+            await EventStore.publish_event(event)
+            logger.info("Published token refresh event")
+        except Exception as e:
+            logger.warning(f"Failed to publish token refresh event: {e}")
+
     async def _token_refresh_loop(self) -> None:
         """Background loop that refreshes the authentication token dynamically."""
         while True:
@@ -161,6 +202,9 @@ class TemporalWorkflowClient(WorkflowClient):
                 self._token_refresh_interval = (
                     self.auth_manager.calculate_refresh_interval()
                 )
+                # Publish token refresh event
+                await self._publish_token_refresh_event()
+
             except asyncio.CancelledError:
                 logger.info("Token refresh loop cancelled")
                 break

application_sdk/constants.py

@@ -51,7 +51,9 @@ SQL_SERVER_MIN_VERSION = os.getenv("ATLAN_SQL_SERVER_MIN_VERSION")
 SQL_QUERIES_PATH = os.getenv("ATLAN_SQL_QUERIES_PATH", "app/sql")
 
 # Output Path Constants
-#: Output path format for workflows (example: objectstore://bucket/artifacts/apps/{application_name}/workflows/{workflow_id}/{workflow_run_id})
+#: Output path format for workflows.
+#:
+#: Example: objectstore://bucket/artifacts/apps/{application_name}/workflows/{workflow_id}/{workflow_run_id}
 WORKFLOW_OUTPUT_PATH_TEMPLATE = (
     "artifacts/apps/{application_name}/workflows/{workflow_id}/{run_id}"
 )
@@ -68,7 +70,9 @@ CLEANUP_BASE_PATHS = [
 ]
 
 # State Store Constants
-#: Path template for state store files (example: objectstore://bucket/persistent-artifacts/apps/{application_name}/{state_type}/{id}/config.json)
+#: Path template for state store files.
+#:
+#: Example: objectstore://bucket/persistent-artifacts/apps/{application_name}/{state_type}/{id}/config.json
 STATE_STORE_PATH_TEMPLATE = (
     "persistent-artifacts/apps/{application_name}/{state_type}/{id}/config.json"
 )
@@ -147,7 +151,9 @@ WORKER_START_EVENT_VERSION = "v1"
 #: Whether to enable Atlan storage upload
 ENABLE_ATLAN_UPLOAD = os.getenv("ENABLE_ATLAN_UPLOAD", "false").lower() == "true"
 # Dapr Client Configuration
-#: Maximum gRPC message length in bytes for Dapr client (default: 100MB)
+#: Maximum gRPC message length in bytes for Dapr client.
+#:
+#: Default: 100MB
 DAPR_MAX_GRPC_MESSAGE_LENGTH = int(
     os.getenv("DAPR_MAX_GRPC_MESSAGE_LENGTH", "104857600")
 )
@@ -250,9 +256,9 @@ REDIS_HOST = os.getenv("REDIS_HOST", "")
 REDIS_PORT = os.getenv("REDIS_PORT", "")
 #: Redis password (required for authenticated Redis instances)
 REDIS_PASSWORD = os.getenv("REDIS_PASSWORD")
-#: Redis Sentinel service name (default: mymaster)
+#: Redis Sentinel service name. Default: mymaster
 REDIS_SENTINEL_SERVICE_NAME = os.getenv("REDIS_SENTINEL_SERVICE_NAME", "mymaster")
-#: Redis Sentinel hosts (comma-separated host:port pairs)
+#: Redis Sentinel hosts as comma-separated host:port pairs
 REDIS_SENTINEL_HOSTS = os.getenv("REDIS_SENTINEL_HOSTS", "")
 #: Whether to enable strict locking
 IS_LOCKING_DISABLED = os.getenv("IS_LOCKING_DISABLED", "true").lower() == "true"

application_sdk/events/models.py

@@ -29,6 +29,7 @@ class ApplicationEventNames(Enum):
     WORKER_END = "worker_end"
     APPLICATION_START = "application_start"
     APPLICATION_END = "application_end"
+    TOKEN_REFRESH = "token_refresh"
 
 
 class WorkflowStates(Enum):
@@ -174,3 +175,26 @@ class WorkerStartEventData(BaseModel):
     max_concurrent_activities: Optional[int]
     workflow_count: int
     activity_count: int
+
+
+class WorkerTokenRefreshEventData(BaseModel):
+    """Model for token refresh event data.
+
+    This model represents the data structure used when publishing token refresh events.
+    It contains information about the token refresh operation and agent status.
+
+    Attributes:
+        application_name: Name of the application the token belongs to.
+        deployment_name: Name of the deployment (e.g., dev, staging, prod).
+        force_refresh: Whether this was a forced refresh or automatic.
+        token_expiry_time: Unix timestamp when the new token expires.
+        time_until_expiry: Seconds until token expires.
+        refresh_timestamp: Unix timestamp when the refresh occurred.
+    """
+
+    application_name: str
+    deployment_name: str
+    force_refresh: bool
+    token_expiry_time: float
+    time_until_expiry: float
+    refresh_timestamp: float

application_sdk/version.py

@@ -2,4 +2,4 @@
 Version information for the application_sdk package.
 """
 
-__version__ = "0.1.1rc60"
+__version__ = "0.1.1rc62"

{atlan_application_sdk-0.1.1rc60.dist-info → atlan_application_sdk-0.1.1rc62.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc60
+Version: 0.1.1rc62
 Summary: Atlan Application SDK is a Python library for developing applications on the Atlan Platform
 Project-URL: Repository, https://github.com/atlanhq/application-sdk
 Project-URL: Documentation, https://github.com/atlanhq/application-sdk/README.md
@@ -22,7 +22,7 @@ Requires-Python: >=3.11
 Requires-Dist: aiohttp>=3.10.0
 Requires-Dist: duckdb-engine>=0.17.0
 Requires-Dist: duckdb>=1.1.3
-Requires-Dist: fastapi[standard]>=0.115.0
+Requires-Dist: fastapi[standard]==0.120.2
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: opentelemetry-exporter-otlp>=1.27.0
 Requires-Dist: psutil>=7.0.0

{atlan_application_sdk-0.1.1rc60.dist-info → atlan_application_sdk-0.1.1rc62.dist-info}/RECORD

@@ -1,6 +1,6 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
-application_sdk/constants.py,sha256=ySrjME6CSoiyjVLgQt0s4dIdgJ0JWIMmMs7WBZBVUpA,11013
-application_sdk/version.py,sha256=2aOWC-Y08sXYBsPMJehGpskX5CTH1TNdtzY-QR-Ub8o,88
+application_sdk/constants.py,sha256=XCRJRiug2GXgeeVZu3Mp1QavRw54bWtzxIZ2l1bIkkA,11028
+application_sdk/version.py,sha256=MFOV2tNUAo7go0hrgOky6QW0SxSDPfPaY9aTrIZC3wg,88
 application_sdk/worker.py,sha256=i5f0AeKI39IfsLO05QkwC6uMz0zDPSJqP7B2byri1VI,7489
 application_sdk/activities/__init__.py,sha256=L5WXkTwOwGtjWAlXrUJRCKGwyIyp3z8fBv8BZVCRFQI,11175
 application_sdk/activities/lock_management.py,sha256=6Wdf3jMKitoarHQP91PIJOoGFz4aaOLS_40c7n1yAOA,3902
@@ -23,7 +23,7 @@ application_sdk/clients/base.py,sha256=TIn3pG89eXUc1XSYf4jk66m1vajWp0WxcCQOOltda
 application_sdk/clients/models.py,sha256=iZOTyH6LO64kozdiUPCFCN0NgLhd_Gtv0lH7ZIPdo8w,1800
 application_sdk/clients/redis.py,sha256=IfAD32vLp88BCvsDTaQtxFHxzHlEx4V7TK7h1HwDDBg,15917
 application_sdk/clients/sql.py,sha256=lXeVu_dute30IaWWK5gHBhjEs2dXp_e0XkOMsbOsq64,19589
-application_sdk/clients/temporal.py,sha256=jC3U8LmW8G6gg-Qmxk0rcAifIGF0KekwP1UkMGXN7RA,18314
+application_sdk/clients/temporal.py,sha256=eSbBOmbRmYU-MLiQ1D8hHTU3-VIJssTYAzC2twLwf38,20200
 application_sdk/clients/utils.py,sha256=zLFOJbTr_6TOqnjfVFGY85OtIXZ4FQy_rquzjaydkbY,779
 application_sdk/clients/workflow.py,sha256=6bSqmA3sNCk9oY68dOjBUDZ9DhNKQxPD75qqE0cfldc,6104
 application_sdk/clients/.cursor/BUGBOT.md,sha256=7nEDUqWBEMI_uU6eK1jCSZGeXoQtLQcKwOrDn8AIDWo,10595
@@ -55,7 +55,7 @@ application_sdk/docgen/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 application_sdk/docgen/parsers/directory.py,sha256=8Kk2sjb-0l2wLO_njdlcuHjv5akoNgmf-FmaDSaE4WM,7751
 application_sdk/docgen/parsers/manifest.py,sha256=3NP-dBTpHAUQa477usMIDaKSb_9xfLE8G3RX0T1Bq2s,3318
 application_sdk/events/__init__.py,sha256=OcbVWDF4ZKRTJXK9UaFVtYEwu-3DHE77S-Sn6jNafUs,204
-application_sdk/events/models.py,sha256=7Esqp3WlbriT2EqT4kNiY_sHtRXRPLj27b8SbeC5Sb0,5121
+application_sdk/events/models.py,sha256=Q8qt1lW0XimtNI2GFd9o6kO3xOuPt9c8yFqzZJVKCv8,6019
 application_sdk/handlers/__init__.py,sha256=3Wf7jCVFR2nYOyHZEc9jj8BQUnHCylFqoezp70J2Df0,1329
 application_sdk/handlers/base.py,sha256=ieWFbv8Gm7vfrrpS-mdMSm-mHGuQY02qiAVX2qPdj3w,2467
 application_sdk/handlers/sql.py,sha256=6A_9xCtkXyNY5gPhImbftzrdPIEWIeTTqjyIewVESHA,17815
@@ -157,8 +157,8 @@ application_sdk/workflows/metadata_extraction/__init__.py,sha256=jHUe_ZBQ66jx8bg
 application_sdk/workflows/metadata_extraction/sql.py,sha256=6ZaVt84n-8U2ZvR9GR7uIJKv5v8CuyQjhlnoRJvDszc,12435
 application_sdk/workflows/query_extraction/__init__.py,sha256=n066_CX5RpJz6DIxGMkKS3eGSRg03ilaCtsqfJWQb7Q,117
 application_sdk/workflows/query_extraction/sql.py,sha256=kT_JQkLCRZ44ZpaC4QvPL6DxnRIIVh8gYHLqRbMI-hA,4826
-atlan_application_sdk-0.1.1rc60.dist-info/METADATA,sha256=vbNpysrGHIBYdJGSe-s8V3km-Amr5xeXIbGunIDHnnI,5730
-atlan_application_sdk-0.1.1rc60.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-atlan_application_sdk-0.1.1rc60.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-0.1.1rc60.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-0.1.1rc60.dist-info/RECORD,,
+atlan_application_sdk-0.1.1rc62.dist-info/METADATA,sha256=8ApEAfPm6ZpGgjwZtDJ2F_rqLkNCM5L-3ZKTKkdkwJg,5730
+atlan_application_sdk-0.1.1rc62.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+atlan_application_sdk-0.1.1rc62.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-0.1.1rc62.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-0.1.1rc62.dist-info/RECORD,,