Джанго косвенное родовое отношение
Я пытался реализовать систему тегов для моего приложения, где разрешены только определенные теги для каждого типа контента.
Я попытался установить тип содержимого для модели Tag и использовать это значение в модели TagAttribution, и получил... интересные результаты.
Код:
from django.db import models
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes import generic
from django.contrib.auth.models import User
class Tag(models.Model):
value = models.CharField(max_length=32)
created_by = models.ForeignKey(User)
appliable_to = models.ForeignKey(ContentType)
def __unicode__(self):
return self.value
class TagAttribution(models.Model):
tag = models.ForeignKey(Tag)
object_id = models.PositiveIntegerField()
content_object = generic.GenericForeignKey('tag__appliable_to', 'object_id')
def __unicode__(self):
return "%s for id %s of class %s" % (self.tag.value, self.object_id, self.content_object.model)
Оболочка теста:
ct = ContentType.objects.get(model='company')
tag = Tag()
tag.value = 'value'
tag.created_by = User.objects.get(id=1)
tag.appliable_to = ct
tag.save()
ta = TagAttribution()
ta.tag = tag
ta.object_id = Company.objects.get(id=1).id
ta.content_object = ta.tag.appliable_to
ta.save()
ta
Выход:
<TagAttribution: value for id 13 of class company>
Я не понимаю это поведение; почему он получил идентификатор 13, если я использовал идентификатор компании 1?
1 ответ
Решение
Ошибка здесь:
ta.content_object = ta.tag.appliable_to
Объект ta.content_object здесь не является объектом Company, а ContentType. Правильный код должен быть:
ta.content_object = Company.objects.get(id=1).id
Кроме того, вам не нужно устанавливать ta.object_id напрямую, это делается полем GenericForeignKey