def get_all_codes():
"""
获取所有代码仓库
:return:
"""
if get_login_user_id() == 0:
return redirect("/user/account")
user_id = get_login_user_id()
lists = CodeModel.select().where(CodeModel.status != 2, CodeModel.user_id == user_id)
objs = list(map(CodeModel.get_dict_from_obj, lists))
response.set_header("Content-Type", "application/json")
return json.dumps(objs)
def download(warehouse, token):
"""
下载代码
:param warehouse: 仓库名称
:param token: 下载凭证
:return:
"""
try:
obj = CodeModel.select().where(CodeModel.warehouse == warehouse, CodeModel.token == token, CodeModel.status == 0).get()
except Exception:
return abort(404)
fp_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__))) + "/data/code"
return static_file(str(obj.id)+".zip", root=fp_path, download=True)
def get_all_codes():
"""
获取所有代码仓库
:return:
"""
if get_login_user_id() == 0:
return redirect("/user/account")
user_id = get_login_user_id()
lists = CodeModel.select().where(CodeModel.status != 2,
CodeModel.user_id == user_id)
objs = list(map(CodeModel.get_dict_from_obj, lists))
response.set_header("Content-Type", "application/json")
return json.dumps(objs)
def upload_zip():
"""
上传文件
:return:
"""
if get_login_user_id() == 0:
return redirect("/user/account")
code_id = int(request.forms.code_id)
obj = CodeModel.select().where(CodeModel.user_id == get_login_user_id(), CodeModel.id == code_id).get()
fp = request.files['file']
fp_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__))) + "/data/code/"
fp.save(fp_path + str(obj.id) + ".zip", overwrite=True)
return "ok"
def download(warehouse, token):
"""
下载代码
:param warehouse: 仓库名称
:param token: 下载凭证
:return:
"""
try:
obj = CodeModel.select().where(CodeModel.warehouse == warehouse,
CodeModel.token == token,
CodeModel.status == 0).get()
except Exception:
return abort(404)
fp_path = os.path.dirname(os.path.dirname(
os.path.realpath(__file__))) + "/data/code"
return static_file(str(obj.id) + ".zip", root=fp_path, download=True)
def upload_zip():
"""
上传文件
:return:
"""
if get_login_user_id() == 0:
return redirect("/user/account")
code_id = int(request.forms.code_id)
obj = CodeModel.select().where(CodeModel.user_id == get_login_user_id(),
CodeModel.id == code_id).get()
fp = request.files['file']
fp_path = os.path.dirname(os.path.dirname(
os.path.realpath(__file__))) + "/data/code/"
fp.save(fp_path + str(obj.id) + ".zip", overwrite=True)
return "ok"
