deb-gnocchi/gnocchi/indexer/alembic/versions/f7d44b47928_uuid_to_binary.py

90 lines
3.3 KiB
Python

#
# Copyright 2015 OpenStack Foundation
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
"""uuid_to_binary
Revision ID: f7d44b47928
Revises: 40c6aae14c3f
Create Date: 2015-04-30 13:29:29.074794
"""
# revision identifiers, used by Alembic.
revision = 'f7d44b47928'
down_revision = '40c6aae14c3f'
branch_labels = None
depends_on = None
from alembic import op
import sqlalchemy_utils.types.uuid
def upgrade():
op.alter_column("metric", "id",
type_=sqlalchemy_utils.types.uuid.UUIDType(binary=True),
nullable=False)
for table in ('resource', 'resource_history', 'metric'):
op.alter_column(table, "created_by_user_id",
type_=sqlalchemy_utils.types.uuid.UUIDType(
binary=True))
op.alter_column(table, "created_by_project_id",
type_=sqlalchemy_utils.types.uuid.UUIDType(
binary=True))
for table in ('resource', 'resource_history'):
op.alter_column(table, "user_id",
type_=sqlalchemy_utils.types.uuid.UUIDType(
binary=True))
op.alter_column(table, "project_id",
type_=sqlalchemy_utils.types.uuid.UUIDType(
binary=True))
# Drop all foreign keys linking to resource.id
for table in ('ceph_account', 'identity', 'volume', 'swift_account',
'ipmi', 'image', 'network', 'stack', 'instance',
'resource_history'):
op.drop_constraint("fk_%s_id_resource_id" % table, table,
type_="foreignkey")
op.drop_constraint("fk_metric_resource_id_resource_id", "metric",
type_="foreignkey")
# Now change the type of resource.id
op.alter_column("resource", "id",
type_=sqlalchemy_utils.types.uuid.UUIDType(binary=True),
nullable=False)
# Now change all the types of $table.id and re-add the FK
for table in ('ceph_account', 'identity', 'volume', 'swift_account',
'ipmi', 'image', 'network', 'stack', 'instance',
'resource_history'):
op.alter_column(
table, "id",
type_=sqlalchemy_utils.types.uuid.UUIDType(binary=True),
nullable=False)
op.create_foreign_key("fk_%s_id_resource_id" % table,
table, "resource",
("id",), ("id",),
ondelete="CASCADE")
op.alter_column("metric", "resource_id",
type_=sqlalchemy_utils.types.uuid.UUIDType(binary=True))
op.create_foreign_key("fk_metric_resource_id_resource_id",
"metric", "resource",
("resource_id",), ("id",),
ondelete="CASCADE")