Add FK's from Record/RecordSet to domain.id

Closes-Bug: #1520523
Change-Id: Id719657384515e40dc1db78836da9ac9c267b001

1 files changed, 19 insertions, 3 deletions

diff --git a/designate/storage/impl_sqlalchemy/migrate_repo/versions/057_placeholder.py b/designate/storage/impl_sqlalchemy/migrate_repo/versions/057_add_r_domain_FK.py
index 5626d695..1e7855db 100644
--- a/designate/storage/impl_sqlalchemy/migrate_repo/versions/057_placeholder.py
+++ b/designate/storage/impl_sqlalchemy/migrate_repo/versions/057_add_r_domain_FK.py
@@ -21,10 +21,26 @@
 # See https://blueprints.launchpad.net/nova/+spec/backportable-db-migrations
 # http://lists.openstack.org/pipermail/openstack-dev/2013-March/006827.html
 
+from migrate.changeset.constraint import ForeignKeyConstraint
+from sqlalchemy.schema import MetaData, Table
+
+# This migration adds back the FKs removed in migration 80, as sqlalchemy
+# migrate seems to need to wait to add FKs to renamed tables.
+
+meta = MetaData()
+
 
 def upgrade(migrate_engine):
-    pass
+    meta.bind = migrate_engine
 
+    domains_table = Table('domains', meta, autoload=True)
+    recordsets_table = Table('recordsets', meta, autoload=True)
+    records_table = Table('records', meta, autoload=True)
 
-def downgrade(migration_engine):
-    pass
+    fks = []
+    fks.append(ForeignKeyConstraint([recordsets_table.c.domain_id],
+                                    [domains_table.c.id], ondelete='CASCADE'))
+    fks.append(ForeignKeyConstraint([records_table.c.domain_id],
+                                    [domains_table.c.id], ondelete='CASCADE'))
+    for fk in fks:
+        fk.create()