83 lines
1.7 KiB
Python
83 lines
1.7 KiB
Python
|
import json
|
|||
|
|
|
|||
|
|
import mysql as sql
|
|||
|
|
import prot_codec as pr
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
SQL=sql.sql()
|
|||
|
|
SQL.init("test_data")
|
|||
|
|
|
|||
|
|
|
|||
|
|
_DATA=None
|
|||
|
|
|
|||
|
|
# 获取文件夹目录，参数：无
|
|||
|
|
def get_dirs_list():
|
|||
|
|
r=SQL.show_tables()
|
|||
|
|
j=[]
|
|||
|
|
for item in r:
|
|||
|
|
j.append(item[0])
|
|||
|
|
js=json.dumps(j)
|
|||
|
|
return js
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 获取文件列表，参数：json{文件的目录}
|
|||
|
|
def get_files_list(j:bytearray):
|
|||
|
|
SQL.table_name=json.loads(j)["dir"]
|
|||
|
|
r=SQL.show()
|
|||
|
|
js=json.dumps(r)
|
|||
|
|
print(js)
|
|||
|
|
return js
|
|||
|
|
|
|||
|
|
# 获取文件，参数：json{文件目录，文件序号}
|
|||
|
|
def get_file_data(j:bytearray):
|
|||
|
|
js=json.loads(j)
|
|||
|
|
name,data=SQL.download_data(js["dir"],js["id"])
|
|||
|
|
print(name,len(data))
|
|||
|
|
global _DATA
|
|||
|
|
_DATA=data
|
|||
|
|
js=json.dumps({'name':name,"size":len(data)})
|
|||
|
|
return js
|
|||
|
|
|
|||
|
|
# 获取指定长度的数据,参数：json{偏移，长度}
|
|||
|
|
def get_data(j:bytearray):
|
|||
|
|
js=json.loads(j)
|
|||
|
|
off=js["off"]
|
|||
|
|
length=js["len"]
|
|||
|
|
if(off<len(_DATA)):
|
|||
|
|
data=_DATA[off:]
|
|||
|
|
if(length<len(data)):
|
|||
|
|
return data[:length]
|
|||
|
|
else:
|
|||
|
|
return data
|
|||
|
|
return bytearray()
|
|||
|
|
|
|||
|
|
_cmd_list={
|
|||
|
|
"get_dirs":get_dirs_list,
|
|||
|
|
"get_files":get_files_list,
|
|||
|
|
"get_file":get_file_data,
|
|||
|
|
"get_data":get_data,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
|
|||
|
|
# dirs_str=get_dirs()
|
|||
|
|
# data=pr.encode(b"dirs",dirs_str.encode("utf-8"))
|
|||
|
|
# print(data.hex(' '))
|
|||
|
|
# print(pr.decode(data))
|
|||
|
|
# files=_cmd_list["get_files"](b"checker_host_app")
|
|||
|
|
# print(json.loads(files))
|
|||
|
|
# print(list(_cmd_list.keys()))
|
|||
|
|
data=_cmd_list["get_file"](json.dumps({"dir":"checker_host_app","id":42}))
|
|||
|
|
print(data)
|
|||
|
|
ret=bytearray()
|
|||
|
|
off=0
|
|||
|
|
while off<len(_DATA):
|
|||
|
|
ret+=_cmd_list["get_data"](json.dumps({"off":off,"len":200}))
|
|||
|
|
off+=200
|
|||
|
|
if(ret==_DATA):
|
|||
|
|
print("check success")
|
|||
|
|
else:
|
|||
|
|
print("check failed")
|