review.fuel-infra Code Review - openstack-build/cinder-build.git/blob

   1 # vim: tabstop=4 shiftwidth=4 softtabstop=4
   2
   3 # Copyright 2011 OpenStack LLC.
   4 # Copyright 2012 Michael Still and Canonical Inc
   5 # All Rights Reserved.
   6 #
   7 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
   8 #    not use this file except in compliance with the License. You may obtain
   9 #    a copy of the License at
  10 #
  11 #         http://www.apache.org/licenses/LICENSE-2.0
  12 #
  13 #    Unless required by applicable law or agreed to in writing, software
  14 #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  15 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  16 #    License for the specific language governing permissions and limitations
  17 #    under the License.
  18
  19 from sqlalchemy import select, Column, ForeignKey, Integer
  20 from sqlalchemy import MetaData, String, Table
  21 from migrate import ForeignKeyConstraint
  22
  23 from cinder import log as logging
  24
  25
  26 LOG = logging.getLogger(__name__)
  27
  28
  29 def upgrade(migrate_engine):
  30     meta = MetaData()
  31     meta.bind = migrate_engine
  32     dialect = migrate_engine.url.get_dialect().name
  33     block_device_mapping = Table('block_device_mapping', meta, autoload=True)
  34     instances = Table('instances', meta, autoload=True)
  35     uuid_column = Column('instance_uuid', String(36))
  36     uuid_column.create(block_device_mapping)
  37
  38     try:
  39         block_device_mapping.update().values(
  40             instance_uuid=select(
  41                 [instances.c.uuid],
  42                 instances.c.id == block_device_mapping.c.instance_id)
  43         ).execute()
  44     except Exception:
  45         uuid_column.drop()
  46         raise
  47
  48     fkeys = list(block_device_mapping.c.instance_id.foreign_keys)
  49     if fkeys:
  50         try:
  51             fkey_name = fkeys[0].constraint.name
  52             ForeignKeyConstraint(
  53                 columns=[block_device_mapping.c.instance_id],
  54                 refcolumns=[instances.c.id],
  55                 name=fkey_name).drop()
  56         except Exception:
  57             LOG.error(_("foreign key constraint couldn't be removed"))
  58             raise
  59
  60     block_device_mapping.c.instance_id.drop()
  61
  62
  63 def downgrade(migrate_engine):
  64     meta = MetaData()
  65     meta.bind = migrate_engine
  66     block_device_mapping = Table('block_device_mapping', meta, autoload=True)
  67     instances = Table('instances', meta, autoload=True)
  68     id_column = Column('instance_id', Integer, ForeignKey('instances.id'))
  69     id_column.create(block_device_mapping)
  70
  71     try:
  72         block_device_mapping.update().values(
  73             instance_id=select(
  74                 [instances.c.id],
  75                 instances.c.uuid == block_device_mapping.c.instance_uuid)
  76         ).execute()
  77     except Exception:
  78         id_column.drop()
  79         raise
  80
  81     block_device_mapping.c.instance_uuid.drop()