From 6e1c5a70c0f756eca93cf9fa917557c377152af7 Mon Sep 17 00:00:00 2001 From: Pea Tyczynska Date: Mon, 8 Feb 2021 18:22:25 +0000 Subject: [PATCH] Make stubbed column on broadcast_message non-nullable But first set any null values there to False. --- app/models.py | 2 +- .../versions/0344_stubbed_not_nullable.py | 36 +++++++++++++++++++ 2 files changed, 37 insertions(+), 1 deletion(-) create mode 100644 migrations/versions/0344_stubbed_not_nullable.py diff --git a/app/models.py b/app/models.py index e1e69dc22..f31a51765 100644 --- a/app/models.py +++ b/app/models.py @@ -2281,7 +2281,7 @@ class BroadcastMessage(db.Model): reference = db.Column(db.String(255), nullable=True) - stubbed = db.Column(db.Boolean, nullable=True) + stubbed = db.Column(db.Boolean, nullable=False) CheckConstraint("created_by_id is not null or api_key_id is not null") diff --git a/migrations/versions/0344_stubbed_not_nullable.py b/migrations/versions/0344_stubbed_not_nullable.py new file mode 100644 index 000000000..eb5e87028 --- /dev/null +++ b/migrations/versions/0344_stubbed_not_nullable.py @@ -0,0 +1,36 @@ +""" + +Revision ID: 0344_stubbed_not_nullable +Revises: 0343_org_billing_details +Create Date: 2021-02-08 18:10:15.533279 + +""" +from alembic import op +import sqlalchemy as sa + + +revision = '0344_stubbed_not_nullable' +down_revision = '0343_org_billing_details' + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.execute("UPDATE broadcast_message SET stubbed = False WHERE stubbed is null") + op.alter_column( + 'broadcast_message', + 'stubbed', + existing_type=sa.BOOLEAN(), + nullable=False + ) + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.alter_column( + 'broadcast_message', + 'stubbed', + existing_type=sa.BOOLEAN(), + nullable=True + ) + # ### end Alembic commands ###