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]