notifications-api/migrations/versions/0289_precompiled_for_all.py

"""

Revision ID: 0289_precompiled_for_all
Revises: 0288_add_go_live_user
Create Date: 2019-05-13 10:44:51.867661

"""

from alembic import op
from sqlalchemy import text

revision = "0289_precompiled_for_all"
down_revision = "0288_add_go_live_user"


def upgrade():
    op.execute(
        "DELETE from service_permissions where permission = 'precompiled_letter'"
    )
    op.execute("DELETE from service_permission_types where name = 'precompiled_letter'")


def downgrade():
    conn = op.get_bind()
    op.execute("INSERT INTO service_permission_types values('precompiled_letter')")
    input_params = {"permission": "precompiled_letter"}
    conn.execute(
        text(
            """
           INSERT INTO
               service_permissions (service_id, permission, created_at)
           SELECT
               id, :permission, now()
           FROM
               services
           WHERE
               NOT EXISTS (
                   SELECT
                   FROM
                       service_permissions
                   WHERE
                       service_id = services.id and
                       permission = :permission
               )
       """
        ),
        input_params,
    )