bugfix for pkname's alias (#624)

2022-04-25 17:44:56 +08:00
parent 801e22e58f
commit 2caa17812a
1 changed files with 2 additions and 1 deletions
--- a/ormar/queryset/queryset.py
+++ b/ormar/queryset/queryset.py
@ -697,7 +697,8 @@ class QuerySet(Generic[T]):
        expr = self.build_select_expression().alias("subquery_for_count")
        expr = sqlalchemy.func.count().select().select_from(expr)
        if distinct:
-            expr_distinct = expr.group_by(self.model_meta.pkname).alias(
+            pk_column_name = self.model.get_column_alias(self.model_meta.pkname)
+            expr_distinct = expr.group_by(pk_column_name).alias(
                "subquery_for_group"
            )
            expr = sqlalchemy.func.count().select().select_from(expr_distinct)