Initial commit

api/models/web.py

@@ -0,0 +1,36 @@
+from sqlalchemy.dialects.postgresql import UUID
+
+from extensions.ext_database import db
+from models.model import Message
+
+
+class SavedMessage(db.Model):
+    __tablename__ = 'saved_messages'
+    __table_args__ = (
+        db.PrimaryKeyConstraint('id', name='saved_message_pkey'),
+        db.Index('saved_message_message_idx', 'app_id', 'message_id', 'created_by'),
+    )
+
+    id = db.Column(UUID, server_default=db.text('uuid_generate_v4()'))
+    app_id = db.Column(UUID, nullable=False)
+    message_id = db.Column(UUID, nullable=False)
+    created_by = db.Column(UUID, nullable=False)
+    created_at = db.Column(db.DateTime, nullable=False, server_default=db.text('CURRENT_TIMESTAMP(0)'))
+
+    @property
+    def message(self):
+        return db.session.query(Message).filter(Message.id == self.message_id).first()
+
+
+class PinnedConversation(db.Model):
+    __tablename__ = 'pinned_conversations'
+    __table_args__ = (
+        db.PrimaryKeyConstraint('id', name='pinned_conversation_pkey'),
+        db.Index('pinned_conversation_conversation_idx', 'app_id', 'conversation_id', 'created_by'),
+    )
+
+    id = db.Column(UUID, server_default=db.text('uuid_generate_v4()'))
+    app_id = db.Column(UUID, nullable=False)
+    conversation_id = db.Column(UUID, nullable=False)
+    created_by = db.Column(UUID, nullable=False)
+    created_at = db.Column(db.DateTime, nullable=False, server_default=db.text('CURRENT_TIMESTAMP(0)'))