1.0

backend/app/models/moment.py

@@ -0,0 +1,61 @@
+"""朋友圈动态模型"""
+
+from datetime import datetime
+
+from sqlalchemy import String, Text, DateTime, ForeignKey, UniqueConstraint
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from app.database import Base
+
+
+class Moment(Base):
+    __tablename__ = "moments"
+
+    id: Mapped[str] = mapped_column(String(36), primary_key=True)
+    user_id: Mapped[str] = mapped_column(String(36), ForeignKey("users.id", ondelete="CASCADE"))
+    content: Mapped[str] = mapped_column(Text, nullable=False)
+    images: Mapped[str | None] = mapped_column(Text, nullable=True)  # JSON array of URLs
+    visibility: Mapped[str] = mapped_column(String(20), default="friends")  # public/friends/private
+    created_at: Mapped[datetime] = mapped_column(DateTime, default=lambda: datetime.utcnow())
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime,
+        default=lambda: datetime.utcnow(),
+        onupdate=lambda: datetime.utcnow(),
+    )
+
+    # 关系
+    user = relationship("User", foreign_keys=[user_id])
+    likes = relationship("MomentLike", back_populates="moment", cascade="all, delete-orphan")
+    comments = relationship("MomentComment", back_populates="moment", cascade="all, delete-orphan")
+
+
+class MomentLike(Base):
+    __tablename__ = "moment_likes"
+    __table_args__ = (
+        UniqueConstraint("moment_id", "user_id", name="uq_moment_like"),
+    )
+
+    id: Mapped[str] = mapped_column(String(36), primary_key=True)
+    moment_id: Mapped[str] = mapped_column(String(36), ForeignKey("moments.id", ondelete="CASCADE"))
+    user_id: Mapped[str] = mapped_column(String(36), ForeignKey("users.id", ondelete="CASCADE"))
+    created_at: Mapped[datetime] = mapped_column(DateTime, default=lambda: datetime.utcnow())
+
+    moment = relationship("Moment", back_populates="likes")
+    user = relationship("User", foreign_keys=[user_id])
+
+
+class MomentComment(Base):
+    __tablename__ = "moment_comments"
+
+    id: Mapped[str] = mapped_column(String(36), primary_key=True)
+    moment_id: Mapped[str] = mapped_column(String(36), ForeignKey("moments.id", ondelete="CASCADE"))
+    user_id: Mapped[str] = mapped_column(String(36), ForeignKey("users.id", ondelete="CASCADE"))
+    content: Mapped[str] = mapped_column(String(500), nullable=False)
+    reply_to_id: Mapped[str | None] = mapped_column(
+        String(36), ForeignKey("moment_comments.id", ondelete="SET NULL"), nullable=True
+    )
+    created_at: Mapped[datetime] = mapped_column(DateTime, default=lambda: datetime.utcnow())
+
+    moment = relationship("Moment", back_populates="comments")
+    user = relationship("User", foreign_keys=[user_id])
+    reply_to = relationship("MomentComment", remote_side=[id])