Limited Workspace (#438)

Author: JR-Morgan

src/specklepy/api/resources/current/active_user_resource.py

@@ -12,6 +12,7 @@
     User,
 )
 from specklepy.core.api.models.current import (
+    LimitedWorkspace,
     PermissionCheckResult,
     ProjectWithPermissions,
     Workspace,
@@ -94,7 +95,7 @@ def get_workspaces(
         metrics.track(metrics.SDK, self.account, {"name": "Active User Get Workspaces"})
         return super().get_workspaces(limit, cursor, filter)
 
-    def get_active_workspace(self) -> Optional[Workspace]:
+    def get_active_workspace(self) -> Optional[LimitedWorkspace]:
         metrics.track(
             metrics.SDK, self.account, {"name": "Active User Get Active Workspace"}
         )

src/specklepy/core/api/models/current.py

@@ -213,15 +213,18 @@ class WorkspaceCreationState(GraphQLBaseModel):
     completed: bool
 
 
-class Workspace(GraphQLBaseModel):
+class LimitedWorkspace(GraphQLBaseModel):
     id: str
     name: str
     role: Optional[str]
     slug: str
     logo: Optional[str]
+    description: Optional[str]
+
+
+class Workspace(LimitedWorkspace):
     created_at: datetime
     updated_at: datetime
     read_only: bool
-    description: Optional[str]
     creation_state: Optional[WorkspaceCreationState]
     permissions: WorkspacePermissionChecks

src/specklepy/core/api/resources/current/active_user_resource.py

@@ -14,6 +14,7 @@
     User,
 )
 from specklepy.core.api.models.current import (
+    LimitedWorkspace,
     PermissionCheckResult,
     ProjectWithPermissions,
     Workspace,
@@ -295,7 +296,7 @@ def get_workspaces(
 
         return response.data.data
 
-    def get_active_workspace(self) -> Optional[Workspace]:
+    def get_active_workspace(self) -> Optional[LimitedWorkspace]:
         """
         This feature is only available on Workspace enabled servers  (server versions
         >=2.23.17) e.g. app.speckle.systems
@@ -310,29 +311,15 @@ def get_active_workspace(self) -> Optional[Workspace]:
                   role
                   slug
                   logo
-                  createdAt
-                  updatedAt
-                  readOnly
                   description
-                  creationState
-                  {
-                    completed
-                  }
-                  permissions {
-                    canCreateProject {
-                      authorized
-                      code
-                      message
-                    }
-                  }
                 }
               }
             }
-            """  # noqa: E501
+            """
         )
 
         response = self.make_request_and_parse_response(
-            DataResponse[Optional[DataResponse[Optional[Workspace]]]],
+            DataResponse[Optional[DataResponse[Optional[LimitedWorkspace]]]],
             QUERY,
         )