Джанго косвенное родовое отношение

Я пытался реализовать систему тегов для моего приложения, где разрешены только определенные теги для каждого типа контента.

Я попытался установить тип содержимого для модели 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

Другие вопросы по тегам