我一直在尝试为我的应用程序实现标记系统,其中每种内容类型只允许使用某些标记。
我尝试在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?
答案 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字段完成