def sleep(sleep_time): timeout, exception = timeouts.next_timeout() if timeout >= 0 and timeout < sleep_time: _orig_sleep(timeout) raise exception else: stdlib_time.sleep(sleep_time)
def new_func(*args, **kwargs): new_func.__doc__ = orig_func.__doc__ new_func.__name__ = orig_func.__name__ timeout, exception = timeouts.next_timeout() if timeout is None: return orig_func(*args, **kwargs) if timeout <= 0: raise exception self.settimeout(timeout) try: return orig_func(*args, **kwargs) except stdlib_socket.timeout: cls, value, traceback = sys.exc_info() raise type(exception), exception, traceback