factor wait code out in wait_util

acrlewis · Mar 22, 2016 · 5e6cdde · 5e6cdde
1 parent 44c49c3
commit 5e6cdde
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/wait_util.py b/wait_util.py
@@ -0,0 +1,24 @@
+import time
+
+
+class TimeoutException(Exception):
+    pass
+
+
+def wait_while(condition, refresh=lambda: True):
+    def waitfor(timeout):
+        waited = 0
+        SLEEPTIME = 5
+
+        refresh()
+        if not condition():
+            return True
+
+        while waited < timeout and condition():
+            time.sleep(SLEEPTIME)
+            waited += SLEEPTIME
+            refresh()
+        if condition():
+            raise TimeoutException()
+
+    return waitfor