pagure/alembic/versions/9cb4580e269a_add_date_updated_to_commit_flags.py

"""Add date_updated to commit flags

Revision ID: 9cb4580e269a
Revises: e3cc5aedb8bb
Create Date: 2018-11-14 11:48:47.436459

"""

import datetime

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '9cb4580e269a'
down_revision = 'e3cc5aedb8bb'


def upgrade():
    """ Add date_updated column to commit_flags table """
    op.add_column(
        'commit_flags',
        sa.Column(
            'date_updated',
            sa.DateTime,
            nullable=True,
            default=datetime.datetime.utcnow,
        )
    )
    op.execute('UPDATE commit_flags SET date_updated=date_created')
    op.alter_column(
        'commit_flags', 'date_updated', existing_type=sa.DateTime,
        nullable=False, existing_nullable=True)


def downgrade():
    """ Drop the date_updated column from the commit_flags table """
    op.drop_column('commit_flags', 'date_updated')