from sqlalchemy import Column, Integer, String, DateTime, Text, func, Enum
from sqlalchemy.ext.declarative import declarative_base
from app.database import Base
import datetime
import enum

class GarmentType(enum.Enum):
    """服装类型枚举"""
    TOP_GARMENT = "TOP_GARMENT"
    BOTTOM_GARMENT = "BOTTOM_GARMENT"

class Dress(Base):
    """服装模型类"""
    __tablename__ = "dresses"
    
    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
    name = Column(String(255), nullable=False, comment="服装名称")
    image_url = Column(String(1024), nullable=True, comment="服装图片URL")
    garment_type = Column(Enum(GarmentType), nullable=True, comment="服装类型(上衣/下衣)")
    
    # 以下是一些可选的附加字段，根据需要可以去掉或添加
    description = Column(Text, nullable=True, comment="服装描述")
    created_at = Column(DateTime, default=datetime.datetime.now(), comment="创建时间")
    updated_at = Column(DateTime, default=datetime.datetime.now(), onupdate=datetime.datetime.now, comment="更新时间")
    
    def __repr__(self):
        return f"<Dress(id={self.id}, name='{self.name}')>"