PMFlow/backend/app/models/task.py

38 lines
1.8 KiB
Python

"""任务模型"""
from .base import Base
from .types import JSONB, ARRAY
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy import BigInteger, Integer, VARCHAR, Text, DECIMAL, TIMESTAMP, ForeignKey
from typing import Optional, List
from datetime import datetime
class Task(Base):
__tablename__ = "tasks"
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
truth_node_id: Mapped[int] = mapped_column(
BigInteger, ForeignKey("truth_nodes.id"), nullable=False
)
project_id: Mapped[int] = mapped_column(
BigInteger, ForeignKey("projects.id"), nullable=False
)
assignee_id: Mapped[Optional[int]] = mapped_column(
BigInteger, ForeignKey("users.id"), nullable=True
)
ai_recommended_assignees: Mapped[Optional[dict]] = mapped_column(JSONB, nullable=True)
ai_breakdown_payload: Mapped[Optional[dict]] = mapped_column(JSONB, nullable=True)
manual_overrides: Mapped[Optional[dict]] = mapped_column(JSONB, nullable=True)
title: Mapped[str] = mapped_column(VARCHAR(500), nullable=False)
description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
required_skill_tag_ids: Mapped[List[int]] = mapped_column(
ARRAY(BigInteger), nullable=False
)
status: Mapped[str] = mapped_column(VARCHAR(50), nullable=False)
priority: Mapped[str] = mapped_column(VARCHAR(50), nullable=False)
estimated_hours: Mapped[Optional[float]] = mapped_column(DECIMAL(5, 2), nullable=True)
actual_hours: Mapped[Optional[float]] = mapped_column(DECIMAL(5, 2), nullable=True)
quality_score: Mapped[Optional[float]] = mapped_column(DECIMAL(3, 1), nullable=True)
due_date: Mapped[Optional[datetime]] = mapped_column(TIMESTAMP, nullable=True)
completed_at: Mapped[Optional[datetime]] = mapped_column(TIMESTAMP, nullable=True)