def remove(self, name):
self.sk.send(package("remove"))
response(self.sk, "name", name)
rst = self.sk.recv(service.configure.msg_buffer).decode()
if rst == service.object_pool.success_msg:
return name + " has been removed"
else:
return rst
def send_file(sk, filename):
response(sk, "size", os.path.getsize(filename))
while not sk.recv(service.configure.msg_buffer).decode() == "data":
pass
fp = open(filename, "rb")
while True:
data = fp.read(service.configure.file_buffer)
if not data:
break
sk.send(data)
fp.close()
def exec(self, file):
self.sk.send(package("execute"))
index = file.rfind(os.path.sep)
filename = file[index + 1:]
response(self.sk, "filename", filename)
send_file(self.sk, file)
wait_cmd(self.sk, "output")
receive_file(self.sk, "output")
with open("output", "r") as fp:
data = fp.read()
os.remove("output")
return data
def compute_svd(self, src_name, dst_name, r=0.9):
self.sk.send(package("svd"))
response(self.sk, "name", src_name)
response(self.sk, "rstname", dst_name)
response(self.sk, "r", str(r))
rst = self.sk.recv(service.configure.msg_buffer).decode()
return rst
def compute_pca(self, src_name, dst_name, n=0):
self.sk.send(package("pca"))
response(self.sk, "name", src_name)
response(self.sk, "rstname", dst_name)
response(self.sk, "n", str(n))
rst = self.sk.recv(service.configure.msg_buffer).decode()
return rst
def load(self, name):
self.sk.send(package("load"))
response(self.sk, "name", name)
filename = self.sk.recv(service.configure.msg_buffer).decode()
if filename == service.configure.no_such_object:
return None
state = self.sk.recv(service.configure.msg_buffer).decode()
if state != service.object_pool.success_msg:
return None
receive_file(self.sk, filename)
if filename.split(".")[1] == "svm":
tmp_svm = calculate.classify.SVM.SVM()
tmp_svm.load(filename)
var = tmp_svm
elif filename.split(".")[1] == "knn":
tmp_knn = calculate.classify.KNN.KNN()
tmp_knn.load(filename)
var = tmp_knn
else:
var = DAL.file.load(filename)
os.remove(filename)
return var
def save(self, var, name):
self.sk.send(package("save"))
data_type = str(type(var))[8:-2]
if data_type == "calculate.classify.SVM.SVM":
data_type = "svm"
elif data_type == "calculate.classify.KNN.KNN":
data_type = "knn"
filename = name + "." + data_type
if data_type == "svm":
var.save(filename)
elif data_type == "knn":
var.save(filename)
else:
DAL.file.save(var, filename)
response(self.sk, "name", name)
response(self.sk, "type", data_type)
state = self.sk.recv(service.configure.msg_buffer).decode()
if state != service.object_pool.success_msg:
return state
send_file(self.sk, filename)
os.remove(filename)
return state
def delete(self, file):
self.sk.send(package("delete"))
response(self.sk, "file", file)
rst = self.sk.recv(service.configure.msg_buffer).decode()
return rst
def compute_feature(self, src_name, dst_name, mtd, option, main, unpack=False):
self.sk.send(package("feature"))
response(self.sk, "name", src_name)
response(self.sk, "rstname", dst_name)
response(self.sk, "mtd", str(mtd))
response(self.sk, "option", str(option))
response(self.sk, "main", str(main))
response(self.sk, "unpack", str(unpack))
rst = self.sk.recv(service.configure.msg_buffer).decode()
return rst
def clone(self, src_name, dst_name):
self.sk.send(package("clone"))
response(self.sk, "name", src_name)
response(self.sk, "clonename", dst_name)
rst = self.sk.recv(service.configure.msg_buffer).decode()
return rst