172 lines
7.1 KiB
Python
172 lines
7.1 KiB
Python
from datetime import datetime
|
|
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import Column, DateTime, Float, ForeignKey, Integer, JSON, String, Text
|
|
from sqlalchemy.dialects.mysql import LONGTEXT
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from app.models.base import Base, TimestampMixin
|
|
|
|
|
|
class ToolJob(Base, TimestampMixin):
|
|
__tablename__ = "tool_jobs"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
|
|
tool_name = Column(String(128), nullable=False, index=True)
|
|
status = Column(String(32), nullable=False, default="pending")
|
|
input_file_name = Column(String(255), nullable=False)
|
|
input_file_path = Column(String(512), nullable=False)
|
|
error_message = Column(Text, nullable=True)
|
|
started_at = Column(DateTime, nullable=True)
|
|
completed_at = Column(DateTime, nullable=True)
|
|
output_summary = Column(JSON, nullable=True)
|
|
|
|
user = relationship("User")
|
|
srs_extraction = relationship(
|
|
"SRSExtraction",
|
|
back_populates="job",
|
|
uselist=False,
|
|
cascade="all, delete-orphan",
|
|
)
|
|
testing_generation = relationship(
|
|
"TestingGeneration",
|
|
back_populates="job",
|
|
uselist=False,
|
|
cascade="all, delete-orphan",
|
|
foreign_keys="TestingGeneration.job_id",
|
|
)
|
|
|
|
|
|
class SRSExtraction(Base, TimestampMixin):
|
|
__tablename__ = "srs_extractions"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
job_id = Column(Integer, ForeignKey("tool_jobs.id", ondelete="CASCADE"), nullable=False, unique=True)
|
|
document_name = Column(String(255), nullable=False)
|
|
document_title = Column(String(255), nullable=False)
|
|
generated_at = Column(DateTime, default=datetime.utcnow, nullable=False)
|
|
total_requirements = Column(Integer, nullable=False, default=0)
|
|
statistics = Column(JSON, nullable=True)
|
|
raw_output = Column(JSON, nullable=True)
|
|
|
|
job = relationship("ToolJob", back_populates="srs_extraction")
|
|
requirements = relationship(
|
|
"SRSRequirement",
|
|
back_populates="extraction",
|
|
cascade="all, delete-orphan",
|
|
order_by="SRSRequirement.sort_order",
|
|
)
|
|
|
|
|
|
class SRSRequirement(Base, TimestampMixin):
|
|
__tablename__ = "srs_requirements"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
extraction_id = Column(Integer, ForeignKey("srs_extractions.id", ondelete="CASCADE"), nullable=False)
|
|
requirement_uid = Column(String(64), nullable=False)
|
|
title = Column(String(255), nullable=False)
|
|
description = Column(LONGTEXT, nullable=False)
|
|
priority = Column(String(16), nullable=False, default="中")
|
|
acceptance_criteria = Column(JSON, nullable=False)
|
|
source_field = Column(String(255), nullable=False)
|
|
section_uid = Column(String(64), nullable=True)
|
|
section_number = Column(String(64), nullable=True)
|
|
section_title = Column(String(255), nullable=True)
|
|
requirement_type = Column(String(64), nullable=True)
|
|
interface_name = Column(String(255), nullable=True)
|
|
interface_type = Column(String(128), nullable=True)
|
|
data_source = Column(String(255), nullable=True)
|
|
data_destination = Column(String(255), nullable=True)
|
|
sort_order = Column(Integer, nullable=False, default=0)
|
|
|
|
extraction = relationship("SRSExtraction", back_populates="requirements")
|
|
|
|
__table_args__ = (
|
|
sa.UniqueConstraint("extraction_id", "requirement_uid", name="uq_srs_extraction_requirement_uid"),
|
|
sa.Index("idx_srs_requirements_extraction_sort", "extraction_id", "sort_order"),
|
|
)
|
|
|
|
|
|
class TestingGeneration(Base, TimestampMixin):
|
|
__tablename__ = "testing_generations"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
job_id = Column(Integer, ForeignKey("tool_jobs.id", ondelete="CASCADE"), nullable=False, unique=True)
|
|
source_job_id = Column(Integer, ForeignKey("tool_jobs.id"), nullable=True, index=True)
|
|
source_document_name = Column(String(255), nullable=False)
|
|
generated_at = Column(DateTime, default=datetime.utcnow, nullable=False)
|
|
total_requirements = Column(Integer, nullable=False, default=0)
|
|
knowledge_base_id = Column(Integer, ForeignKey("knowledge_bases.id"), nullable=True, index=True)
|
|
generated_file = Column(JSON, nullable=False)
|
|
|
|
job = relationship("ToolJob", back_populates="testing_generation", foreign_keys=[job_id])
|
|
|
|
|
|
class CodeKnowledgeBase(Base, TimestampMixin):
|
|
__tablename__ = "code_knowledge_bases"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
|
|
name = Column(String(255), nullable=False)
|
|
project_path = Column(String(1024), nullable=True)
|
|
vector_path = Column(String(1024), nullable=False)
|
|
metadata_path = Column(String(1024), nullable=False)
|
|
graph_path = Column(String(1024), nullable=False)
|
|
status = Column(String(32), nullable=False, default="active", index=True)
|
|
metadata_summary = Column(JSON, nullable=True)
|
|
|
|
user = relationship("User")
|
|
consistency_jobs = relationship(
|
|
"ConsistencyJob",
|
|
back_populates="code_kb",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
|
|
|
|
class ConsistencyJob(Base, TimestampMixin):
|
|
__tablename__ = "consistency_jobs"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
|
|
srs_extraction_id = Column(Integer, ForeignKey("srs_extractions.id"), nullable=False, index=True)
|
|
code_kb_id = Column(Integer, ForeignKey("code_knowledge_bases.id"), nullable=False, index=True)
|
|
status = Column(String(32), nullable=False, default="pending", index=True)
|
|
total_requirements = Column(Integer, nullable=False, default=0)
|
|
completed_requirements = Column(Integer, nullable=False, default=0)
|
|
output_summary = Column(JSON, nullable=True)
|
|
error_message = Column(Text, nullable=True)
|
|
started_at = Column(DateTime, nullable=True)
|
|
completed_at = Column(DateTime, nullable=True)
|
|
|
|
user = relationship("User")
|
|
srs_extraction = relationship("SRSExtraction")
|
|
code_kb = relationship("CodeKnowledgeBase", back_populates="consistency_jobs")
|
|
results = relationship(
|
|
"ConsistencyResult",
|
|
back_populates="job",
|
|
cascade="all, delete-orphan",
|
|
order_by="ConsistencyResult.id",
|
|
)
|
|
|
|
|
|
class ConsistencyResult(Base, TimestampMixin):
|
|
__tablename__ = "consistency_results"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
job_id = Column(Integer, ForeignKey("consistency_jobs.id", ondelete="CASCADE"), nullable=False, index=True)
|
|
requirement_uid = Column(String(64), nullable=False, index=True)
|
|
verdict = Column(String(32), nullable=False, index=True)
|
|
coverage_score = Column(Float, nullable=False, default=0.0)
|
|
confidence = Column(Float, nullable=False, default=0.0)
|
|
matched_functions = Column(JSON, nullable=False)
|
|
covered_points = Column(JSON, nullable=False)
|
|
missing_points = Column(JSON, nullable=False)
|
|
conflict_points = Column(JSON, nullable=False)
|
|
call_chain_evidence = Column(JSON, nullable=False)
|
|
suggestion = Column(Text, nullable=True)
|
|
raw_judgment = Column(JSON, nullable=True)
|
|
|
|
job = relationship("ConsistencyJob", back_populates="results")
|
|
|