Merge pull request #595 from atlanhq/APP-5915

APP-5915 : Add "Workflow Name" as an optional param in `credentials.get_all()`

Author: Aryamanz29

pyatlan/client/credential.py

@@ -83,13 +83,15 @@ def get_all(
         filter: Optional[Dict[str, Any]] = None,
         limit: Optional[int] = None,
         offset: Optional[int] = None,
+        workflow_name: Optional[str] = None,
     ) -> CredentialListResponse:
         """
         Retrieves all credentials.
 
         :param filter: (optional) dictionary specifying the filter criteria.
         :param limit: (optional) maximum number of credentials to retrieve.
         :param offset: (optional) number of credentials to skip before starting retrieval.
+        :param workflow_name: (optional) name of the workflow to retrieve credentials for.
         :returns: CredentialListResponse instance.
         :raises: AtlanError on any error during API invocation.
         """
@@ -101,6 +103,17 @@ def get_all(
         if offset is not None:
             params["offset"] = offset
 
+        if workflow_name is not None:
+            if filter is None:
+                filter = {}
+
+            if workflow_name.startswith("atlan-"):
+                workflow_name = "default-" + workflow_name[len("atlan-") :]
+
+            filter["name"] = f"{workflow_name}-0"
+
+            params["filter"] = dumps(filter)
+
         raw_json = self._client._call_api(
             GET_ALL_CREDENTIALS.format_path_with_params(), query_params=params
         )