Add API functions for team roles and user interactions.

webknossos/Changelog.md

@@ -15,6 +15,7 @@ For upgrade instructions, please check the respective _Breaking Changes_ section
 ### Breaking Changes
 
 ### Added
+- Webknossos API functions were added: `Team.get_list()`, `Team.add("new_name")`, `User.assign_team_roles("teamName", isTeamManager: True)` and `RemoteDataset.explore_and_add_remote()` are available now. [#1196](https://github.com/scalableminds/webknossos-libs/pull/1196)
 
 ### Changed
 

webknossos/webknossos/administration/user.py

@@ -2,7 +2,12 @@
 
 import attr
 
-from ..client.api_client.models import ApiLoggedTimeGroupedByMonth, ApiUser
+from ..client.api_client.models import (
+    ApiLoggedTimeGroupedByMonth,
+    ApiTeamAdd,
+    ApiTeamMembership,
+    ApiUser,
+)
 from ..client.context import _get_api_client
 
 
@@ -82,6 +87,25 @@ def get_all_managed_users(cls) -> List["User"]:
         api_users = client.user_list()
         return [cls._from_api_user(i) for i in api_users]
 
+    def assign_team_roles(self, team_name: str, is_team_manager: bool) -> None:
+        """Assigns the specified roles to the user for the specified team."""
+        client = _get_api_client(enforce_auth=True)
+        api_user = client.user_by_id(self.user_id)
+        if team_name in [team.name for team in api_user.teams]:
+            api_user.teams = [
+                team
+                if team.name != team_name
+                else ApiTeamMembership(team.id, team.name, is_team_manager)
+                for team in api_user.teams
+            ]
+        else:
+            api_user.teams.append(
+                ApiTeamMembership(
+                    Team.get_by_name(team_name).id, team_name, is_team_manager
+                )
+            )
+        client.user_update(api_user)
+
 
 @attr.frozen
 class Team:
@@ -99,6 +123,22 @@ def get_by_name(cls, name: str) -> "Team":
                 return cls(api_team.id, api_team.name, api_team.organization)
         raise KeyError(f"Could not find team {name}.")
 
+    @classmethod
+    def get_list(cls) -> List["Team"]:
+        """Returns all teams of the current user."""
+        client = _get_api_client(enforce_auth=True)
+        api_teams = client.team_list()
+        return [
+            cls(api_team.id, api_team.name, api_team.organization)
+            for api_team in api_teams
+        ]
+
+    @classmethod
+    def add(cls, team_name: str) -> None:
+        """Adds a new team with the specified name."""
+        client = _get_api_client(enforce_auth=True)
+        client.team_add(ApiTeamAdd(team_name))
+
 
 @attr.frozen
 class LoggedTime:

webknossos/webknossos/client/api_client/models.py

@@ -41,6 +41,11 @@ class ApiTeam:
     organization: str
 
 
+@attr.s(auto_attribs=True)
+class ApiTeamAdd:
+    name: str
+
+
 @attr.s(auto_attribs=True)
 class ApiBoundingBox:
     top_left: Tuple[int, int, int]
@@ -102,6 +107,14 @@ class ApiDataset:
     description: Optional[str] = None
 
 
+@attr.s(auto_attribs=True)
+class ApiDatasetExploreAndAddRemote:
+    remote_uri: str
+    dataset_name: str
+    folder_path: Optional[str] = None
+    data_store_name: Optional[str] = None
+
+
 @attr.s(auto_attribs=True)
 class ApiDatasetAnnounceUpload:
     dataset_name: str
@@ -215,6 +228,7 @@ class ApiTaskCreationResult:
 class ApiTeamMembership:
     id: str
     name: str
+    is_team_manager: bool
 
 
 @attr.s(auto_attribs=True)

webknossos/webknossos/client/api_client/wk_api_client.py

@@ -7,6 +7,7 @@
     ApiAnnotation,
     ApiAnnotationUploadResult,
     ApiDataset,
+    ApiDatasetExploreAndAddRemote,
     ApiDatasetIsValidNewNameResponse,
     ApiDataStore,
     ApiDataStoreToken,
@@ -20,6 +21,7 @@
     ApiTaskCreationResult,
     ApiTaskParameters,
     ApiTeam,
+    ApiTeamAdd,
     ApiUser,
     ApiWkBuildInfo,
 )
@@ -104,6 +106,15 @@ def dataset_is_valid_new_name(
         route = f"/datasets/{organization_name}/{dataset_name}/isValidNewName"
         return self._get_json(route, ApiDatasetIsValidNewNameResponse)
 
+    def dataset_explore_and_add_remote(
+        self, dataset: ApiDatasetExploreAndAddRemote
+    ) -> None:
+        route = "/datasets/exploreAndAddRemote"
+        self._post_json(
+            route,
+            dataset,
+        )
+
     def datastore_list(self) -> List[ApiDataStore]:
         route = "/datastores"
         return self._get_json(route, List[ApiDataStore])
@@ -176,10 +187,18 @@ def user_logged_time(self, user_id: str) -> ApiLoggedTimeGroupedByMonth:
         route = f"/users/{user_id}/loggedTime"
         return self._get_json(route, ApiLoggedTimeGroupedByMonth)
 
+    def user_update(self, user: ApiUser) -> None:
+        route = "/users/{user.id}"
+        self._patch_json(route, user)
+
     def team_list(self) -> List[ApiTeam]:
         route = "/teams"
         return self._get_json(route, List[ApiTeam])
 
+    def team_add(self, team: ApiTeamAdd) -> None:
+        route = "/teams"
+        self._post_json(route, team)
+
     def token_generate_for_data_store(self) -> ApiDataStoreToken:
         route = "/userToken/generate"
         return self._post_with_json_response(route, ApiDataStoreToken)

webknossos/webknossos/dataset/dataset.py

@@ -38,7 +38,11 @@
 from webknossos.dataset._metadata import DatasetMetadata
 from webknossos.geometry.vec_int import VecIntLike
 
-from ..client.api_client.models import ApiDataset, ApiMetadata
+from ..client.api_client.models import (
+    ApiDataset,
+    ApiDatasetExploreAndAddRemote,
+    ApiMetadata,
+)
 from ..geometry.vec3_int import Vec3Int, Vec3IntLike
 from ._array import ArrayException, ArrayInfo, BaseArray
 from ._utils import pims_images
@@ -2202,6 +2206,25 @@ def allowed_teams(self, allowed_teams: Sequence[Union[str, "Team"]]) -> None:
                 self._organization_id, self._dataset_name, team_ids
             )
 
+    @classmethod
+    def explore_and_add_remote(
+        cls, dataset_uri: PathLike, dataset_name: str, folder_path: str
+    ) -> "RemoteDataset":
+        from ..client.context import _get_api_client
+
+        (context, dataset_name, organisation_id, sharing_token) = cls._parse_remote(
+            dataset_name
+        )
+
+        with context:
+            client = _get_api_client()
+            dataset = ApiDatasetExploreAndAddRemote(
+                UPath(dataset_uri).resolve().as_uri(), dataset_name, folder_path
+            )
+            client.dataset_explore_and_add_remote(dataset)
+
+            return cls.open_remote(dataset_name, organisation_id, sharing_token)
+
     @property
     def folder(self) -> RemoteFolder:
         return RemoteFolder.get_by_id(self._get_dataset_info().folder_id)