65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
"""Celery 任务"""
|
|
from celery import Task
|
|
from pathlib import Path
|
|
import shutil
|
|
import logging
|
|
|
|
from ..core.celery_app import celery_app
|
|
from ..core.docker_client import DockerManager
|
|
from ..database import SessionLocal
|
|
from ..models.job import Job, JobStatus
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
@celery_app.task(bind=True)
|
|
def run_bttoxin_analysis(
|
|
self,
|
|
job_id: str,
|
|
input_dir: str,
|
|
output_dir: str,
|
|
sequence_type: str = "nucl",
|
|
scaf_suffix: str = ".fna",
|
|
threads: int = 4
|
|
):
|
|
"""执行分析任务"""
|
|
db = SessionLocal()
|
|
|
|
try:
|
|
job = db.query(Job).filter(Job.id == job_id).first()
|
|
job.status = JobStatus.RUNNING
|
|
db.commit()
|
|
|
|
self.update_state(
|
|
state='PROGRESS',
|
|
meta={'current': 20, 'total': 100, 'status': 'Running analysis...'}
|
|
)
|
|
|
|
docker_manager = DockerManager()
|
|
result = docker_manager.run_bttoxin_digger(
|
|
input_dir=Path(input_dir),
|
|
output_dir=Path(output_dir),
|
|
sequence_type=sequence_type,
|
|
scaf_suffix=scaf_suffix,
|
|
threads=threads
|
|
)
|
|
|
|
if result['success']:
|
|
job.status = JobStatus.COMPLETED
|
|
job.logs = result.get('logs', '')
|
|
else:
|
|
job.status = JobStatus.FAILED
|
|
job.error_message = result.get('error', 'Analysis failed')
|
|
|
|
db.commit()
|
|
|
|
return {'job_id': job_id, 'status': job.status}
|
|
|
|
except Exception as e:
|
|
logger.error(f"Task failed: {e}")
|
|
job.status = JobStatus.FAILED
|
|
job.error_message = str(e)
|
|
db.commit()
|
|
raise
|
|
finally:
|
|
db.close()
|