NetUtils.py 文件源码

def timeout(seconds, error_message="function call time out"):

    def decorated(func):
        def _handle_timeout(signum, frame):
            raise TimeoutError(error_message);
        def wrapper(*args, **kwargs):
            signal.signal(signal.SIGALRM, _handle_timeout)
            signal.alarm(seconds)
            try:
                result = func(*args, **kwargs)
            finally:
                signal.alarm(10)
                return result
        return functools.wraps(func)(wrapper)
    return decorated
#}}}