Initial commit: BtToxin Pipeline project structure

backend/app/workers/tasks.py

@@ -0,0 +1,64 @@
+"""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()