Fixed #23659 -- Kept annotate() args ordering
Thanks Loic Bistuer and Simon Charette for the review.
This commit is contained in:
parent
947af46db3
commit
374c14b7fd
|
@ -2,7 +2,7 @@
|
|||
The main QuerySet implementation. This provides the public API for the ORM.
|
||||
"""
|
||||
|
||||
from collections import deque
|
||||
from collections import deque, OrderedDict
|
||||
import copy
|
||||
import sys
|
||||
import warnings
|
||||
|
@ -786,27 +786,29 @@ class QuerySet(object):
|
|||
Return a query set in which the returned objects have been annotated
|
||||
with data aggregated from related fields.
|
||||
"""
|
||||
aggrs = OrderedDict() # To preserve ordering of args
|
||||
for arg in args:
|
||||
if arg.default_alias in kwargs:
|
||||
raise ValueError("The named annotation '%s' conflicts with the "
|
||||
"default name for another annotation."
|
||||
% arg.default_alias)
|
||||
kwargs[arg.default_alias] = arg
|
||||
aggrs[arg.default_alias] = arg
|
||||
aggrs.update(kwargs)
|
||||
|
||||
names = getattr(self, '_fields', None)
|
||||
if names is None:
|
||||
names = set(self.model._meta.get_all_field_names())
|
||||
for aggregate in kwargs:
|
||||
for aggregate in aggrs:
|
||||
if aggregate in names:
|
||||
raise ValueError("The annotation '%s' conflicts with a field on "
|
||||
"the model." % aggregate)
|
||||
|
||||
obj = self._clone()
|
||||
|
||||
obj._setup_aggregate_query(list(kwargs))
|
||||
obj._setup_aggregate_query(list(aggrs))
|
||||
|
||||
# Add the aggregates to the query
|
||||
for (alias, aggregate_expr) in kwargs.items():
|
||||
for (alias, aggregate_expr) in aggrs.items():
|
||||
obj.query.add_aggregate(aggregate_expr, self.model, alias,
|
||||
is_summary=False)
|
||||
|
||||
|
|
|
@ -740,6 +740,17 @@ class AggregationTests(TestCase):
|
|||
list(qs), list(Book.objects.values_list("name", flat=True))
|
||||
)
|
||||
|
||||
def test_values_list_annotation_args_ordering(self):
|
||||
"""
|
||||
Annotate *args ordering should be preserved in values_list results.
|
||||
**kwargs comes after *args.
|
||||
Regression test for #23659.
|
||||
"""
|
||||
books = Book.objects.values_list("publisher__name").annotate(
|
||||
Count("id"), Avg("price"), Avg("authors__age"), avg_pgs=Avg("pages")
|
||||
).order_by("-publisher__name")
|
||||
self.assertEqual(books[0], ('Sams', 1, 23.09, 45.0, 528.0))
|
||||
|
||||
def test_annotation_disjunction(self):
|
||||
qs = Book.objects.annotate(n_authors=Count("authors")).filter(
|
||||
Q(n_authors=2) | Q(name="Python Web Development with Django")
|
||||
|
|
Loading…
Reference in New Issue