"""Adding column to store edited_by and edited_on a PR comment

Revision ID: 15ea3c2cf83d

Revises: 1cd0a853c697

Create Date: 2015-11-09 16:18:47.192088

"""

# revision identifiers, used by Alembic.

revision = '15ea3c2cf83d'

down_revision = '1cd0a853c697'

from alembic import op

import sqlalchemy as sa

def upgrade():

    ''' Add the columns editor_id and edited_on to the table

    pull_request_comments.

    '''

    op.add_column(

        'pull_request_comments',

        sa.Column(

            'editor_id',

            sa.Integer,

            sa.ForeignKey('users.id', onupdate='CASCADE'),

            nullable=True)

    )

    op.add_column(

        'pull_request_comments',

        sa.Column(

            'edited_on',

            sa.DateTime,

            nullable=True)

    )

def downgrade():

    ''' Remove the columns editor_id and edited_on from the table

    pull_request_comments.

    '''

    op.drop_column('pull_request_comments', 'editor_id')

    op.drop_column('pull_request_comments', 'edited_on')