Migrate workspaces

Author: codetheweb

seamapi/workspaces.py

@@ -6,14 +6,13 @@
     ResetSandBoxResponse,
 )
 from typing import Optional, List, Union
-import requests
 
 from seamapi.utils.convert_to_id import to_workspace_id
 
 
 class Workspaces(AbstractWorkspaces):
     """
-    A class used to retreive workspace data
+    A class used to retrieve workspace data
     through interaction with Seam API
 
     ...
@@ -69,21 +68,15 @@ def list(
         """
 
         workspace_id = None if workspace is None else to_workspace_id(workspace)
-        res = requests.get(
-            f"{self.seam.api_url}/workspaces/list",
+        res = self.seam.make_request(
+            "GET",
+            "/workspaces/list",
             params={"workspace_id": workspace_id},
-            headers={"Authorization": f"Bearer {self.seam.api_key}"},
         )
-        if res.status_code == 404:
-            raise Exception("workspaces not found")  # TODO custom exception
-        if res.status_code != 200:
-            raise Exception(res.text)
-        res_json = res.json()
-        return res_json["workspaces"]
+        return res["workspaces"]
 
     def get(
-        self,
-        workspace: Optional[Union[WorkspaceId, Workspace]] = None,
+        self
     ) -> Workspace:
         """Gets a workspace.
 
@@ -103,21 +96,14 @@ def get(
         ------
             Workspace
         """
-        workspace_id = None if workspace is None else to_workspace_id(workspace)
-        res = requests.get(
-            f"{self.seam.api_url}/workspaces/get",
-            params={"workspace_id": workspace_id},
-            headers={"Authorization": f"Bearer {self.seam.api_key}"},
+        res = self.seam.make_request(
+            "GET",
+            "/workspaces/get",
         )
-        if res.status_code == 404:
-            raise Exception("workspace not found")  # TODO custom exception
-        if res.status_code != 200:
-            raise Exception(res.text)
-        res_json = res.json()
         return Workspace(
-            workspace_id=res_json["workspace"]["workspace_id"],
-            name=res_json["workspace"]["name"],
-            is_sandbox=res_json["workspace"]["is_sandbox"],
+            workspace_id=res["workspace"]["workspace_id"],
+            name=res["workspace"]["name"],
+            is_sandbox=res["workspace"]["is_sandbox"],
         )
 
     def reset_sandbox(self) -> None:
@@ -132,13 +118,10 @@ def reset_sandbox(self) -> None:
         ------
             ResetSandBoxResponse
         """
-
-        res = requests.post(
-            f"{self.seam.api_url}/workspaces/reset_sandbox",
-            headers={"Authorization": f"Bearer {self.seam.api_key}"},
+        res = self.seam.make_request(
+            "POST",
+            "/workspaces/reset_sandbox",
         )
-        if not res.ok:
-            raise Exception(res.text)
 
         return ResetSandBoxResponse(
             message="Successfully reset workspace sandbox",