From 85ab4401c798b460dc348c92e98bf3f84df0c7a4 Mon Sep 17 00:00:00 2001 From: Izan Date: Wed, 10 Nov 2021 22:44:54 +0000 Subject: [PATCH] Add `dm_sent` field to infractions model --- ..._add_dm_sent_field_to_infractions_model.py | 26 +++++++++++++++++++ .../database/models/api/bot/infraction.py | 3 +++ 2 files changed, 29 insertions(+) create mode 100644 alembic/versions/3d9c7f4c3e4c_add_dm_sent_field_to_infractions_model.py diff --git a/alembic/versions/3d9c7f4c3e4c_add_dm_sent_field_to_infractions_model.py b/alembic/versions/3d9c7f4c3e4c_add_dm_sent_field_to_infractions_model.py new file mode 100644 index 0000000..4bb0344 --- /dev/null +++ b/alembic/versions/3d9c7f4c3e4c_add_dm_sent_field_to_infractions_model.py @@ -0,0 +1,26 @@ +""" +Add DM Sent field to infractions model. + +Revision ID: 3d9c7f4c3e4c +Revises: c551b4b47800 +Create Date: 2021-11-10 22:26:00.101691 +""" +import sqlalchemy as sa +from alembic import op + + +# revision identifiers, used by Alembic. +revision = '3d9c7f4c3e4c' +down_revision = 'c551b4b47800' +branch_labels = None +depends_on = None + + +def upgrade() -> None: + """Commands auto generated by Alembic.""" + op.add_column('api_infraction', sa.Column('dm_sent', sa.Boolean(), nullable=True)) + + +def downgrade() -> None: + """Commands auto generated by Alembic.""" + op.drop_column('api_infraction', 'dm_sent') diff --git a/api/core/database/models/api/bot/infraction.py b/api/core/database/models/api/bot/infraction.py index bec20c8..b24ff51 100644 --- a/api/core/database/models/api/bot/infraction.py +++ b/api/core/database/models/api/bot/infraction.py @@ -48,6 +48,9 @@ class Infraction(Base): # Whether the infraction is a shadow infraction. hidden = Column(Boolean, nullable=False) + # Whether a DM was sent to the user when infraction was applied. + dm_sent = Column(Boolean, nullable=True) + actor_id = Column( ForeignKey( "api_user.id", deferrable=True, initially="DEFERRED", ondelete="CASCADE"