在Django queryset上注释以逗号分隔的相关项列表

Question

从表面上看,这似乎是一个奇怪的要求,但我需要一个逗号分隔的模型相关项目串.从任何教程中获取作者/书籍模型示例,这是我目前正在做的事情:

authors = Authors.objects.all().prefetch_related('books')
for author in authors:
    author.book_titles = ', '.join([book.title for book in author.books.all()])

它并不重,但感觉多余.就像数据库可以做到的那样.在一个理想的世界里,我觉得我应该能够用这些奇特的新数据库功能之一来注释它 .这是一个幻想的例子,使用一个名为的伪造函数Joiner(..):

Authors.objects.annotate(book_titles=Joiner('books__title', separator=', ')

那可能吗？如果是这样,怎么样？

Answer 1

from django.db.models import Aggregate, CharField, Value

class GroupConcat(Aggregate):
    function = 'GROUP_CONCAT'
    template = '%(function)s(%(expressions)s)'

    def __init__(self, expression, delimiter, **extra):
        output_field = extra.pop('output_field', CharField())
        delimiter = Value(delimiter)
        super(GroupConcat, self).__init__(
            expression, delimiter, output_field=output_field, **extra)

    def as_postgresql(self, compiler, connection):
        self.function = 'STRING_AGG'
        return super(GroupConcat, self).as_sql(compiler, connection)

用法:

Author.objects.annotate(book_titles=GroupConcat('book__title', ', '))

自定义聚合.这应该适用于SQLite,MySQL和PostgreSQL.