Commit 9918c111 authored by Anssi Kääriäinen's avatar Anssi Kääriäinen
Browse files

Fixed #21413 -- resolve_columns fields misalignment

parent 9cc3371a

django/contrib/gis/db/models/sql/compiler.py

+1 −1
Original line number Diff line number Diff line
@@ -133,7 +133,7 @@ class GeoSQLCompiler(compiler.SQLCompiler): 
            if table in only_load and field.column not in only_load[table]:

                continue

            if as_pairs:

                result.append((alias, field.column))

                result.append((alias, field))

                aliases.add(alias)

                continue

            # This part of the function is customized for GeoQuery. We

django/db/models/sql/compiler.py

+8 −7
Original line number Diff line number Diff line
@@ -295,7 +295,7 @@ class SQLCompiler(object): 
            if table in only_load and column not in only_load[table]:

                continue

            if as_pairs:

                result.append((alias, column))

                result.append((alias, field))

                aliases.add(alias)

                continue

            if with_aliases and column in col_aliases:
@@ -625,10 +625,10 @@ class SQLCompiler(object): 
            _, _, _, joins, _ = self.query.setup_joins(

                [f.name], opts, root_alias)

            alias = joins[-1]

            columns, aliases = self.get_default_columns(start_alias=alias,

            columns, _ = self.get_default_columns(start_alias=alias,

                    opts=f.rel.to._meta, as_pairs=True)

            self.query.related_select_cols.extend(

                SelectInfo(col, field) for col, field in zip(columns, f.rel.to._meta.concrete_fields))

                SelectInfo((col[0], col[1].column), col[1]) for col in columns)

            if restricted:

                next = requested.get(f.name, {})

            else:
@@ -652,11 +652,10 @@ class SQLCompiler(object): 
                alias = joins[-1]

                from_parent = (opts.model if issubclass(model, opts.model)

                               else None)

                columns, aliases = self.get_default_columns(start_alias=alias,

                columns, _ = self.get_default_columns(start_alias=alias,

                    opts=model._meta, as_pairs=True, from_parent=from_parent)

                self.query.related_select_cols.extend(

                    SelectInfo(col, field) for col, field

                    in zip(columns, model._meta.concrete_fields))

                    SelectInfo((col[0], col[1].column), col[1]) for col in columns)

                next = requested.get(f.related_query_name(), {})

                self.fill_related_selections(model._meta, alias, cur_depth + 1,

                                             next, restricted)
@@ -699,8 +698,10 @@ class SQLCompiler(object): 
                        # found in get_columns(). It would be nice to clean this up.

                        if self.query.select:

                            fields = [f.field for f in self.query.select]

                        else:

                        elif self.query.default_cols:

                            fields = self.query.get_meta().concrete_fields

                        else:

                            fields = []

                        fields = fields + [f.field for f in self.query.related_select_cols]



                        # If the field was deferred, exclude it from being passed

tests/model_inheritance_regress/tests.py

+6 −0
Original line number Diff line number Diff line
@@ -460,3 +460,9 @@ class ModelInheritanceTest(TestCase): 
        )



        Senator.objects.get(pk=senator.pk)



    def test_inheritance_resolve_columns(self):

        Restaurant.objects.create(name='Bobs Cafe', address="Somewhere",

                                  serves_pizza=True, serves_hot_dogs=True)

        p = Place.objects.all().select_related('restaurant')[0]

        self.assertIsInstance(p.restaurant.serves_pizza, bool)