Blame alembic/versions/6190226bed0_add_the_updated_on_column_to_pull_.py

Pierre-Yves Chibon bcd466
"""Add the updated_on column to pull-requests
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
Revision ID: 6190226bed0
Pierre-Yves Chibon bcd466
Revises: 257a7ce22682
Pierre-Yves Chibon bcd466
Create Date: 2015-09-29 15:32:58.229183
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
"""
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
# revision identifiers, used by Alembic.
Pierre-Yves Chibon bcd466
revision = '6190226bed0'
Pierre-Yves Chibon bcd466
down_revision = '257a7ce22682'
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
from alembic import op
Pierre-Yves Chibon bcd466
import sqlalchemy as sa
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
def upgrade():
Pierre-Yves Chibon bcd466
    ''' Add the column updated_on to the table pull_requests.
Pierre-Yves Chibon bcd466
    '''
Pierre-Yves Chibon bcd466
    op.add_column(
Pierre-Yves Chibon bcd466
        'pull_requests',
Pierre-Yves Chibon bcd466
        sa.Column(
Pierre-Yves Chibon bcd466
            'updated_on',
Pierre-Yves Chibon bcd466
            sa.DateTime,
Pierre-Yves Chibon bcd466
            nullable=True,
Pierre-Yves Chibon bcd466
            default=sa.func.now(),
Pierre-Yves Chibon bcd466
            onupdate=sa.func.now()
Pierre-Yves Chibon bcd466
        )
Pierre-Yves Chibon bcd466
    )
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
    op.execute('''UPDATE "pull_requests" SET updated_on=date_created;''')
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
    op.alter_column(
Pierre-Yves Chibon bcd466
        'pull_requests',
Pierre-Yves Chibon bcd466
        column_name='updated_on',
Pierre-Yves Chibon bcd466
        nullable=False,
Pierre-Yves Chibon bcd466
        existing_nullable=True)
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
Pierre-Yves Chibon bcd466
def downgrade():
Pierre-Yves Chibon bcd466
    ''' Remove the column updated_on from the table pull_requests.
Pierre-Yves Chibon bcd466
    '''
Pierre-Yves Chibon bcd466
    op.drop_column('pull_requests', 'updated_on')