From ee58bbeb5e8d870835c03d7f36b9976a0e8b857f Mon Sep 17 00:00:00 2001 From: jeanpouget-abadie Date: Fri, 5 Dec 2014 12:19:45 -0500 Subject: timeout decorator better usage --- jpa_test/timeout.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'jpa_test/timeout.py') 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) -- cgit v1.2.3-70-g09d2