Files
notifications-api/migrations/versions/0336_broadcast_msg_content_2.py
2024-05-23 10:18:02 -07:00

44 lines
997 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 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)