Ability to add or subtract value on workspace capacity

predat · Feb 19, 2016 · 864598f · 864598f
1 parent 77798f6
commit 864598f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pyIsis/connection.py b/pyIsis/connection.py
@@ -260,7 +260,12 @@ def create_workspace(self, name='default_name', capacity=100, *options):
     def update_workspace_capacity(self, name="", capacity=None):
         workspace = self.get_workspace_details(name)
         modify_workspace = copy.deepcopy(workspace)
-        modify_workspace.ioByteCount = int(capacity)
+        if '+' in capacity:
+            modify_workspace.ioByteCount =  modify_workspace.ioByteCount + (int(capacity) * 1024)
+        elif '-' in capacity:
+            modify_workspace.ioByteCount =  modify_workspace.ioByteCount - (int(capacity) * 1024)
+        else:
+            modify_workspace.ioByteCount = int(capacity)
         return self._client.service.ModifyWorkspaceDetails(self.token,
                                                            workspace,
                                                            modify_workspace)