feat: task_submit/control/modify/restart 加 owner 校验(通用引擎任务放行)
This commit is contained in:
parent
75c6d9d003
commit
c3508867d4
@ -8,6 +8,11 @@ if not task_id:
|
|||||||
if action not in ('pause', 'resume', 'cancel'):
|
if action not in ('pause', 'resume', 'cancel'):
|
||||||
return json.dumps({"success": False, "message": "action必须是pause/resume/cancel"}, ensure_ascii=False)
|
return json.dumps({"success": False, "message": "action必须是pause/resume/cancel"}, ensure_ascii=False)
|
||||||
|
|
||||||
|
_uid = await get_user()
|
||||||
|
_ok, _emsg = await check_task_owner(task_id, _uid)
|
||||||
|
if not _ok:
|
||||||
|
return json.dumps({"success": False, "message": _emsg}, ensure_ascii=False)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if action == 'pause':
|
if action == 'pause':
|
||||||
result = await pipeline_pause(tenant_id, task_id)
|
result = await pipeline_pause(tenant_id, task_id)
|
||||||
|
|||||||
@ -7,6 +7,11 @@ rerun_from = params_kw.get('rerun_from', 'node')
|
|||||||
if not task_id or not step_name:
|
if not task_id or not step_name:
|
||||||
return json.dumps({"success": False, "message": "缺少task_id或step_name"}, ensure_ascii=False)
|
return json.dumps({"success": False, "message": "缺少task_id或step_name"}, ensure_ascii=False)
|
||||||
|
|
||||||
|
_uid = await get_user()
|
||||||
|
_ok, _emsg = await check_task_owner(task_id, _uid)
|
||||||
|
if not _ok:
|
||||||
|
return json.dumps({"success": False, "message": _emsg}, ensure_ascii=False)
|
||||||
|
|
||||||
# Parse content
|
# Parse content
|
||||||
if isinstance(content_raw, str):
|
if isinstance(content_raw, str):
|
||||||
try:
|
try:
|
||||||
|
|||||||
@ -4,6 +4,11 @@ task_id = params_kw.get('task_id', '')
|
|||||||
if not task_id:
|
if not task_id:
|
||||||
return json.dumps({"success": False, "message": "缺少task_id"}, ensure_ascii=False)
|
return json.dumps({"success": False, "message": "缺少task_id"}, ensure_ascii=False)
|
||||||
|
|
||||||
|
_uid = await get_user()
|
||||||
|
_ok, _emsg = await check_task_owner(task_id, _uid)
|
||||||
|
if not _ok:
|
||||||
|
return json.dumps({"success": False, "message": _emsg}, ensure_ascii=False)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
result = await pipeline_restart(tenant_id, task_id)
|
result = await pipeline_restart(tenant_id, task_id)
|
||||||
return result
|
return result
|
||||||
|
|||||||
@ -8,6 +8,10 @@ if not pipeline_id:
|
|||||||
if not title:
|
if not title:
|
||||||
return json.dumps({"success": False, "message": "缺少任务标题"}, ensure_ascii=False)
|
return json.dumps({"success": False, "message": "缺少任务标题"}, ensure_ascii=False)
|
||||||
|
|
||||||
|
_ok, _emsg = await check_tenant_owner(tenant_id, owner_id)
|
||||||
|
if not _ok:
|
||||||
|
return json.dumps({"success": False, "message": _emsg}, ensure_ascii=False)
|
||||||
|
|
||||||
# Collect optional params
|
# Collect optional params
|
||||||
task_params = {}
|
task_params = {}
|
||||||
for key in ['input_audio', 'input_video', 'input_text', 'lyrics', 'mode', 'scene']:
|
for key in ['input_audio', 'input_video', 'input_text', 'lyrics', 'mode', 'scene']:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user