Django间接通用关系

时间:2012-03-01 21:51:49

标签: python django tags generic-relations

我一直在尝试为我的应用程序实现标记系统,其中每种内容类型只允许使用某些标记。

我尝试在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>

我不明白这种行为;如果我使用公司ID 1,它为什么会得到id 13?

1 个答案:

答案 0 :(得分:3)

错误在这里:

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字段完成