1 files changed, 54 insertions, 0 deletions
diff --git a/keystone-moon/keystone/contrib/oauth1/migrate_repo/versions/002_fix_oauth_tables_fk.py b/keystone-moon/keystone/contrib/oauth1/migrate_repo/versions/002_fix_oauth_tables_fk.py
new file mode 100644
index 00000000..d39df8d5
--- /dev/null
+++ b/keystone-moon/keystone/contrib/oauth1/migrate_repo/versions/002_fix_oauth_tables_fk.py
@@ -0,0 +1,54 @@
+# Copyright 2013 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.
+
+import sqlalchemy as sql
+
+from keystone.common.sql import migration_helpers
+
+
+def upgrade(migrate_engine):
+    # Upgrade operations go here. Don't create your own engine; bind
+    # migrate_engine to your metadata
+    meta = sql.MetaData()
+    meta.bind = migrate_engine
+
+    consumer_table = sql.Table('consumer', meta, autoload=True)
+    request_token_table = sql.Table('request_token', meta, autoload=True)
+    access_token_table = sql.Table('access_token', meta, autoload=True)
+
+    constraints = [{'table': request_token_table,
+                    'fk_column': 'consumer_id',
+                    'ref_column': consumer_table.c.id},
+                   {'table': access_token_table,
+                    'fk_column': 'consumer_id',
+                    'ref_column': consumer_table.c.id}]
+    if meta.bind != 'sqlite':
+        migration_helpers.add_constraints(constraints)
+
+
+def downgrade(migrate_engine):
+    meta = sql.MetaData()
+    meta.bind = migrate_engine
+    consumer_table = sql.Table('consumer', meta, autoload=True)
+    request_token_table = sql.Table('request_token', meta, autoload=True)
+    access_token_table = sql.Table('access_token', meta, autoload=True)
+
+    constraints = [{'table': request_token_table,
+                    'fk_column': 'consumer_id',
+                    'ref_column': consumer_table.c.id},
+                   {'table': access_token_table,
+                    'fk_column': 'consumer_id',
+                    'ref_column': consumer_table.c.id}]
+    if migrate_engine.name != 'sqlite':
+        migration_helpers.remove_constraints(constraints)