OpenStack Study: 010_normalize_identity_migration.py

**** CubicPower OpenStack Study ****

def is_enabled(enabled):

    # no explicit value means enabled

    if enabled is True or enabled is None:

        return True

    if (isinstance(enabled, six.string_types)

            and enabled.lower() in DISABLED_VALUES):

        return False

    return bool(enabled)

**** CubicPower OpenStack Study ****

def downgrade_user_table(meta, migrate_engine, session):

    user_table = Table('user', meta, autoload=True)

    for user in session.query(user_table).all():

        extra = json.loads(user.extra)

        extra['password'] = user.password

        extra['enabled'] = '%r' % is_enabled(user.enabled)

        values = {'extra': json.dumps(extra)}

        update = user_table.update().\

            where(user_table.c.id == user.id).\

            values(values)

        migrate_engine.execute(update)

**** CubicPower OpenStack Study ****

def downgrade_tenant_table(meta, migrate_engine, session):

    tenant_table = Table('tenant', meta, autoload=True)

    for tenant in session.query(tenant_table).all():

        extra = json.loads(tenant.extra)

        extra['description'] = tenant.description

        extra['enabled'] = '%r' % is_enabled(tenant.enabled)

        values = {'extra': json.dumps(extra)}

        update = tenant_table.update().\

            where(tenant_table.c.id == tenant.id).\

            values(values)

        migrate_engine.execute(update)

**** CubicPower OpenStack Study ****

def upgrade_user_table(meta, migrate_engine, session):

    user_table = Table('user', meta, autoload=True)

    for user in session.query(user_table).all():

        extra = json.loads(user.extra)

        values = {'password': extra.pop('password', None),

                  'enabled': is_enabled(extra.pop('enabled', True)),

                  'extra': json.dumps(extra)}

        update = user_table.update().\

            where(user_table.c.id == user.id).\

            values(values)

        migrate_engine.execute(update)

**** CubicPower OpenStack Study ****

def upgrade_tenant_table(meta, migrate_engine, session):

    tenant_table = Table('tenant', meta, autoload=True)

    for tenant in session.query(tenant_table):

        extra = json.loads(tenant.extra)

        values = {'description': extra.pop('description', None),

                  'enabled': is_enabled(extra.pop('enabled', True)),

                  'extra': json.dumps(extra)}

        update = tenant_table.update().\

            where(tenant_table.c.id == tenant.id).\

            values(values)

        migrate_engine.execute(update)

**** CubicPower OpenStack Study ****

def upgrade(migrate_engine):

    meta = MetaData()

    meta.bind = migrate_engine

    session = sessionmaker(bind=migrate_engine)()

    upgrade_user_table(meta, migrate_engine, session)

    upgrade_tenant_table(meta, migrate_engine, session)

    session.commit()

**** CubicPower OpenStack Study ****

def downgrade(migrate_engine):

    meta = MetaData()

    meta.bind = migrate_engine

    session = sessionmaker(bind=migrate_engine)()

    downgrade_user_table(meta, migrate_engine, session)

    downgrade_tenant_table(meta, migrate_engine, session)

    session.commit()