notifications-api/migrations/versions/0336_broadcast_msg_content_2.py

"""

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 sqlalchemy.dialects import postgresql
from sqlalchemy.orm.session import Session

from notifications_utils.template import BroadcastMessageTemplate

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)