def getImage(key, width, height): #get node nodeAddr,nodePort = getNodeAddr(key) nodeSocket = do_connect(nodeAddr, nodePort) #send method util.w_send(nodeSocket, "1") req = WPB_pb2.ReqGet() req.imageID = key req.width = width req.height = height req.keep_proportion = True util.w_send_pb(nodeSocket,req) response = WPB_pb2.ReqResponse() respBuff = util.w_recv(nodeSocket) response.ParseFromString(respBuff) if response.rspCode == WPB_pb2.REQ_FAILED: print "ERROR GET IMAGE" return #get content imgContent = util.w_recv(nodeSocket) nodeSocket.close() return imgContent
def uploadImage(imgName, imgContent): #get file content content = imgContent reqKey = hashlib.md5(content).hexdigest() #print "content key is ",reqKey nodeAddr,nodePort = getNodeAddr(reqKey) nodeSocket = do_connect(nodeAddr, nodePort) #send method util.w_send(nodeSocket, "2") req = WPB_pb2.ReqSet() req.md5 = reqKey req.imageName = imgName util.w_send_pb(nodeSocket, req) #send file content util.w_send(nodeSocket, content) #get Response response = WPB_pb2.ReqResponse() respBuff = util.w_recv(nodeSocket) response.ParseFromString(respBuff) print "resp:new name is:", response.newname nodeSocket.close() return response.newname