atlan-application-sdk 0.1.1rc61__tar.gz → 0.1.1rc62__tar.gz

atlan_application_sdk-0.1.1rc62/.github/release_notes.md

@@ -0,0 +1,7 @@
+## v0.1.1rc62 (November 03, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc61...v0.1.1rc62
+
+### Features
+
+- Adding a new event whenever Token refreshes (#802) (by @Garavitey in [2e32ab4](https://github.com/atlanhq/application-sdk/commit/2e32ab4))

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## v0.1.1rc62 (November 03, 2025)
+
+Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc61...v0.1.1rc62
+
+### Features
+
+- Adding a new event whenever Token refreshes (#802) (by @Garavitey in [2e32ab4](https://github.com/atlanhq/application-sdk/commit/2e32ab4))
+
+
 ## v0.1.1rc61 (October 30, 2025)
 
 Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc60...v0.1.1rc61

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 0.1.1rc61
+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

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/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

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/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

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/application_sdk/version.py

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

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "atlan-application-sdk"
-version = "0.1.1rc61"
+version = "0.1.1rc62"
 description = "Atlan Application SDK is a Python library for developing applications on the Atlan Platform"
 license = "Apache-2.0"
 authors = [

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/tests/unit/clients/test_temporal_client.py

@@ -337,3 +337,110 @@ async def test_get_workflow_run_status_client_not_loaded(
     """Test get_workflow_run_status when client is not loaded."""
     with pytest.raises(ValueError, match="Client is not loaded"):
         await temporal_client.get_workflow_run_status("test_workflow_id", "test_run_id")
+
+
+@patch("application_sdk.clients.temporal.logger")
+@patch("application_sdk.clients.temporal.EventStore.publish_event")
+@patch("application_sdk.clients.temporal.time.time")
+@patch("application_sdk.clients.temporal.DEPLOYMENT_NAME", "test_deployment")
+async def test_publish_token_refresh_event_success(
+    mock_time: Mock,
+    mock_publish_event: AsyncMock,
+    mock_logger: MagicMock,
+    temporal_client: TemporalWorkflowClient,
+):
+    """Test successful token refresh event publishing."""
+    # Arrange
+    mock_time.return_value = 1234567890.0
+    mock_auth_manager = MagicMock()
+    mock_auth_manager.get_token_expiry_time.return_value = 1234567895.0
+    mock_auth_manager.get_time_until_expiry.return_value = 300.0
+    temporal_client.auth_manager = mock_auth_manager
+
+    # Act
+    await temporal_client._publish_token_refresh_event()
+
+    # Assert
+    mock_time.assert_called_once()
+    mock_auth_manager.get_token_expiry_time.assert_called_once()
+    mock_auth_manager.get_time_until_expiry.assert_called_once()
+    mock_publish_event.assert_called_once()
+
+    # Verify the event structure
+    call_args = mock_publish_event.call_args
+    event = call_args[0][0]  # First positional argument
+
+    assert event.event_type == "application_events"
+    assert event.event_name == "token_refresh"
+    assert event.data["application_name"] == "test_app"
+    assert event.data["deployment_name"] == "test_deployment"
+    assert event.data["force_refresh"] is True
+    assert event.data["token_expiry_time"] == 1234567895.0
+    assert event.data["time_until_expiry"] == 300.0
+    assert event.data["refresh_timestamp"] == 1234567890.0
+
+    mock_logger.info.assert_called_once_with("Published token refresh event")
+    mock_logger.warning.assert_not_called()
+
+
+@patch("application_sdk.clients.temporal.logger")
+@patch("application_sdk.clients.temporal.EventStore.publish_event")
+@patch("application_sdk.clients.temporal.time.time")
+@patch("application_sdk.clients.temporal.DEPLOYMENT_NAME", "test_deployment")
+async def test_publish_token_refresh_event_with_none_expiry_times(
+    mock_time: Mock,
+    mock_publish_event: AsyncMock,
+    mock_logger: MagicMock,
+    temporal_client: TemporalWorkflowClient,
+):
+    """Test token refresh event publishing with None expiry times."""
+    # Arrange
+    mock_time.return_value = 1234567890.0
+    mock_auth_manager = MagicMock()
+    mock_auth_manager.get_token_expiry_time.return_value = None
+    mock_auth_manager.get_time_until_expiry.return_value = None
+    temporal_client.auth_manager = mock_auth_manager
+
+    # Act
+    await temporal_client._publish_token_refresh_event()
+
+    # Assert
+    call_args = mock_publish_event.call_args
+    event = call_args[0][0]  # First positional argument
+
+    assert event.data["token_expiry_time"] == 0
+    assert event.data["time_until_expiry"] == 0
+
+    mock_logger.info.assert_called_once_with("Published token refresh event")
+    mock_logger.warning.assert_not_called()
+
+
+@patch("application_sdk.clients.temporal.logger")
+@patch("application_sdk.clients.temporal.EventStore.publish_event")
+@patch("application_sdk.clients.temporal.time.time")
+async def test_publish_token_refresh_event_exception_handling(
+    mock_time: Mock,
+    mock_publish_event: AsyncMock,
+    mock_logger: MagicMock,
+    temporal_client: TemporalWorkflowClient,
+):
+    """Test token refresh event publishing handles exceptions gracefully."""
+    # Arrange
+    mock_time.return_value = 1234567890.0
+    mock_auth_manager = MagicMock()
+    mock_auth_manager.get_token_expiry_time.return_value = 1234567895.0
+    mock_auth_manager.get_time_until_expiry.return_value = 300.0
+    temporal_client.auth_manager = mock_auth_manager
+
+    # Simulate event publishing failure
+    mock_publish_event.side_effect = Exception("Event store connection failed")
+
+    # Act - should not raise exception
+    await temporal_client._publish_token_refresh_event()
+
+    # Assert
+    mock_publish_event.assert_called_once()
+    mock_logger.info.assert_not_called()
+    mock_logger.warning.assert_called_once_with(
+        "Failed to publish token refresh event: Event store connection failed"
+    )

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/tests/unit/services/test_eventstore.py

@@ -209,6 +209,7 @@ class TestApplicationEventNames:
             "worker_end",
             "application_start",
             "application_end",
+            "token_refresh",
         ]
         actual_names = [name.value for name in ApplicationEventNames]
         assert actual_names == expected_names

{atlan_application_sdk-0.1.1rc61 → atlan_application_sdk-0.1.1rc62}/uv.lock

@@ -185,7 +185,7 @@ wheels = [
 
 [[package]]
 name = "atlan-application-sdk"
-version = "0.1.1rc61"
+version = "0.1.1rc62"
 source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },

atlan_application_sdk-0.1.1rc61/.github/release_notes.md

@@ -1,3 +0,0 @@
-## v0.1.1rc61 (October 30, 2025)
-
-Full Changelog: https://github.com/atlanhq/application-sdk/compare/v0.1.1rc60...v0.1.1rc61