Question

我的数据库中有2个模型：Movie和Comment（它有一个指向Movie的外键）。

from django.db import models


class Movie(models.Model):
    title = models.CharField(max_length=120, blank=False, unique=True)

class Comment(models.Model):
    movie = models.ForeignKey(movie_models.Movie, null=False, on_delete=models.CASCADE)
    body = models.TextField(blank=True)

我想做的是创建一个Movies的排名，该排名由相关Comments的数量决定。

我可以注释大量注释并正确排序结果：

from django.db.models import Count

qs = Movie.objects.annotate(total_comments=Count('comment')).order_by('-total_comments')

这给了我正确的查询集，但我想做一件事-在每行上标注“排名”。

所以我的问题是：如何对结果的每一行进行“排名”注释？请注意，具有相同评论量的电影必须具有相同的等级。

|  movie_title | total_comments | rank |
|--------------|----------------|------|
| mov1         | 10             | 1    |
| mov2         | 5              | 2    |
| mov3         | 5              | 2    |
| mov4         | 3              | 3    |

我尝试使用window functions，因为其中一些示例对我而言是合法的：

from django.db.models import F, Window

dense_rank = Window(
    expression=DenseRank(),
    order_by=F('total_comments').desc()
)

qs = Movie.objects.annotate(total_comments=Count('comment')).annotate(rank=dense_rank)

但是运行此查询会引发django.db.utils.OperationalError: near "(": syntax error

Answer 1

如果查询的排序正确，

RowNumber就足够了：

from django.db.models.expressions import F, Window
from django.db.models.functions.window import RowNumber

qs = (Movie.objects
    .annotate(total_comments=Count('comment'))
    .order_by('total_comments')
    .annotate(rank = Window(expression=RowNumber())
)

但是由于行号是记录序列中固有的，因此Kevin建议使用简单的Python方法（例如enumerate）是正确的，因为无论如何您都将在记录集上进行迭代。

如果您希望记录集的排序与排名的排序不同，那么将Window与单独的order_by参数一起使用是有意义的。

我不确定所有后端是否支持RowNumber。

Django ORM：根据相关模型计数创建排名

1 个答案: