"""Add an origin field on pr_to_issue

Revision ID: 003fcd9e8860

Revises: 2b1743f77436

Create Date: 2019-03-12 14:55:59.316861

"""

from alembic import op

import sqlalchemy as sa

# revision identifiers, used by Alembic.

revision = '003fcd9e8860'

down_revision = '2b1743f77436'

def upgrade():

    ''' Add a column to record if the custom field should trigger a email

    notification.

    '''

    op.add_column(

        'pr_to_issue',

        sa.Column(

            'origin', sa.String(32), index=True

        )

    )

def downgrade():

    ''' Remove the key_notify column.

    '''

    op.drop_column('pr_to_issue', 'origin')