diff options
| author | jeanpouget-abadie <jean.pougetabadie@gmail.com> | 2014-12-05 12:19:45 -0500 |
|---|---|---|
| committer | jeanpouget-abadie <jean.pougetabadie@gmail.com> | 2014-12-05 12:19:45 -0500 |
| commit | ee58bbeb5e8d870835c03d7f36b9976a0e8b857f (patch) | |
| tree | 90c5a767ab9db9cd5247be60029bf3a92d6bda28 /jpa_test/timeout.py | |
| parent | 67772e10f297d23e8d99b3901d044a3bb3345214 (diff) | |
| download | cascades-ee58bbeb5e8d870835c03d7f36b9976a0e8b857f.tar.gz | |
timeout decorator better usage
Diffstat (limited to 'jpa_test/timeout.py')
| -rw-r--r-- | jpa_test/timeout.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/jpa_test/timeout.py b/jpa_test/timeout.py index 52d7d92..d7381c3 100644 --- a/jpa_test/timeout.py +++ b/jpa_test/timeout.py @@ -3,10 +3,13 @@ import errno import os import signal +class TimeoutError(Exception): + pass + def timeout(seconds=10, error_message=os.strerror(errno.ETIME)): def decorator(func): def _handle_timeout(signum, frame): - raise Exception(error_message) + raise TimeoutError(error_message) def wrapper(*args, **kwargs): signal.signal(signal.SIGALRM, _handle_timeout) |