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 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"