"""Add namespace to project
 
Revision ID: 350efb3f6baf
Revises: 1640c7d75e5f
Create Date: 2016-08-30 22:02:07.645138
 
"""
 
# revision identifiers, used by Alembic.
revision = '350efb3f6baf'
down_revision = '1640c7d75e5f'
 
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 namespace to the table projects.
    '''
    op.add_column(
        'projects',
        sa.Column('namespace', sa.String(255), nullable=True, index=True)
    )
 
    # Update all the existing projects
    engine = op.get_bind()
    Session = sa.orm.scoped_session(sa.orm.sessionmaker())
    Session.configure(bind=engine)
    session = Session()
    for project in session.query(model.Project).filter(
            model.Project.name.ilike('%/%')).all():
        nspace, name = project.name.split('/', 1)
        project.name = name
        project.namespace = nspace
        session.add(project)
    session.commit()
 
 
def downgrade():
    ''' Remove the column namespace from the table projects.
    '''
    # Update all the existing projects
    engine = op.get_bind()
    Session = sa.orm.scoped_session(sa.orm.sessionmaker())
    Session.configure(bind=engine)
    session = Session()
    for project in session.query(model.Project).filter(
            model.Project.namespace != None).all():
        if project.namespace.strip():
            project.name = '%s/%s' % (project.namespace, project.name)
            session.add(project)
    session.commit()
 
    op.drop_column('projects', 'namespace')