"""
设计图片模型
存储每个设计的多视角图片
"""
from sqlalchemy import Column, BigInteger, Integer, String, Text, DateTime, ForeignKey
from sqlalchemy.sql import func
from sqlalchemy.orm import relationship

from ..database import Base


class DesignImage(Base):
    """设计图片表 - 存储多视角设计图"""
    __tablename__ = "design_images"

    id = Column(BigInteger, primary_key=True, autoincrement=True, comment="图片ID")
    design_id = Column(BigInteger, ForeignKey("designs.id", ondelete="CASCADE"), nullable=False, comment="关联设计ID")
    view_name = Column(String(20), nullable=False, comment="视角名称: 效果图/正面图/侧面图/背面图")
    image_url = Column(Text, nullable=True, comment="图片URL路径")
    model_used = Column(String(50), nullable=True, comment="使用的AI模型: flux-dev/seedream-4.5")
    prompt_used = Column(Text, nullable=True, comment="实际使用的英文prompt")
    sort_order = Column(Integer, default=0, comment="排序")
    model_3d_url = Column(Text, nullable=True, comment="3D模型URL(.glb)")
    created_at = Column(DateTime, server_default=func.now(), comment="创建时间")

    # 关联关系
    design = relationship("Design", back_populates="images")

    def __repr__(self):
        return f"<DesignImage(id={self.id}, view='{self.view_name}')>"