diff --git a/src/codeocean/capsule.py b/src/codeocean/capsule.py index 784ac34..980e8c4 100644 --- a/src/codeocean/capsule.py +++ b/src/codeocean/capsule.py @@ -5,7 +5,7 @@ from typing import Optional, Iterator from requests_toolbelt.sessions import BaseUrlSession -from codeocean.components import Ownership, SortOrder, SearchFilter +from codeocean.components import Ownership, SortOrder, SearchFilter, Permissions from codeocean.computation import Computation from codeocean.data_asset import DataAssetAttachParams, DataAssetAttachResults from codeocean.enum import StrEnum @@ -92,6 +92,12 @@ def list_computations(self, capsule_id: str) -> list[Computation]: return [Computation.from_dict(c) for c in res.json()] + def update_permissions(self, capsule_id: str, permissions: Permissions): + self.client.post( + f"capsules/{capsule_id}/permissions", + json=permissions.to_dict(), + ) + def attach_data_assets( self, capsule_id: str,