new module for interruptable threads

stamparm · stamparm · commit 60f04f0a4197 · 2010-04-06T14:33:57.000Z
diff --git a/lib/utils/timeout.py b/lib/utils/timeout.py
@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+
+import threading
+
+def timeout(func, args=(), kwargs={}, timeout_duration=1, default=None):
+    class InterruptableThread(threading.Thread):
+        def __init__(self):
+            threading.Thread.__init__(self)
+            self.result = None
+
+        def run(self):
+            try:
+                self.result = func(*args, **kwargs)
+            except:
+                self.result = default
+
+    thread = InterruptableThread()
+    thread.start()
+    thread.join(timeout_duration)
+    if thread.isAlive():
+        return default
+    else:
+        return thread.result
