Skip to content

Commit 8d464ff

Browse files
authored
feat(user-data): Scope credentials fetch by organization and workspace (#232)
1 parent 40ae38d commit 8d464ff

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

dreadnode/api/client.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -753,14 +753,17 @@ def export_timeseries(
753753

754754
# User data access
755755

756-
def get_user_data_credentials(self) -> UserDataCredentials:
756+
def get_user_data_credentials(
757+
self, organization_id: UUID, workspace_id: UUID
758+
) -> UserDataCredentials:
757759
"""
758760
Retrieves user data credentials for secondary storage access.
759761
760762
Returns:
761763
The user data credentials object.
762764
"""
763-
response = self._request("GET", "/user-data/credentials")
765+
params = {"org_id": str(organization_id), "workspace_id": str(workspace_id)}
766+
response = self._request("GET", "/user-data/credentials", params=params)
764767
return UserDataCredentials(**response.json())
765768

766769
# Container registry access

dreadnode/main.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -687,8 +687,11 @@ def initialize(self) -> None:
687687
if self._api is not None:
688688
api = self._api
689689
self._credential_manager = CredentialManager(
690-
credential_fetcher=lambda: api.get_user_data_credentials()
690+
credential_fetcher=lambda: api.get_user_data_credentials(
691+
self._organization.id, self._workspace.id
692+
)
691693
)
694+
692695
self._credential_manager.initialize()
693696

694697
self._fs = self._credential_manager.get_filesystem()

0 commit comments

Comments
 (0)