diff --git a/models/message.py b/models/message.py new file mode 100644 index 0000000..5f4a28d --- /dev/null +++ b/models/message.py @@ -0,0 +1,20 @@ +from sqlalchemy import Column, String, DateTime, ForeignKey +from sqlalchemy.orm import relationship +from sqlalchemy.sql import func +from core.database import Base +import uuid + +class Message(Base): + __tablename__ = "messages" + + id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4())) + sender_id = Column(String, ForeignKey("users.id"), nullable=False, index=True) + recipient_id = Column(String, ForeignKey("users.id"), nullable=False, index=True) + content = Column(String, nullable=False) + is_read = Column(Boolean, default=False) + + created_at = Column(DateTime, default=func.now()) + updated_at = Column(DateTime, default=func.now(), onupdate=func.now()) + + sender = relationship("User", foreign_keys=[sender_id], back_populates="sent_messages") + recipient = relationship("User", foreign_keys=[recipient_id], back_populates="received_messages") \ No newline at end of file