fix: 修复任务完成字段错误

pull/7/head
梁浩 5 months ago
parent db94e35efd
commit 0ce6747b76

@ -162,7 +162,7 @@ class TaskService:
'flow_id': task.flow_id,
'created_at': task.created_at.isoformat() if task.created_at else None,
'started_at': task.started_at.isoformat() if task.started_at else None,
'completed_at': task.completed_at.isoformat() if task.completed_at else None
'finished_at': task.finished_at.isoformat() if task.finished_at else None
}
# 如果任务失败尝试从RQ获取错误信息
@ -228,7 +228,7 @@ class TaskService:
failed_status = TaskStatus.query.filter_by(task_status_code='failed').first()
if failed_status:
task.tasks_status_id = failed_status.task_status_id
task.completed_at = datetime.utcnow()
task.finished_at = datetime.utcnow()
db.session.commit()
return True

@ -85,7 +85,7 @@ def run_evaluate_task(task_id, clean_ref_dir, clean_output_dir,
completed_status = TaskStatus.query.filter_by(task_status_code='completed').first()
if completed_status:
task.tasks_status_id = completed_status.task_status_id
task.completed_at = datetime.utcnow()
task.finished_at = datetime.utcnow()
db.session.commit()
logger.info(f"Evaluate task {task_id} completed")
@ -98,7 +98,7 @@ def run_evaluate_task(task_id, clean_ref_dir, clean_output_dir,
failed_status = TaskStatus.query.filter_by(task_status_code='failed').first()
if failed_status:
task.tasks_status_id = failed_status.task_status_id
task.completed_at = datetime.utcnow()
task.finished_at = datetime.utcnow()
db.session.commit()
return {'success': False, 'error': str(e)}

@ -112,7 +112,7 @@ def run_heatmap_task(task_id, original_image_path, perturbed_image_path,
completed_status = TaskStatus.query.filter_by(task_status_code='completed').first()
if completed_status:
task.tasks_status_id = completed_status.task_status_id
task.completed_at = datetime.utcnow()
task.finished_at = datetime.utcnow()
db.session.commit()
logger.info(f"Heatmap task {task_id} completed")
@ -125,7 +125,7 @@ def run_heatmap_task(task_id, original_image_path, perturbed_image_path,
failed_status = TaskStatus.query.filter_by(task_status_code='failed').first()
if failed_status:
task.tasks_status_id = failed_status.task_status_id
task.completed_at = datetime.utcnow()
task.finished_at = datetime.utcnow()
db.session.commit()
return {'success': False, 'error': str(e)}

Loading…
Cancel
Save