pagure/alembic/versions/32d636cb5e00_display_name_in_groups.py

"""display_name_in_groups

Revision ID: 32d636cb5e00
Revises: 43df5e588a87
Create Date: 2016-08-13 02:54:27.199948

"""

# revision identifiers, used by Alembic.
revision = '32d636cb5e00'
down_revision = '43df5e588a87'

from alembic import op
import sqlalchemy as sa


def upgrade():
    ''' Add attributes display name and description in PagureGroup '''
    op.add_column(
        'pagure_group',
        sa.Column(
            'display_name',
            sa.String(255),
            nullable=True,
            unique=True,
        )
    )

    op.execute('''UPDATE "pagure_group" SET display_name=group_name; ''')

    op.alter_column(
        'pagure_group',
        column_name='display_name',
        nullable=False,
        existing_nullable=True
        )

    op.add_column(
        'pagure_group',
        sa.Column(
            'description',
            sa.String(255),
            nullable=True,
        )
    )


def downgrade():
    ''' Remove attributes display name and description in PagureGroup '''

    op.drop_column('pagure_group', 'display_name')
    op.drop_column('pagure_group', 'description')