openstack/aodh
Code Issues Proposed changes

Merge "Fix missing foreign keys."

Browse Source
This commit is contained in:
Jenkins 2013-07-24 06:00:18 +00:00 committed by Gerrit Code Review
commit 618c788520
1 changed files with 67 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
ceilometer/storage/sqlalchemy/migrate_repo/versions/012_add_missing_foreign_keys.py Normal file
View File

@ -0,0 +1,67 @@
# -*- encoding: utf-8 -*-
#
#
# 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.
from migrate import ForeignKeyConstraint
from sqlalchemy import MetaData, Table
TABLES = ['resource', 'sourceassoc', 'user',
'project', 'meter', 'source', 'alarm']
INDEXES = {
"resource": (('user_id', 'user', 'id'),
('project_id', 'project', 'id')),
"sourceassoc": (('user_id', 'user', 'id'),
('project_id', 'project', 'id'),
('resource_id', 'resource', 'id'),
('meter_id', 'meter', 'id'),
('source_id', 'source', 'id')),
"alarm": (('user_id', 'user', 'id'),
('project_id', 'project', 'id')),
"meter": (('user_id', 'user', 'id'),
('project_id', 'project', 'id'),
('resource_id', 'resource', 'id'),)
}
def upgrade(migrate_engine):
meta = MetaData(bind=migrate_engine)
load_tables = dict((table_name, Table(table_name, meta, autoload=True))
for table_name in TABLES)
for table_name, indexes in INDEXES.items():
table = load_tables[table_name]
for column, ref_table_name, ref_column_name in indexes:
ref_table = load_tables[ref_table_name]
params = {'columns': [table.c[column]],
'refcolumns': [ref_table.c[ref_column_name]]}
if migrate_engine.name == 'mysql':
params['name'] = "_".join(('fk', table_name, column))
fkey = ForeignKeyConstraint(**params)
fkey.create()
def downgrade(migrate_engine):
meta = MetaData(bind=migrate_engine)
load_tables = dict((table_name, Table(table_name, meta, autoload=True))
for table_name in TABLES)
for table_name, indexes in INDEXES.items():
table = load_tables[table_name]
for column, ref_table_name, ref_column_name in indexes:
ref_table = load_tables[ref_table_name]
params = {'columns': [table.c[column]],
'refcolumns': [ref_table.c[ref_column_name]]}
if migrate_engine.name == 'mysql':
params['name'] = "_".join(('fk', table_name, column))
fkey = ForeignKeyConstraint(**params)
fkey.drop()