From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Fri, 9 Nov 2007 15:11:38 +0000 (+0000)
Subject: a tweak, still requiring test coverage, to add more "foreign key equivalents" to... 
X-Git-Tag: rel_0_4_1~39
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=d54ca6a62afbccab260e5b5e704be8e45e2a026d;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

a tweak, still requiring test coverage, to add more "foreign key equivalents" to the equivalent_columns collection
---

diff --git a/lib/sqlalchemy/orm/mapper.py b/lib/sqlalchemy/orm/mapper.py
index 3cfe7b3f61..aaff050285 100644
--- a/lib/sqlalchemy/orm/mapper.py
+++ b/lib/sqlalchemy/orm/mapper.py
@@ -519,12 +519,15 @@ class Mapper(object):
             if mapper.inherit_condition is not None:
                 vis.traverse(mapper.inherit_condition)
 
-        for col in (self.primary_key_argument or self.pks_by_table[self.mapped_table]):
-            if not col.foreign_keys:
-                result.setdefault(col, util.Set()).add(col)
-            else:
-                for fk in col.foreign_keys:
-                    result.setdefault(fk.column, util.Set()).add(col)
+        # TODO: matching of cols to foreign keys might better be generalized
+        # into general column translation (i.e. corresponding_column)
+        for column in (self.primary_key_argument or self.pks_by_table[self.mapped_table]):
+            for col in column.proxy_set:
+                if not col.foreign_keys:
+                    result.setdefault(col, util.Set()).add(col)
+                else:
+                    for fk in col.foreign_keys:
+                        result.setdefault(fk.column, util.Set()).add(col)
 
         return result