aidress/app/models/dress.py
2025-03-21 17:06:54 +08:00

27 lines
1.2 KiB
Python

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}')>"