GenericForeignKey получает неправильный идентификатор при использовании с моделью с UUIDField

Когда используешь GenericForeignKey вместе с UUIDFieldКаков рекомендуемый способ получить набор запросов "реальной модели" из набора запросов общих объектов?

Вот модели, с которыми я тестирую:

import uuid
from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.db import models

class Foo(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4)

class Generic(models.Model):
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.CharField(max_length=255)
    content_object = GenericForeignKey()

и это то, что я пробовал до сих пор:

>>> from django.db.models import Subquery
>>> from foo.models import Foo, Generic

>>> f = Foo.objects.create()
>>> g = Generic.objects.create(content_object=f)
>>> Foo.objects.filter(id__in=Subquery(Generic.objects.all().values('object_id')))
<QuerySet []>

>>> Generic.objects.get().object_id
'997eaf64-a115-4f48-b3ac-8cbcc21274a8'
>>> Foo.objects.get().pk
UUID('997eaf64-a115-4f48-b3ac-8cbcc21274a8')

Я предполагаю, что это связано с сохранением UUID без дефисов для UUIDField, Я не могу сделать object_id в UUIDField либо потому, что мне нужны другие модели, которые имеют целые числа и строки в качестве первичных ключей.

Я использую Django 1.11, но я также тестировал Django 2.0, который имеет ту же проблему.

2 ответа

Решение

Главная проблема в explicit type casts

Итак, идея @Alasdair, вы можете попробовать:

foo_content_type = ContentType.objects.get_for_model(Foo)
gids = Generic.objects.filter(content_type=foo_content_type)
# create list of uuid strings
gids = list(gids.values_list('object_id', flat=True))
Foo.objects.filter(pk__in=gids)

Другое решение: вы можете добавить uuid поле к Generic моделей. Например:

class Generic(models.Model):
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.CharField(max_length=255)
    content_object = GenericForeignKey()
    uuid4 = models.UUIDField(blank=True, null=True)

    def save(self, *args, **kwargs):
        try:
            self.uuid4 = uuid.UUID(self.object_id)
        except Exception as e:
            pass
        super().save(*args, **kwargs)

и запрос будет выглядеть так:

foo_content_type = ContentType.objects.get_for_model(Foo)
gids = Generic.objects.filter(content_type=foo_content_type).values('uuid4')
Foo.objects.filter(pk__in=gids)

Попробуйте удалить Subquery(), Вы также хотите фильтровать по типу контента.

foo_content_type = ContentType.objects.get_for_model(Foo)
Foo.objects.filter(
    id__in=Generic.objects.filter(content_type=foo_content_type).values('object_id'), 
)
Другие вопросы по тегам