pagure/alembic/versions/644ef887bb6f_add_close_status.py

"""Add close status

Revision ID: 644ef887bb6f
Revises: 368fd931cf7f
Create Date: 2016-10-04 15:38:41.908679

"""

# revision identifiers, used by Alembic.
revision = '644ef887bb6f'
down_revision = '368fd931cf7f'

from alembic import op
import sqlalchemy as sa


try:
    from pagure.lib import model
except ImportError:
    import sys
    sys.path.insert(0, '.')
    from pagure.lib import model


def upgrade():
    ''' Add the column _close_status to the table projects.
    '''
    op.add_column(
        'projects',
        sa.Column('_close_status', sa.Text, nullable=True)
    )
    op.add_column(
        'issues',
        sa.Column('close_status', sa.Text, nullable=True)
    )

    engine = op.get_bind()
    Session = sa.orm.scoped_session(sa.orm.sessionmaker())
    Session.configure(bind=engine)
    session = Session()

    # Update all the existing projects
    statuses = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate']
    for project in session.query(model.Project).all():
        project.close_status = statuses
        session.add(project)
    session.commit()

    # Add the status 'Closed' for issues
    ticket_stat = model.StatusIssue(status='Closed')
    session.add(ticket_stat)
    session.commit()

    # Set the close_status for all the closed tickets
    op.execute('''UPDATE "issues" SET "close_status"=status where status != 'Open'; ''')

    # Mark all the tickets as closed
    op.execute('''UPDATE "issues" SET status='Closed' where status != 'Open';  ''')

    # Remove the old status
    op.execute('''DELETE FROM "status_issue" WHERE "status" NOT IN ('Open', 'Closed'); ''')


def downgrade():
    ''' Add the column _close_status to the table projects.
    '''
    engine = op.get_bind()
    Session = sa.orm.scoped_session(sa.orm.sessionmaker())
    Session.configure(bind=engine)
    session = Session()

    statuses = ['Invalid', 'Insufficient data', 'Fixed', 'Duplicate']
    for status in statuses:
        ticket_stat = model.StatusIssue(status=status)
        session.add(ticket_stat)
        session.commit()

    # Set the close_status for all the closed tickets
    op.execute('''UPDATE "issues" SET status=close_status where status != 'Open'; ''')

    # Remove the old status
    op.execute('''DELETE FROM "status_issue" WHERE status = 'Closed'; ''')

    op.drop_column('projects', '_close_status')
    op.drop_column('issues', 'close_status')