feat(lock): add wait

Signed-off-by: Chris Snow <[email protected]>

Author: snowch

hpecp/cli/lock.py

@@ -70,12 +70,17 @@ def list(
             print(json.dumps(response))
 
     @base.intercept_exception
-    def create(
-        self,
-        reason,
-    ):
+    def create(self, reason, timeout_secs=300):
         """Create a lock."""
-        print(base.get_client().lock.create(reason), file=sys.stdout)
+        response = base.get_client().lock.create(reason, timeout_secs)
+        if response is False:
+            print(
+                "Unabled to lock within '{}'".format(timeout_secs),
+                file=sys.stderr,
+            )
+        else:
+            # reponse contains lock ID
+            print(response, file=sys.stderr)
 
     @base.intercept_exception
     def delete(

hpecp/lock.py

@@ -45,13 +45,15 @@ def list(self):
         """Retrieve the locks"""
         return self.get()
 
-    def create(self, reason=None):
+    def create(self, reason=None, timeout_secs=300):
         """Create a new lock.
 
         Arguments
         ---------
         reason: str
             Provide a reason for the lock.
+        timeout_secs: int
+            Time to wait for lock to be successful
 
         Raises
         ------
@@ -64,7 +66,21 @@ def create(self, reason=None):
             data=data,
             description="lock/set_lock",
         )
-        return CaseInsensitiveDict(response.headers)["Location"]
+        id = CaseInsensitiveDict(response.headers)["Location"]
+
+        if timeout_secs == 0:
+            return id
+        else:
+            try:
+                polling.poll(
+                    lambda: self.get()["locked"],
+                    step=10,
+                    poll_forever=False,
+                    timeout=timeout_secs,
+                )
+                return id
+            except polling.TimeoutException:
+                return False
 
     def delete(self, lock_id):
         """Delete a lock.