From 78a2617703bc1dada409f126db5c3db214913ff4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Anssi=20K=C3=A4=C3=A4ri=C3=A4inen?= Date: Mon, 20 Jan 2014 18:28:16 +0200 Subject: [PATCH] Fixed #21787 -- regression in MTI .exclude() queries --- django/db/models/sql/query.py | 8 ++++++-- tests/queries/tests.py | 18 ++++++++++++++++++ 2 files changed, 24 insertions(+), 2 deletions(-) diff --git a/django/db/models/sql/query.py b/django/db/models/sql/query.py index dcf89fe8bf..838c2914d9 100644 --- a/django/db/models/sql/query.py +++ b/django/db/models/sql/query.py @@ -1334,6 +1334,7 @@ class Query(object): """ path, names_with_path = [], [] for pos, name in enumerate(names): + cur_names_with_path = (name, []) if name == 'pk': name = opts.pk.name try: @@ -1359,19 +1360,22 @@ class Query(object): targets = (final_field.rel.get_related_field(),) opts = int_model._meta path.append(PathInfo(final_field.model._meta, opts, targets, final_field, False, True)) + cur_names_with_path[1].append(PathInfo(final_field.model._meta, opts, targets, final_field, False, True)) if hasattr(field, 'get_path_info'): pathinfos = field.get_path_info() if not allow_many: for inner_pos, p in enumerate(pathinfos): if p.m2m: - names_with_path.append((name, pathinfos[0:inner_pos + 1])) + cur_names_with_path[1].extend(pathinfos[0:inner_pos + 1]) + names_with_path.append(cur_names_with_path) raise MultiJoin(pos + 1, names_with_path) last = pathinfos[-1] path.extend(pathinfos) final_field = last.join_field opts = last.to_opts targets = last.target_fields - names_with_path.append((name, pathinfos)) + cur_names_with_path[1].extend(pathinfos) + names_with_path.append(cur_names_with_path) else: # Local non-relational field. final_field = field diff --git a/tests/queries/tests.py b/tests/queries/tests.py index 03cfc71afe..b303d5f603 100644 --- a/tests/queries/tests.py +++ b/tests/queries/tests.py @@ -3225,3 +3225,21 @@ class ValuesJoinPromotionTests(TestCase): self.assertEqual(qs.count(), 1) tblname = connection.ops.quote_name(ObjectB._meta.db_table) self.assertTrue(' LEFT OUTER JOIN %s' % tblname in str(qs.query)) + + +class ForeignKeyToBaseExcludeTests(TestCase): + def test_ticket_21787(self): + sc1 = SpecialCategory.objects.create(special_name='sc1', name='sc1') + sc2 = SpecialCategory.objects.create(special_name='sc2', name='sc2') + sc3 = SpecialCategory.objects.create(special_name='sc3', name='sc3') + c1 = CategoryItem.objects.create(category=sc1) + CategoryItem.objects.create(category=sc2) + self.assertQuerysetEqual( + SpecialCategory.objects.exclude( + categoryitem__id=c1.pk).order_by('name'), + [sc2, sc3], lambda x: x + ) + self.assertQuerysetEqual( + SpecialCategory.objects.filter(categoryitem__id=c1.pk), + [sc1], lambda x: x + )