68 lines
2.7 KiB
Python
68 lines
2.7 KiB
Python
|
import uuid
|
|||
|
|
from sqlalchemy import Column, String, Integer, DateTime, ForeignKey, Text, Float
|
|||
|
|
from sqlalchemy.sql import func
|
|||
|
|
from sqlalchemy.orm import relationship
|
|||
|
|
from app.db.base import Base
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AuditTask(Base):
|
|||
|
|
__tablename__ = "audit_tasks"
|
|||
|
|
|
|||
|
|
id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
|
|||
|
|
project_id = Column(String, ForeignKey("projects.id"), nullable=False)
|
|||
|
|
created_by = Column(String, ForeignKey("users.id"), nullable=False)
|
|||
|
|
|
|||
|
|
task_type = Column(String, nullable=False)
|
|||
|
|
status = Column(String, default="pending", index=True)
|
|||
|
|
branch_name = Column(String, nullable=True)
|
|||
|
|
|
|||
|
|
exclude_patterns = Column(Text, default="[]")
|
|||
|
|
scan_config = Column(Text, default="{}")
|
|||
|
|
|
|||
|
|
# Stats
|
|||
|
|
total_files = Column(Integer, default=0)
|
|||
|
|
scanned_files = Column(Integer, default=0)
|
|||
|
|
total_lines = Column(Integer, default=0)
|
|||
|
|
issues_count = Column(Integer, default=0)
|
|||
|
|
quality_score = Column(Float, default=0.0)
|
|||
|
|
|
|||
|
|
started_at = Column(DateTime(timezone=True), nullable=True)
|
|||
|
|
completed_at = Column(DateTime(timezone=True), nullable=True)
|
|||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|||
|
|
|
|||
|
|
# Relationships
|
|||
|
|
project = relationship("Project", back_populates="tasks")
|
|||
|
|
creator = relationship("User", foreign_keys=[created_by])
|
|||
|
|
issues = relationship("AuditIssue", back_populates="task", cascade="all, delete-orphan")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AuditIssue(Base):
|
|||
|
|
__tablename__ = "audit_issues"
|
|||
|
|
|
|||
|
|
id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
|
|||
|
|
task_id = Column(String, ForeignKey("audit_tasks.id"), nullable=False)
|
|||
|
|
|
|||
|
|
file_path = Column(String, nullable=False)
|
|||
|
|
line_number = Column(Integer, nullable=True)
|
|||
|
|
column_number = Column(Integer, nullable=True)
|
|||
|
|
issue_type = Column(String, nullable=False)
|
|||
|
|
severity = Column(String, nullable=False) # critical, high, medium, low
|
|||
|
|
|
|||
|
|
# 问题信息
|
|||
|
|
title = Column(String, nullable=True) # 问题标题
|
|||
|
|
message = Column(Text, nullable=True) # 兼容旧字段，同title
|
|||
|
|
description = Column(Text, nullable=True) # 详细描述
|
|||
|
|
suggestion = Column(Text, nullable=True) # 修复建议
|
|||
|
|
code_snippet = Column(Text, nullable=True) # 问题代码片段
|
|||
|
|
ai_explanation = Column(Text, nullable=True) # AI解释（JSON格式的xai字段）
|
|||
|
|
|
|||
|
|
status = Column(String, default="open") # open, resolved, false_positive
|
|||
|
|
resolved_by = Column(String, ForeignKey("users.id"), nullable=True)
|
|||
|
|
resolved_at = Column(DateTime(timezone=True), nullable=True)
|
|||
|
|
|
|||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|||
|
|
|
|||
|
|
# Relationships
|
|||
|
|
task = relationship("AuditTask", back_populates="issues")
|
|||
|
|
resolver = relationship("User", foreign_keys=[resolved_by])
|