summaryrefslogtreecommitdiff
path: root/django/db/models/sql/compiler.py
diff options
context:
space:
mode:
authorSimon Charette <charette.s@gmail.com>2022-09-22 23:57:06 -0400
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-10-06 09:34:31 +0200
commit04518e310d4552ff7595a34f5a7f93487d78a406 (patch)
tree0cb94a0dfc2b8c63d3da0ceb90d41b048b067695 /django/db/models/sql/compiler.py
parentc58a8acd413ccc992dd30afd98ed900897e1f719 (diff)
downloaddjango-04518e310d4552ff7595a34f5a7f93487d78a406.tar.gz
Refs #33308 -- Enabled explicit GROUP BY and ORDER BY aliases.
This ensures explicit grouping from using values() before annotating an aggregate function groups by selected aliases if supported. The GROUP BY feature is disabled on Oracle because it doesn't support it.
Diffstat (limited to 'django/db/models/sql/compiler.py')
-rw-r--r--django/db/models/sql/compiler.py23
1 files changed, 18 insertions, 5 deletions
diff --git a/django/db/models/sql/compiler.py b/django/db/models/sql/compiler.py
index f566546307..e4605918d9 100644
--- a/django/db/models/sql/compiler.py
+++ b/django/db/models/sql/compiler.py
@@ -131,9 +131,12 @@ class SQLCompiler:
# Converts string references to expressions.
for expr in self.query.group_by:
if not hasattr(expr, "as_sql"):
- expressions.append(self.query.resolve_ref(expr))
- else:
- expressions.append(expr)
+ expr = self.query.resolve_ref(expr)
+ if not self.connection.features.allows_group_by_refs and isinstance(
+ expr, Ref
+ ):
+ expr = expr.source
+ expressions.append(expr)
# Note that even if the group_by is set, it is only the minimal
# set to group by. So, we need to add cols in select, order_by, and
# having into the select in any case.
@@ -344,7 +347,13 @@ class SQLCompiler:
if not self.query.standard_ordering:
field = field.copy()
field.reverse_ordering()
- yield field, False
+ if isinstance(field.expression, F) and (
+ annotation := self.query.annotation_select.get(
+ field.expression.name
+ )
+ ):
+ field.expression = Ref(field.expression.name, annotation)
+ yield field, isinstance(field.expression, Ref)
continue
if field == "?": # random
yield OrderBy(Random()), False
@@ -432,6 +441,10 @@ class SQLCompiler:
"""
result = []
seen = set()
+ replacements = {
+ expr: Ref(alias, expr)
+ for alias, expr in self.query.annotation_select.items()
+ }
for expr, is_ref in self._order_by_pairs():
resolved = expr.resolve_expression(self.query, allow_joins=True, reuse=None)
@@ -461,7 +474,7 @@ class SQLCompiler:
q.add_annotation(expr_src, col_name)
self.query.add_select_col(resolved, col_name)
resolved.set_source_expressions([RawSQL(f"{order_by_idx}", ())])
- sql, params = self.compile(resolved)
+ sql, params = self.compile(resolved.replace_expressions(replacements))
# Don't add the same column twice, but the order direction is
# not taken into account so we strip it. When this entire method
# is refactored into expressions, then we can check each part as we