diff options
author | Peter Fong <k2fong@hotmail.com> | 2017-04-11 11:34:41 -0700 |
---|---|---|
committer | Peter Fong <k2fong@hotmail.com> | 2017-04-11 11:34:41 -0700 |
commit | f51f52daf1fc511dd3650abe454f7f1502230007 (patch) | |
tree | 3dea2be8382b67ee2e96f9d03dcc2d0edf4f9649 | |
parent | d790f5d419b1fe8b15156f740c2031a2d4c9e7dd (diff) | |
download | timeout-decorator-f51f52daf1fc511dd3650abe454f7f1502230007.tar.gz |
Fixed an issue for decorating class methods
-rw-r--r-- | timeout_decorator/timeout_decorator.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/timeout_decorator/timeout_decorator.py b/timeout_decorator/timeout_decorator.py index 466f33d..aadc586 100644 --- a/timeout_decorator/timeout_decorator.py +++ b/timeout_decorator/timeout_decorator.py @@ -73,7 +73,10 @@ def timeout(seconds=None, use_signals=True, timeout_exception=TimeoutError): signal.signal(signal.SIGALRM, old) return new_function else: - return _Timeout(function, timeout_exception, seconds) + @wraps(function) + def new_function(*args, **kwargs): + return _Timeout(function, timeout_exception, seconds) + return new_function return decorate |