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'),
)