pvsune · May 31, 2020 15:03 · May 31, 2020
diff --git a/timeout.py b/timeout.py
@@ -0,0 +1,24 @@
+import logging
+from multiprocessing import Pool, TimeoutError
+
+
+logging.basicConfig(level=logging.INFO)
+
+
+def timeout(func, args=None, kwds=None, timeout=10):
+    """Call function and wait until timeout.
+
+    Cannot make into a decorator, fails in pickling function.
+    """
+    args = args or []
+    kwds = kwds or {}
+    with Pool(processes=1) as pool:
+        res = pool.apply_async(func, args, kwds)
+        try:
+            return res.get(timeout)
+        except TimeoutError:
+            logging.exception(
+                'Calling %s%s timed out after %ss',
+                func.__name__, tuple(args), timeout
+            )
+            raise