Files
notifications-api/migrations/versions/0337_broadcast_msg_api.py
Cliff Hill 1157f5639d black, isort, flake8
Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
2023-12-08 21:43:52 -05:00

32 lines
912 B
Python

"""
Revision ID: 0337_broadcast_msg_api
Revises: 0336_broadcast_msg_content_2
Create Date: 2020-12-04 15:06:22.544803
"""
import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql
revision = "0337_broadcast_msg_api"
down_revision = "0336_broadcast_msg_content_2"
def upgrade():
op.alter_column("broadcast_message", "created_by_id", nullable=True)
op.add_column(
"broadcast_message", sa.Column("api_key_id", postgresql.UUID(), nullable=True)
)
op.create_foreign_key(None, "broadcast_message", "api_keys", ["api_key_id"], ["id"])
op.add_column(
"broadcast_message",
sa.Column("reference", sa.String(length=255), nullable=True),
)
def downgrade():
op.alter_column("broadcast_message", "created_by_id", nullable=False)
op.drop_column("broadcast_message", "api_key_id")
op.add_column("broadcast_message", "reference")