Django在表单中使用多对多

时间:2018-06-26 12:56:12

标签: django django-models django-forms django-related-manager

我有一个数据模型,其中我使用一个手动中间表来建立m2m关系。 以django doc中的经典示例为基础:

from django.db import models

INSTRUMENT_CHOICES = (
    ('guitar', 'Guitar'),
    ('bass', 'Bass Guitar'),
    ('drum', 'Drum'),
    ('keyboard', 'Keyboard'),
)

class Person(models.Model):
    name = models.CharField(max_length=128)

    def __str__(self):
        return self.name

class Group(models.Model):
    name = models.CharField(max_length=128)
    members = models.ManyToManyField(Person, through='Leadership')

    def __str__(self):
        return self.name

    def get_leadership():
        return self.leadership_set.first()


class Leadership(models.Model):
    person = models.ForeignKey(Person, on_delete=models.CASCADE)
    group = models.ForeignKey(Group, on_delete=models.CASCADE)
    instrument = models.CharField('Playing Instrument', choices=INSTRUMENT_CHOICES, 
                                   max_length=15, 
                                   null=True, 
                                   blank=False)
    class Meta:
        unique_together = ('person', 'group')

当我创建一个新的组时,我还想指定谁将成为组长,并且为此关系,还要指定他将在该组中扮演哪种乐器。

考虑到缺乏有关该主题的文档,真正令我困惑的是如何处理表单中的这种关系。

这是我随附的表格:

class InstrumentField(forms.ChoiceField):
    def __init__(self, *args, **kwargs):
        super().__init__(INSTRUMENT_CHOICES, *args, **kwargs)


class GroupForm(forms.ModelForm):
    instrument = InstrumentField(required=False)

    class Meta:
        model = Group
        fields = ['name', 
                  'members'
                  'instrument']  # This works but it's not correctly initalized in case of edit form
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        if self.instance.pk is not None:  # editing 
            # PROBLEM: this doesn't work
            self.fields["instrument"].initial = self.instance.get_leadership().instrument

    def save(self, commit=True):
        group = super().save(commit=False)
        if commit:
            group.save()
            if 'instrument' in self.changed_data:
                leader = self.cleaned_data.get('members').first()  
                instrument = self.cleaned_data['instrument']

                Leadership.objects.update_or_create(person=leader, group=group, defaults={'instrument': instrument})

        return group

按照Django文档中的建议,我正在手动实例化Leadership对象(请参见表单save方法)。

我无法解决的是在进行表单编辑时如何填充instrument字段。我尝试在__init__中执行此操作:首先,我检查我们是否处于“编辑”模式(实例具有pk),然后获取相关的Leadership对象(请参见Group.get_leadership)然后从中提取instrument,然后将其分配给fields["instrument"].initial

这不起作用。

我可以检查是否设置了initial值,但是当我渲染表单时,将显示默认选择值(INSTRUMENT_CHOICES的第一个值)。

我在这里想念什么? 是否有更好的方法或更好的文档介绍如何使用表单中的通过模型处理m2m?

0 个答案:

没有答案