Underground-Digital/Workflow-Engine

1from extensions.ext_database import db2 3from .model import Message4from .types import StringUUID5 6 7class SavedMessage(db.Model):8    __tablename__ = "saved_messages"9    __table_args__ = (10        db.PrimaryKeyConstraint("id", name="saved_message_pkey"),11        db.Index("saved_message_message_idx", "app_id", "message_id", "created_by_role", "created_by"),12    )13 14    id = db.Column(StringUUID, server_default=db.text("uuid_generate_v4()"))15    app_id = db.Column(StringUUID, nullable=False)16    message_id = db.Column(StringUUID, nullable=False)17    created_by_role = db.Column(db.String(255), nullable=False, server_default=db.text("'end_user'::character varying"))18    created_by = db.Column(StringUUID, nullable=False)19    created_at = db.Column(db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)"))20 21    @property22    def message(self):23        return db.session.query(Message).filter(Message.id == self.message_id).first()24 25 26class PinnedConversation(db.Model):27    __tablename__ = "pinned_conversations"28    __table_args__ = (29        db.PrimaryKeyConstraint("id", name="pinned_conversation_pkey"),30        db.Index("pinned_conversation_conversation_idx", "app_id", "conversation_id", "created_by_role", "created_by"),31    )32 33    id = db.Column(StringUUID, server_default=db.text("uuid_generate_v4()"))34    app_id = db.Column(StringUUID, nullable=False)35    conversation_id = db.Column(StringUUID, nullable=False)36    created_by_role = db.Column(db.String(255), nullable=False, server_default=db.text("'end_user'::character varying"))37    created_by = db.Column(StringUUID, nullable=False)38    created_at = db.Column(db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)"))39