如何使用 django 获取每个作者最新的 3 本书

Question

使用以下 django 模型：

class Author(models.Model):
   name = models.CharField(max_length=100)
   age = models.IntegerField()

class Book(models.Model):
    name = models.CharField(max_length=300)
    author = models.ForeignKey(Author)
    pubdate = models.DateField()
    class Meta:
        ordering = ('-pubdate')

如何获取每位作者最新出版的五本书？

我曾考虑过迭代每个作者并将该作者出版的书籍切片为 5。

for a in Author.objects.all():
    books = Book.objects.filter(author = a )[:5]
    print books #and/or process the entries... 

但是，如果表有很多记录（可能有数千本书），这可能会很慢且效率低下。

那么，有没有其他方法可以使用 django （或 sql 查询）来完成此任务？

Answer 1

我建议：

for a in Author.objects.all():
    books = a.book_set.all().order_by('-pub_date')[:5]
    print books #and/or process the entries... 

或者，如果顺序应该始终相同，正如您定义的 Meta 那样，

    books = a.book_set.all()[:5]

应该可以解决问题

Answer 2

如果您担心查询的速度，请在 pubdate 字段上构建索引：

pubdate = models.DateField(db_index=True)

这应该避免每次运行查询时扫描整个表。

postgres 中的 SQL 类似于：

select b1.name, b1.author
from books b1
where b1.id in (
    select b2.id
    from books b2
    where b1.author = b2.author
    order by b2.pubdate desc
    limit 3)
order by b1.author, b1.name

Answer 3

对我来说，我有‍‍ 3 位作者，每个作者都至少有 300 篇帖子。
我测试了两种方法，差异非常引人注目！
根据django-debug-toolbar显示，第一种方法有3个类似的查询，其中3是这里的作者数量，它需要总共 8 个查询（+180 毫秒） 和 CPU（400 - 800 毫秒），但第二个总共需要 5 个查询（+80 毫秒） ) 和 CPU (+70 - 90 ms)。

• 我还看到有一个额外的查询SELECT ... FROMauthor，我无法弄清楚为什么它被生成它似乎根本没有被使用。

from .models import Author, Po‍‍st

最简单和最熟悉的方法

from itertools import chain

authors = Author.objects.all().prefetch_related("posts")

posts = list(chain.from_iterable(author.posts.order_by("-created")[:3]
             for author in authors)
        )

更复杂和不熟悉的方法

from django.db.models import OuterRef, Prefetch, Subquery

subquery = Subquery(
        Post.objects.filter(author__id=OuterRef("author__id"))
        .order_by("-created")
        .values_list("id", flat=True)[:4]
)

authors = Author.objects.all().prefetch_related(
    Prefetch("posts", queryset=Post.objects.filter(id__in=subquery))
)

posts = list(chain.from_iterable(source.posts.all()
            for source in feed_sources)
)

如果您只需要帖子而不是作者：

以下获取数据的方式会少做一次查询，并且只返回帖子。

subquery = Subquery(
    Post.objects.filter(author__id=OuterRef("author__id"))
    .order_by("-created")
    .values_list("id", flat=True)[:4]
)

posts = (
    Post.objects.select_related("author")
    .exclude(author=an_author)
    .filter(id__in=subquery)
)