mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-11 07:42:20 -05:00
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
"""
|
|
Revision ID: 0363_cancelled_by_api_key
|
|
Revises: 0362_broadcast_msg_event
|
|
Create Date: 2022-02-09 14:05:27.750234
|
|
"""
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
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 ###
|