Save message IDs

3 files changed, 3 insertions, 0 deletions

diff --git a/models.py b/models.py
index 3e60640..b100703 100644
--- a/models.py
+++ b/models.py
@@ -8,6 +8,7 @@ class Message(Base):
 
     id = Column(Integer, primary_key=True, autoincrement=True)
     timestamp = Column(String, nullable=False)
+    message_id = Column(Integer, nullable=False)
     author_id = Column(Integer, nullable=False)
     channel_id = Column(Integer, nullable=False)
     stickers = Column(JSON, nullable=False, default=list)
diff --git a/src/cogs/archive.py b/src/cogs/archive.py
index 5e80709..51adbdf 100644
--- a/src/cogs/archive.py
+++ b/src/cogs/archive.py
@@ -105,6 +105,7 @@ class Archive(commands.Cog):
 
             db_message = Message(
                 timestamp=message.created_at.isoformat(),
+                message_id=message.id,
                 author_id=message.author.id,
                 channel_id=channel.id,
                 stickers=[sticker.name for sticker in message.stickers],
diff --git a/src/schemas/message_schema.py b/src/schemas/message_schema.py
index 1738f3b..e135a2e 100644
--- a/src/schemas/message_schema.py
+++ b/src/schemas/message_schema.py
@@ -3,6 +3,7 @@ from pydantic import BaseModel
 
 class Message(BaseModel):
     timestamp: str
+    message_id: int
     author_id: int
     channel_id: int
     stickers: list[str]