在Django中自动填充一组通用的多个字段?

时间:2015-02-10 16:49:11

标签: python django django-contenttypes

我正在尝试将this回答与此one结合起来,并进行一些循环。

在创建角色时,我想添加值为0的所有可能技能,但我对如何遵循上述答案感到困惑。

我有这个混音:

class CrossCharacterMixin(models.Model):
    cross_character_types = models.Q(app_label='mage', model='mage')
    content_type = models.ForeignKey(ContentType, limit_choices_to=cross_character_types,
                                     null=True, blank=True)
    object_id = models.PositiveIntegerField(null=True)
    content_object = GenericForeignKey('content_type', 'object_id')

    class Meta:
        abstract = True

(最终,cross_character_types将被扩展)

这个模型:

class CharacterSkillLink(Trait, CrossCharacterMixin):
    PRIORITY_CHOICES = (
        (1, 'Primary'), (2, 'Secondary'), (3, 'Tertiary')
    )
    skill = models.ForeignKey('SkillAbility')
    priority = models.PositiveSmallIntegerField(
        choices=PRIORITY_CHOICES, default=None)
    speciality = models.CharField(max_length=200, null=True, blank=True)

    def __str__(self):
        spec_string = " (" + self.speciality + ")" if self.speciality else ""
        return self.skill.skill.label + spec_string

我开始写的是NWODCharacter模型:

def save(self, *args, **kwargs):
    if not self.pk:
        character_skills_through = CharacterSkillLink.content_object.model

        CharacterSkillLink.objects.bulk_create([
            [character_skills_through(skill=SkillAbility(
                skill), content_object=self) for skill in SkillAbility.Skills]
        ])

    super(NWODCharacter, self).save(*args, **kwargs)

这不起作用,因为我认为我没有传递正确的物体。

虽然基于此answer

from django.db import models

class Users(models.Model):
    pass

class Sample(models.Model):
    users = models.ManyToManyField(Users)

Users().save()
Users().save()

# Access the through model directly
ThroughModel = Sample.users.through

users = Users.objects.filter(pk__in=[1,2])

sample_object = Sample()
sample_object.save()

ThroughModel.objects.bulk_create([
    ThroughModel(users_id=users[0].pk, sample_id=sample_object.pk),
    ThroughModel(users_id=users[1].pk, sample_id=sample_object.pk)
])

在这种情况下,我的ThroughModel是什么?是CharacterSkillLink.content_object.model吗?

我如何在我的场景中执行此操作?我很抱歉,如果这是微不足道的,但我很难理解它。

1 个答案:

答案 0 :(得分:1)

在我看来,在这种情况下,CharacterSkillLink本身就是您的直通模型......它通常将内容类型加入SkillAbility

如果您考虑一下,如果您正在执行bulk_create,那么您传入的对象必须与您正在进行bulk_create的模型相同。< / p>

所以我认为你想要这样的东西:

def save(self, *args, **kwargs):
    initialise_skill_links = not self.pk
    super(NWODCharacter, self).save(*args, **kwargs)
    if initialise_skill_links:
        CharacterSkillLink.objects.bulk_create([
            CharacterSkillLink(
                skill=SkillAbility.objects.get_or_create(skill=skill)[0],
                content_object=self
            )
            for skill in SkillAbility.Skills
        ])

请注意,[]内有bulk_createSkillAbility.objects.get_or_create()太多了。

另外我认为你应该使用SkillAbility() ...作为外键,你需要相关的对象存在。如果它已经存在,那么只是执行{{1}}将无法从数据库中获取它,如果不存在则不会将其保存到数据库中。