62 lines
2.6 KiB
Python
62 lines
2.6 KiB
Python
"""朋友圈动态模型"""
|
|
|
|
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])
|