"""任务模型"""
from sqlalchemy import Column, String, Integer, DateTime, JSON, Enum, Text
from sqlalchemy.sql import func
import enum

from ..database import Base


class JobStatus(str, enum.Enum):
    """任务状态"""
    PENDING = "pending"      # 等待进入队列
    QUEUED = "queued"        # 已排队，等待执行
    RUNNING = "running"      # 正在执行
    COMPLETED = "completed"  # 执行完成
    FAILED = "failed"        # 执行失败


class Job(Base):
    __tablename__ = "jobs"

    id = Column(String, primary_key=True, index=True)
    celery_task_id = Column(String, nullable=True, index=True)
    status = Column(Enum(JobStatus), default=JobStatus.PENDING, index=True)

    input_files = Column(JSON)
    sequence_type = Column(String, default="nucl")
    scaf_suffix = Column(String, default=".fna")
    threads = Column(Integer, default=4)

    # 分析参数
    min_identity = Column(Integer, default=80)  # 存储为百分比 (0-100)
    min_coverage = Column(Integer, default=60)
    allow_unknown_families = Column(Integer, default=0)  # 0 = False, 1 = True
    require_index_hit = Column(Integer, default=1)

    result_url = Column(String, nullable=True)
    logs = Column(Text, nullable=True)
    error_message = Column(Text, nullable=True)

    # 队列位置
    queue_position = Column(Integer, nullable=True)

    # 进度信息
    current_stage = Column(String, nullable=True)  # digger, shoter, plots, bundle
    progress_percent = Column(Integer, default=0)

    created_at = Column(DateTime(timezone=True), server_default=func.now(), index=True)
    started_at = Column(DateTime(timezone=True), nullable=True)
    completed_at = Column(DateTime(timezone=True), nullable=True)
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())