我如何使用GenericForeignKeys注解具有GenericRelations的django-polymorphic模型到其他模型?

a1o7rhls  于 2023-01-21  发布在  Go
关注(0)|答案(1)|浏览(113)

我有一个名为Content的父模型,它继承自Django polymorphic。这是一个简化的示例,但我有一个继承自ContentPost模型。
Content模型上,注意我有一个名为notesGenericRelation(Note)

    • 我尝试做的是**用一个notes的计数来注解所有的Content对象,这和你在下面的for循环中得到的结果完全一样。
for content in Content.objects.all():
    print(content.notes.count())

下面是一个完全可复制的简化示例。

重现问题

1.设置新的Django项目,创建超级用户,添加django-polymorphic到项目中,复制/粘贴模型。进行迁移和迁移。我的应用程序被称为myapp
1.打开www.example.com shell,导入Post模型,然后运行Post.make_entries(n=30) manage.py shell, import Post model, and run Post.make_entries(n=30)
1.运行Post.notes_count_answer(),它将返回一个数字列表。这些数字是带注解的Content PolymorphicQuerySet应该显示的。例如:

Post.notes_count_answer()
[3, 2, 3, 1, 3, 1, 3, 1, 2, 1, 2, 2, 3, 3, 3, 1, 3, 3, 2, 3, 2, 3, 2, 1, 2, 1, 1, 1, 1, 2]

列表中的第一个数字3表示第一个Post有3个notes

我尝试了哪些方法(从简单到复杂)

基本的

>>> Content.objects.all().annotate(notes_count=Count('notes')).values('notes_count')
<PolymorphicQuerySet [{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, '...(remaining elements truncated)...']>

求救信号/微弱的尝试

Content.objects.all().prefetch_related('notes').annotate(notes_count=Count('notes')).values('notes_count')
<PolymorphicQuerySet [{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, {'notes_count': 0}, 
{'notes_count': 0}, '...(remaining elements truncated)...']>

子查询?

>>> Content.objects.all().annotate(notes_count=Subquery(
Note.objects.filter(object_id=OuterRef('pk'), content_type_id=OuterRef('polymorphic_ctype_id')).order_by(
    'object_id').annotate(c=Count('object_id')).values('c'))).values('notes_count')

<PolymorphicQuerySet [{'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, 
{'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, 
{'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, 
{'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, 
{'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, {'notes_count': 1}, 
{'notes_count': 1}, '...(remaining elements truncated)...']>

很近?

Content.objects.all().annotate(
    notes_count=Count(Subquery(
        Note.objects.filter(
            object_id=OuterRef('pk'), content_type_id=OuterRef('polymorphic_ctype_id')
        ).order_by('object_id')))).values('notes_count')

# error message
line 357, in execute
    return Database.Cursor.execute(self, query, params)
django.db.utils.OperationalError: sub-select returns 4 columns - expected 1

我确实尝试了Subquery的许多不同变体,但一直无法在注解中获得正确的音符计数。

预期结果:

你的不会是精确的,但数据是生成的,但这是一个想法。

<PolymorphicQuerySet [{'notes_count': 3}, {'notes_count': 2}, {'notes_count': 3}, 
{'notes_count': 1}, {'notes_count': 3}, {'notes_count': 1}, {'notes_count': 3}, 
{'notes_count': 1}, {'notes_count': 2}, {'notes_count': 1}, {'notes_count': 2}, 
{'notes_count': 2}, {'notes_count': 3}, {'notes_count': 3}, {'notes_count': 3}, 
{'notes_count': 1}, {'notes_count': 3}, {'notes_count': 3}, {'notes_count': 2}, 
{'notes_count': 3}, {'notes_count': 2}, {'notes_count': 3}, {'notes_count': 2}, 
{'notes_count': 1}, '...(remaining elements truncated)...']>

要求. txt

Django==4.1.5
django-polymorphic==3.1.0

网站settings.py

INSTALLED_APPS = [
    'django.contrib.admin',
    'django.contrib.auth',
    'polymorphic',
    'django.contrib.contenttypes',
    'django.contrib.sessions',
    'django.contrib.messages',
    'django.contrib.staticfiles',

    'myapp.apps.MyappConfig',
]

网站models.py

from django.contrib.contenttypes.fields import GenericRelation, GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.core.validators import MaxValueValidator, MinValueValidator
from django.db import models
from django.conf import settings
from polymorphic.models import PolymorphicModel

from django.contrib.auth import get_user_model

class Vote(models.Model):

    value = models.IntegerField(default=0, validators=[MinValueValidator(-1), MaxValueValidator(1)])

    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    content_object = GenericForeignKey('content_type', 'object_id')

    def __str__(self):
        return str(self.value)

class Note(models.Model):
    body = models.TextField()

    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    content_object = GenericForeignKey('content_type', 'object_id')

    def __str__(self):
        return str(self.id)

class Content(PolymorphicModel):
    user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
    title = models.CharField(max_length=100)
    votes = GenericRelation(Vote)  # reverse generic relation
    notes = GenericRelation(Note)  # reverse generic relation

    def __str__(self):
        return str(self.pk)

class Post(Content):
    content = models.TextField(blank=True)

    def __str__(self):
        return str(self.pk)

    @staticmethod
    def make_entries(n=5):
        import random
        user = get_user_model().objects.first()
        for i in range(1, n+1, 1):
            vote_count = random.randrange(0, 5)
            note_count = random.randrange(0,3)
            p = Post.objects.create(
                user=user,
                title=f'Post #{i}',
                content=f'Content for post {i}',
            )
            content_type = ContentType.objects.get_for_model(p)
            Vote.objects.create(
                value=vote_count,
                content_type=content_type,
                object_id=p.id
            )
            for j in range(note_count + 1):
                Note.objects.create(
                    body=f'Note {j}',
                    object_id=p.id,
                    content_type=content_type
                )

    @staticmethod
    def notes_count_answer():
        return [content.notes.count() for content in Content.objects.all()]
mrphzbgm

mrphzbgm1#

成功了。我想关键是知道Subquery需要返回一个值(一个计数),并在Subquery内部执行计数。我经常摆弄Count()函数,并把头撞到墙上。

from django.db.models import Count, OuterRef, Subquery
from django.db.models.functions import Coalesce

Content.objects.annotate(notes_count=Coalesce(
    Subquery(
        Note.objects.filter(
            object_id=OuterRef('pk'),
        ).order_by('object_id').values('object_id').annotate(
            count=Count('object_id')
        ).values('count')
    ), 0
))

相关问题