python-sqlalchemy/fix-regression-in-ambiguous...

116 lines
4.3 KiB
Diff

diff --git a/lib/sqlalchemy/sql/util.py b/lib/sqlalchemy/sql/util.py
index 3077840..8719b26 100644
--- a/lib/sqlalchemy/sql/util.py
+++ b/lib/sqlalchemy/sql/util.py
@@ -20,6 +20,7 @@ from .annotation import _shallow_annotate # noqa
from .base import _from_objects
from .base import ColumnSet
from .ddl import sort_tables # noqa
+from .elements import _expand_cloned
from .elements import _find_columns # noqa
from .elements import _label_reference
from .elements import _textual_label_reference
@@ -149,6 +150,16 @@ def find_left_clause_to_join_from(clauses, join_to, onclause):
idx.append(i)
break
+ if len(idx) > 1:
+ # this is the same "hide froms" logic from
+ # Selectable._get_display_froms
+ toremove = set(
+ chain(*[_expand_cloned(f._hide_froms) for f in clauses])
+ )
+ idx = [
+ i for i in idx if clauses[i] not in toremove
+ ]
+
# onclause was given and none of them resolved, so assume
# all indexes can match
if not idx and onclause is not None:
diff --git a/test/orm/test_joins.py b/test/orm/test_joins.py
index bb9747c..4fe04f7 100644
--- a/test/orm/test_joins.py
+++ b/test/orm/test_joins.py
@@ -1559,6 +1559,81 @@ class JoinTest(QueryTest, AssertsCompiledSQL):
"ON dingalings.address_id = addresses_1.id",
)
+ def test_clause_present_in_froms_twice_w_onclause(self):
+ # test [ticket:4584]
+ Order, Address, Dingaling, User = (
+ self.classes.Order,
+ self.classes.Address,
+ self.classes.Dingaling,
+ self.classes.User,
+ )
+
+ sess = create_session()
+
+ a1 = aliased(Address)
+
+ q = sess.query(Order).select_from(Order, a1, User)
+ assert_raises_message(
+ sa.exc.InvalidRequestError,
+ "Can't determine which FROM clause to join from, there are "
+ "multiple FROMS which can join to this entity. "
+ "Try adding an explicit ON clause to help resolve the ambiguity.",
+ q.outerjoin,
+ a1,
+ )
+
+ # the condition which occurs here is: Query._from_obj contains both
+ # "a1" by itself as well as a join that "a1" is part of.
+ # find_left_clause_to_join_from() needs to include removal of froms
+ # that are in the _hide_froms of joins the same way
+ # Selectable._get_display_froms does.
+ q = sess.query(Order).select_from(Order, a1, User)
+ q = q.outerjoin(a1, a1.id == Order.address_id)
+ q = q.outerjoin(User, a1.user_id == User.id)
+
+ self.assert_compile(
+ q,
+ "SELECT orders.id AS orders_id, orders.user_id AS orders_user_id, "
+ "orders.address_id AS orders_address_id, "
+ "orders.description AS orders_description, "
+ "orders.isopen AS orders_isopen "
+ "FROM orders "
+ "LEFT OUTER JOIN addresses AS addresses_1 "
+ "ON addresses_1.id = orders.address_id "
+ "LEFT OUTER JOIN users ON addresses_1.user_id = users.id",
+ )
+
+ def test_clause_present_in_froms_twice_wo_onclause(self):
+ # test [ticket:4584]
+ Order, Address, Dingaling, User = (
+ self.classes.Order,
+ self.classes.Address,
+ self.classes.Dingaling,
+ self.classes.User,
+ )
+
+ sess = create_session()
+
+ a1 = aliased(Address)
+
+ # the condition which occurs here is: Query._from_obj contains both
+ # "a1" by itself as well as a join that "a1" is part of.
+ # find_left_clause_to_join_from() needs to include removal of froms
+ # that are in the _hide_froms of joins the same way
+ # Selectable._get_display_froms does.
+ q = sess.query(User).select_from(Dingaling, a1, User)
+ q = q.outerjoin(a1, User.id == a1.user_id)
+ q = q.outerjoin(Dingaling)
+
+ self.assert_compile(
+ q,
+ "SELECT users.id AS users_id, users.name AS users_name "
+ "FROM users LEFT OUTER JOIN addresses AS addresses_1 "
+ "ON users.id = addresses_1.user_id "
+ "LEFT OUTER JOIN dingalings "
+ "ON addresses_1.id = dingalings.address_id",
+ )
+
def test_multiple_adaption(self):
Item, Order, User = (
self.classes.Item,