Merge pull request #32 from yxd-hde/temp-fix-for-windows

Add timer timeout implementation for windows.

Author: freedomofkeima

lambda_local/timeout.py

@@ -4,7 +4,9 @@
 '''
 
 import signal
+import threading
 from contextlib import contextmanager
+from six.moves import _thread
 
 
 class TimeoutException(Exception):
@@ -13,11 +15,21 @@ class TimeoutException(Exception):
 
 @contextmanager
 def time_limit(seconds):
-    def signal_handler(signum, frame):
-        raise TimeoutException("Timeout after {} seconds.".format(seconds))
-    signal.signal(signal.SIGALRM, signal_handler)
-    signal.alarm(seconds)
-    try:
-        yield
-    finally:
-        signal.alarm(0)
+    if hasattr(signal, "SIGALRM"):
+        def signal_handler(signum, frame):
+            raise TimeoutException("Timeout after {} seconds.".format(seconds))
+        signal.signal(signal.SIGALRM, signal_handler)
+        signal.alarm(seconds)
+        try:
+            yield
+        finally:
+            signal.alarm(0)
+    else:
+        timer = threading.Timer(seconds, lambda: _thread.interrupt_main())
+        timer.start()
+        try:
+            yield
+        except KeyboardInterrupt:
+            raise TimeoutException("Timeout after {} seconds.".format(seconds))
+        finally:
+            timer.cancel()