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

42 lines
995 B
Python

"""
Revision ID: 0336_broadcast_msg_content_2
Revises: 0335_broadcast_msg_content
Create Date: 2020-12-04 15:06:22.544803
"""
import sqlalchemy as sa
from alembic import op
from notifications_utils.template import BroadcastMessageTemplate
from sqlalchemy.dialects import postgresql
from sqlalchemy.orm.session import Session
revision = "0336_broadcast_msg_content_2"
down_revision = "0335_broadcast_msg_content"
def upgrade():
conn = op.get_bind()
results = conn.execute(
sa.text(
"""
UPDATE
broadcast_message
SET
content = templates_history.content
FROM
templates_history
WHERE
broadcast_message.content is NULL and
broadcast_message.template_id = templates_history.id and
broadcast_message.template_version = templates_history.version
;
"""
)
)
def downgrade():
op.alter_column("broadcast_message", "content", nullable=True)