init

src/models/database.py

@@ -115,7 +115,7 @@ class RolePermission(Base):
         return f"<RolePermission(role_id={self.role_id}, permission_id={self.permission_id})>"
 
 class STPFile(Base):
-    """STP源文件元数据表"""
+    """STP源文件元数据表 - 支持同一文件多次上传"""
     __tablename__ = "stp_files"
     
     id = Column(Integer, primary_key=True, index=True)
@@ -127,19 +127,27 @@ class STPFile(Base):
     object_url = Column(String(1000), nullable=True)          # 预签名URL（可选）
     
     # 文件信息
-    original_filename = Column(String(255), nullable=False)
+    original_filename = Column(String(255), nullable=False, index=True)  # 添加索引支持按文件名查询
     file_size = Column(Integer, nullable=False)
-    file_hash = Column(String(64), unique=True, index=True)  # SHA256 hash
+    file_hash = Column(String(64), index=True)  # 移除unique约束，允许同一文件多次上传
     mime_type = Column(String(50), default="application/octet-stream")
     
+    # 上传批次标识 - 用于区分同一文件的多次上传
+    upload_batch = Column(String(36), index=True)  # UUID批次号
+    
     # 时间戳
     upload_time = Column(DateTime, default=func.now())
     processed_time = Column(DateTime, nullable=True)
     
     # 状态
-    status = Column(String(20), default="pending")  # pending, processing, completed, failed
+    status = Column(String(20), default="pending", index=True)  # pending, processing, completed, failed
     error_message = Column(Text, nullable=True)
     
+    # 分析摘要 - 快速查询字段
+    volume = Column(Float, nullable=True)  # 体积 mm³
+    surface_area = Column(Float, nullable=True)  # 表面积 mm²
+    product_weight = Column(Float, nullable=True)  # 产品重量 g
+    
     # 保留旧字段以兼容
     file_path = Column(String(500), nullable=True)  # 本地路径（已弃用）
     file_content = Column(LargeBinary, nullable=True)  # 本地存储（已弃用）
@@ -148,14 +156,15 @@ class STPFile(Base):
     # 关联关系
     user = relationship("User", back_populates="stp_files")
     geometry_data = relationship("GeometryData", back_populates="stp_file", uselist=False)
-    # 网格数据：一条 STP 记录对应一条网格摘要记录（详细 JSON 在 RustFS）
     mesh_data = relationship("MeshData", back_populates="stp_file", uselist=False)
     mold_cavity_data = relationship("MoldCavityData", back_populates="stp_file", uselist=False)
     html_file = relationship("HTMLFile", back_populates="stp_file", uselist=False)
     analysis_metrics = relationship("AnalysisMetrics", back_populates="stp_file", uselist=False)
+    feature_detections = relationship("FeatureDetection", back_populates="stp_file")
+    design_recommendations = relationship("DesignRecommendation", back_populates="stp_file")
     
     def __repr__(self):
-        return f"<STPFile(id={self.id}, object_key='{self.object_key}', status='{self.status}')>"
+        return f"<STPFile(id={self.id}, original_filename='{self.original_filename}', status='{self.status}')>"
 
 class GeometryData(Base):
     """几何数据JSON元数据表"""
@@ -360,6 +369,9 @@ class FeatureDetection(Base):
     # 关联的几何数据
     geometry_data_id = Column(Integer, ForeignKey("geometry_data.id"), nullable=True)
     
+    # 关联关系
+    stp_file = relationship("STPFile", back_populates="feature_detections")
+    
     def __repr__(self):
         return f"<FeatureDetection(id={self.id}, feature_type='{self.feature_type}', confidence={self.confidence})>"
 
@@ -388,6 +400,9 @@ class DesignRecommendation(Base):
     created_at = Column(DateTime, default=func.now())
     updated_at = Column(DateTime, nullable=True)
     
+    # 关联关系
+    stp_file = relationship("STPFile", back_populates="design_recommendations")
+    
     def __repr__(self):
         return f"<DesignRecommendation(id={self.id}, rec_type='{self.rec_type}', priority='{self.priority}')>"