socketUtils.py

#socketUtils.py

import socket
import pickle
import select

MINER_SERVER_PORT = 5005
WALLET_SERVER_PORT = 5006
BUFFER_SIZE = 1024


def newServerConnection(ip_addr, port=MINER_SERVER_PORT):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    #s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    s.bind((ip_addr, port))
    s.listen()

    return s


def recvObj(socket):
    inputs, outpus, errs = select.select([socket], [], [socket], 6)
    if socket in inputs:
        new_sock, addr = socket.accept()

        all_data = b'';
        while True:
            data = new_sock.recv(BUFFER_SIZE)
            if not data: break
            all_data += data

        return pickle.loads(all_data)

    return None

def sendObj(ip_addr, inObj, port=MINER_SERVER_PORT):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.connect((ip_addr, port))

    data = pickle.dumps(inObj)

    s.send(data)
    s.close()

    return  False

if __name__ == '__main__':
    server = newServerConnection('localhost')
    O = recvObj(server)

    print("Success!") #If returns after time, then successful!
    #print(O)
    server.close()