mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-12 00:02:36 -05:00
c55d917b28
This column has been superseded by a new column named created_by_api_key_id. Also create constraint checking that we know who created broadcast Also move data so that constraint is met before instatiating it.
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
"""
|
|
|
|
Revision ID: 0364_drop_old_column
|
|
Revises: 0363_cancelled_by_api_key
|
|
Create Date: 2022-01-25 18:05:27.750234
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision = '0364_drop_old_column'
|
|
down_revision = '0363_cancelled_by_api_key'
|
|
|
|
|
|
def upgrade():
|
|
# move data over
|
|
op.execute("UPDATE broadcast_message SET created_by_api_key_id=api_key_id WHERE created_by_api_key_id IS NULL")
|
|
op.create_check_constraint(
|
|
"ck_broadcast_message_created_by_not_null",
|
|
"broadcast_message",
|
|
"created_by_id is not null or created_by_api_key_id is not null"
|
|
)
|
|
op.drop_column('broadcast_message', 'api_key_id')
|
|
|
|
|
|
def downgrade():
|
|
op.add_column('broadcast_message', sa.Column('api_key_id', postgresql.UUID(), autoincrement=False, nullable=True))
|
|
op.execute("UPDATE broadcast_message SET api_key_id=created_by_api_key_id") # move data over
|
|
op.drop_constraint(
|
|
"ck_broadcast_message_created_by_not_null",
|
|
"broadcast_message"
|
|
)
|