egg/Document_Translator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

16th_fix

Browse Source
This commit is contained in:
beabigegg
2025-09-04 18:34:05 +08:00
parent f093f4bbc2
commit 6eabdb2f07
35 changed files with 1097 additions and 212 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
app/api/jobs.py
View File

@@ -40,8 +40,8 @@ def get_user_jobs():
# 驗證分頁參數
page, per_page = validate_pagination(page, per_page)
# 建立查詢
query = TranslationJob.query.filter_by(user_id=g.current_user_id)
# 建立查詢（排除軟刪除的記錄）
query = TranslationJob.query.filter_by(user_id=g.current_user_id).filter(TranslationJob.deleted_at.is_(None))
# 狀態篩選
if status and status != 'all':
@@ -118,8 +118,8 @@ def get_job_detail(job_uuid):
# 驗證 UUID 格式
validate_job_uuid(job_uuid)
# 取得任務
job = TranslationJob.query.filter_by(job_uuid=job_uuid).first()
# 取得任務（排除軟刪除的記錄）
job = TranslationJob.query.filter_by(job_uuid=job_uuid).filter(TranslationJob.deleted_at.is_(None)).first()
if not job:
return jsonify(create_response(
@@ -194,8 +194,8 @@ def retry_job(job_uuid):
# 驗證 UUID 格式
validate_job_uuid(job_uuid)
# 取得任務
job = TranslationJob.query.filter_by(job_uuid=job_uuid).first()
# 取得任務（排除軟刪除的記錄）
job = TranslationJob.query.filter_by(job_uuid=job_uuid).filter(TranslationJob.deleted_at.is_(None)).first()
if not job:
return jsonify(create_response(
@@ -373,13 +373,13 @@ def get_queue_status():
@jobs_bp.route('/<job_uuid>/cancel', methods=['POST'])
@jwt_login_required
def cancel_job(job_uuid):
"""取消任務（僅限 PENDING 狀態）"""
"""取消任務（支援 PENDING 和 PROCESSING 狀態）"""
try:
# 驗證 UUID 格式
validate_job_uuid(job_uuid)
# 取得任務
job = TranslationJob.query.filter_by(job_uuid=job_uuid).first()
# 取得任務（排除軟刪除的記錄）
job = TranslationJob.query.filter_by(job_uuid=job_uuid).filter(TranslationJob.deleted_at.is_(None)).first()
if not job:
return jsonify(create_response(
@@ -396,16 +396,28 @@ def cancel_job(job_uuid):
message='無權限操作此任務'
)), 403
# 只能取消等待中的任務
if job.status != 'PENDING':
# 只能取消等待中或處理中的任務
if job.status not in ['PENDING', 'PROCESSING']:
return jsonify(create_response(
success=False,
error='CANNOT_CANCEL',
message='只能取消等待中的任務'
message='只能取消等待中或處理中的任務'
)), 400
# 如果是處理中的任務，需要中斷 Celery 任務
if job.status == 'PROCESSING':
try:
from app.services.celery_service import revoke_task
# 嘗試撤銷 Celery 任務
revoke_task(job.job_uuid)
logger.info(f"Celery task revoked for job: {job.job_uuid}")
except Exception as celery_error:
logger.warning(f"Failed to revoke Celery task for job {job.job_uuid}: {celery_error}")
# 即使撤銷失敗也繼續取消任務，因為用戶請求取消
# 更新任務狀態為失敗（取消）
job.update_status('FAILED', error_message='使用者取消任務')
cancel_message = f'使用者取消任務 (原狀態: {job.status})'
job.update_status('FAILED', error_message=cancel_message)
# 記錄取消日誌
SystemLog.info(
@@ -469,13 +481,16 @@ def delete_job(job_uuid):
message='無權限操作此任務'
)), 403
# 檢查任務狀態 - 不能刪除正在處理中的任務
# 如果是處理中的任務，先嘗試中斷 Celery 任務
if job.status == 'PROCESSING':
return jsonify(create_response(
success=False,
error='CANNOT_DELETE',
message='無法刪除正在處理中的任務'
)), 400
try:
from app.services.celery_service import revoke_task
# 嘗試撤銷 Celery 任務
revoke_task(job.job_uuid)
logger.info(f"Celery task revoked before deletion for job: {job.job_uuid}")
except Exception as celery_error:
logger.warning(f"Failed to revoke Celery task before deletion for job {job.job_uuid}: {celery_error}")
# 即使撤銷失敗也繼續刪除任務，因為用戶要求刪除
# 刪除任務相關檔案
import os
@@ -506,11 +521,10 @@ def delete_job(job_uuid):
from app import db
# 刪除資料庫記錄
db.session.delete(job)
db.session.commit()
# 軟刪除資料庫記錄（保留數據供報表使用）
job.soft_delete()
logger.info(f"Job deleted by user: {job_uuid}")
logger.info(f"Job soft deleted by user: {job_uuid}")
return jsonify(create_response(
success=True,