attach/detach data asset from cloud workstation session computation

Author: jake-valsamis

src/codeocean/computation.py

@@ -3,13 +3,16 @@
 from dataclasses import dataclass, field
 from dataclasses_json import dataclass_json
 from requests_toolbelt.sessions import BaseUrlSession
-from typing import Optional
+from typing import Optional, TYPE_CHECKING
 from time import sleep, time
 from warnings import warn
 
 from codeocean.enum import StrEnum
 from codeocean.folder import FileURLs, Folder, DownloadFileURL
 
+if TYPE_CHECKING:
+    from codeocean.data_asset import DataAssetAttachParams, DataAssetAttachResults
+
 
 class ComputationState(StrEnum):
     """Current state of a computation during its execution lifecycle."""
@@ -324,6 +327,27 @@ def wait_until_completed(
 
             sleep(polling_interval)
 
+    def attach_data_assets(
+        self,
+        computation_id: str,
+        attach_params: list["DataAssetAttachParams"],
+    ) -> list["DataAssetAttachResults"]:
+        """Attach one or more data assets to a cloud workstation session computation."""
+        from codeocean.data_asset import DataAssetAttachResults
+
+        res = self.client.post(
+            f"computations/{computation_id}/data_assets",
+            json=[j.to_dict() for j in attach_params],
+        )
+        return [DataAssetAttachResults.from_dict(c) for c in res.json()]   
+
+    def detach_data_assets(self, computation_id: str, data_assets: list[str]):
+        """Detach one or more data assets from a cloud workstation session computation by their IDs."""
+        self.client.delete(
+            f"computations/{computation_id}/data_assets/",
+            json=data_assets,
+        )
+
     def list_computation_results(self, computation_id: str, path: str = "") -> Folder:
         """List result files and folders generated by a computation
         at the specified path. Empty path retrieves the /results root folder."""

src/codeocean/data_asset.py

@@ -4,14 +4,16 @@
 from dataclasses import dataclass, field
 from requests_toolbelt.sessions import BaseUrlSession
 from time import sleep, time
-from typing import Optional, Iterator
+from typing import Optional, Iterator, TYPE_CHECKING
 from warnings import warn
 
 from codeocean.components import Ownership, SortOrder, SearchFilter, Permissions
-from codeocean.computation import PipelineProcess, Param
 from codeocean.enum import StrEnum
 from codeocean.folder import FileURLs, Folder, DownloadFileURL
 
+if TYPE_CHECKING:
+    from codeocean.computation import PipelineProcess, Param
+
 
 class DataAssetType(StrEnum):
     """Type of data asset indicating its content and purpose."""