Джанго косвенное родовое отношение - PullRequest
Новая
       19

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

2 голосов
/ 02 марта 2012

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

Я пытался установить тип контента для модели тегов и использовать этозначение в модели 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 Ответ

3 голосов
/ 02 марта 2012

Ошибка здесь: 

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

...