From: Jason Kirtland <jek@discorporate.us>
Date: Tue, 18 Dec 2007 19:37:03 +0000 (+0000)
Subject: Apply default cascade rules for firebird self-ref ForeignKeys.
X-Git-Tag: rel_0_4_2~29
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=1dc4546aa95d3adfff910430afe922bf137bf595;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

Apply default cascade rules for firebird self-ref ForeignKeys.
---

diff --git a/test/testlib/schema.py b/test/testlib/schema.py
index 6ca15bb6c4..96a81d5bea 100644
--- a/test/testlib/schema.py
+++ b/test/testlib/schema.py
@@ -1,4 +1,5 @@
 import testbase
+from testlib import testing
 schema = None
 
 __all__ = 'Table', 'Column',
@@ -11,7 +12,7 @@ def Table(*args, **kw):
     global schema
     if schema is None:
         from sqlalchemy import schema
-        
+
     test_opts = dict([(k,kw.pop(k)) for k in kw.keys()
                       if k.startswith('test_')])
 
@@ -22,6 +23,33 @@ def Table(*args, **kw):
             if 'test_needs_fk' in test_opts or 'test_needs_acid' in test_opts:
                 kw['mysql_engine'] = 'InnoDB'
 
+    # Apply some default cascading rules for self-referential foreign keys.
+    # MySQL InnoDB has some issues around seleting self-refs too.
+    if testing.against('firebird'):
+        table_name = args[0]
+        unpack = (testing.config.db.dialect.
+                  identifier_preparer.unformat_identifiers)
+
+        # Only going after ForeignKeys in Columns.  May need to
+        # expand to ForeignKeyConstraint too.
+        fks = [fk
+               for col in args if isinstance(col, schema.Column)
+               for fk in col.args if isinstance(fk, schema.ForeignKey)]
+
+        for fk in fks:
+            # root around in raw spec
+            ref = fk._colspec
+            if isinstance(ref, schema.Column):
+                name = ref.table.name
+            else:
+                name = unpack(ref)[-2]
+            print name, table_name
+            if name == table_name:
+                if fk.ondelete is None:
+                    fk.ondelete = 'CASCADE'
+                if fk.onupdate is None:
+                    fk.onupdate = 'CASCADE'
+
     return schema.Table(*args, **kw)
 
 def Column(*args, **kw):