OpenStack Study: 019_migrate_image_locations.py

**** CubicPower OpenStack Study ****

def get_images_table(meta):

    return sqlalchemy.Table('images', meta, autoload=True)

**** CubicPower OpenStack Study ****

def get_image_locations_table(meta):

    return sqlalchemy.Table('image_locations', meta, autoload=True)

**** CubicPower OpenStack Study ****

def upgrade(migrate_engine):

    meta = sqlalchemy.schema.MetaData(migrate_engine)

    images_table = get_images_table(meta)

    image_locations_table = get_image_locations_table(meta)

    image_records = images_table.select().execute().fetchall()

    for image in image_records:

        if image.location is not None:

            values = {

                'image_id': image.id,

                'value': image.location,

                'created_at': image.created_at,

                'updated_at': image.updated_at,

                'deleted': image.deleted,

                'deleted_at': image.deleted_at,

            }

            image_locations_table.insert(values=values).execute()

**** CubicPower OpenStack Study ****

def downgrade(migrate_engine):

    meta = sqlalchemy.schema.MetaData(migrate_engine)

    images_table = get_images_table(meta)

    image_locations_table = get_image_locations_table(meta)

    image_records = image_locations_table.select().execute().fetchall()

    for image_location in image_records:

        images_table.update(values={'location': image_location.value})\

                    .where(images_table.c.id == image_location.image_id)\

                    .execute()