notifications-api/migrations/versions/0149_add_crown_to_services.py

"""

Revision ID: 0149_add_crown_to_services
Revises: 0148_add_letters_as_pdf_svc_perm
Create Date: 2017-12-04 12:13:35.268712

"""
import sqlalchemy as sa
from alembic import op

revision = "0149_add_crown_to_services"
down_revision = "0148_add_letters_as_pdf_svc_perm"


def upgrade():
    op.add_column("services", sa.Column("crown", sa.Boolean(), nullable=True))
    op.execute(
        """
        update services set crown = True
        where organisation_type = 'central'
    """
    )
    op.execute(
        """
        update services set crown = True
        where organisation_type is null
    """
    )
    op.execute(
        """
        update services set crown = False
        where crown is null
    """
    )
    op.alter_column("services", "crown", nullable=False)

    op.add_column("services_history", sa.Column("crown", sa.Boolean(), nullable=True))
    op.execute(
        """
        update services_history set crown = True
        where organisation_type = 'central'
    """
    )
    op.execute(
        """
        update services_history set crown = True
        where organisation_type is null
    """
    )
    op.execute(
        """
        update services_history set crown = False
        where crown is null
    """
    )
    op.alter_column("services_history", "crown", nullable=False)


def downgrade():
    op.drop_column("services", "crown")
    op.drop_column("services_history", "crown")