"""Adjust constraints on pull_request_flags

Revision ID: 01e58ee9eccb

Revises: 6119fbbcc8e9

Create Date: 2017-11-16 16:50:47.278252

"""

from alembic import op

import sqlalchemy as sa

# revision identifiers, used by Alembic.

revision = '01e58ee9eccb'

down_revision = '6119fbbcc8e9'

def upgrade():

    """ Remove the unique constraints on UID in pull_request_flags and make

    it a composite unique constraint on UID + pull_request_uid.

    """

    # alter the constraints

    op.drop_constraint('pull_request_flags_uid_key', 'pull_request_flags')

    op.create_unique_constraint(

            "pull_request_flags_uid_pull_request_uid_key",

            'pull_request_flags',

            ["uid", "pull_request_uid"]

    )

def downgrade():

    """ Remove the composite unique constraints on UID + pull_request_uid

    in pull_request_flags and make it an unique constraint on UID .

    """

    op.drop_constraint(

        'pull_request_flags_uid_pull_request_uid_key',

        'pull_request_flags')

    op.create_unique_constraint(

            "pull_request_flags_uid_key",

            'pull_request_flags',

            ["uid"]

    )