diff --git a/app/task_worker.py b/app/task_worker.py index 2428d8e..f2b7a00 100644 --- a/app/task_worker.py +++ b/app/task_worker.py @@ -743,6 +743,9 @@ def do_task_upload(params): taskno = check_param({'taskno': request.form.get('taskno')}, 'taskno') if not taskno: return json.dumps(make_common_res(2, '任务id异常,请检查后重试')) + userid = check_param({'userid': request.form.get('userid')}, 'userid') + if not userid: + return json.dumps(make_common_res(4, '用户id缺失,请检查后重试')) if len(request.files.keys()) < 1: return json.dumps(make_common_res(1, '文件不存在')) suc_file_num, suc_mysql_num = 0, 0 @@ -756,7 +759,7 @@ def do_task_upload(params): os.chmod(f'./app/static/task_file/{nodeid}/task_no_{taskno}', 0o777) file.save(f'./app/static/task_file/{nodeid}/task_no_{taskno}/{file.filename}') suc_file_num += 1 - download_url = f'/api/download_task_file?nodeid={nodeid}&taskno={taskno}&file_name={file.filename}' + download_url = f'/api/download_task_file?nodeid={nodeid}&taskno={taskno}&file_name={file.filename}&area_id={area_id}&userid={userid}' ret = db_task.insert_upload_file_record(taskno, nodeid, download_url, area_id) if ret == 1: suc_mysql_num += 1