darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-08 22:22:24 -05:00
Code Issues Packages Projects Releases Wiki Activity

Migration adding cancelled_by_api_key_id to broadcast_message

Browse Source 
table. Also add created_by_api_key_id column that will supersede
api_key_id_column.

We do migration and code in separate PRs to make it easier to
downgrade if anything goes wrong.
This commit is contained in:
Pea Tyczynska
2022-02-09 14:30:25 +00:00
parent 9ce6d2fe92
commit 00b63ca5c7
1 changed files with 45 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
migrations/versions/0363_cancelled_by_api_key.py Normal file
View File

@@ -0,0 +1,45 @@
"""
Revision ID: 0363_cancelled_by_api_key
Revises: 0362_broadcast_msg_event
Create Date: 2022-02-09 14:05:27.750234
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
revision = '0363_cancelled_by_api_key'
down_revision = '0362_broadcast_msg_event'
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.add_column('broadcast_message', sa.Column('created_by_api_key_id', postgresql.UUID(as_uuid=True), nullable=True))
op.add_column(
'broadcast_message', sa.Column('cancelled_by_api_key_id', postgresql.UUID(as_uuid=True), nullable=True)
)
op.drop_constraint('broadcast_message_api_key_id_fkey', 'broadcast_message', type_='foreignkey')
op.create_foreign_key(
'broadcast_message_created_by_api_key_id_fkey',
'broadcast_message',
'api_keys',
['created_by_api_key_id'],
['id']
)
op.create_foreign_key(
'broadcast_message_cancelled_by_api_key_id_fkey',
'broadcast_message',
'api_keys',
['cancelled_by_api_key_id'],
['id']
)
# ### end Alembic commands ###
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_constraint('broadcast_message_created_by_api_key_id_fkey', 'broadcast_message', type_='foreignkey')
op.drop_constraint('broadcast_message_cancelled_by_api_key_id_fkey', 'broadcast_message', type_='foreignkey')
op.create_foreign_key('broadcast_message_api_key_id_fkey', 'broadcast_message', 'api_keys', ['api_key_id'], ['id'])
op.drop_column('broadcast_message', 'cancelled_by_api_key_id')
op.drop_column('broadcast_message', 'created_by_api_key_id')
# ### end Alembic commands ###