diff --git a/migrations/versions/0363_cancelled_by_api_key.py b/migrations/versions/0363_cancelled_by_api_key.py new file mode 100644 index 000000000..20856a287 --- /dev/null +++ b/migrations/versions/0363_cancelled_by_api_key.py @@ -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 ###