MalmoEnv/malmoenv/comms.py [26:68]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
retry_timeout = 10


def retry(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        retry_exc = None
        for i in range(retry_count):
            try:
                return func(*args, **kwargs)
            except Exception as e:
                if retry_exc is None:
                    retry_exc = e
                if i < retry_count - 1:
                    print("Pause before retry on " + str(e))
                    time.sleep(retry_timeout)
        raise retry_exc
    return wrapper


def send_message(sock, data):
    length = len(data)
    sock.sendall(struct.pack('!I', length))
    sock.sendall(data)


def recv_message(sock):
    lengthbuf = recvall(sock, 4)
    if not lengthbuf:
        return None
    length, = struct.unpack('!I', lengthbuf)
    return recvall(sock, length)


def recvall(sock, count):
    buf = b''
    while count:
        newbuf = sock.recv(count)
        if not newbuf:
            return None
        buf += newbuf
        count -= len(newbuf)
    return buf
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



MalmoEnv/proxyenv/comms.py [26:68]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
retry_timeout = 10


def retry(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        retry_exc = None
        for i in range(retry_count):
            try:
                return func(*args, **kwargs)
            except Exception as e:
                if retry_exc is None:
                    retry_exc = e
                if i < retry_count - 1:
                    print("Pause before retry on " + str(e))
                    time.sleep(retry_timeout)
        raise retry_exc
    return wrapper


def send_message(sock, data):
    length = len(data)
    sock.sendall(struct.pack('!I', length))
    sock.sendall(data)


def recv_message(sock):
    lengthbuf = recvall(sock, 4)
    if not lengthbuf:
        return None
    length, = struct.unpack('!I', lengthbuf)
    return recvall(sock, length)


def recvall(sock, count):
    buf = b''
    while count:
        newbuf = sock.recv(count)
        if not newbuf:
            return None
        buf += newbuf
        count -= len(newbuf)
    return buf
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -