notifications-api/migrations/versions/0283_platform_admin_not_live.py

"""empty message

Revision ID: 0283_platform_admin_not_live
Revises: 0282_add_count_as_live
Create Date: 2016-10-25 17:37:27.660723

"""

# revision identifiers, used by Alembic.
from sqlalchemy import text

revision = "0283_platform_admin_not_live"
down_revision = "0282_add_count_as_live"

import sqlalchemy as sa
from alembic import op

STATEMENT = """
    UPDATE
        services
    SET
        count_as_live = :count_as_live
    FROM
        users
    WHERE
        services.created_by_id = users.id and
        users.platform_admin is true
    ;
"""


def upgrade():
    conn = op.get_bind()
    input_params = {"count_as_live": "false"}
    conn.execute(text(STATEMENT), input_params)


def downgrade():
    conn = op.get_bind()
    input_params = {"count_as_live": "true"}
    conn.execute(text(STATEMENT), input_params)