Django中内联表单集的外键限制查询集

问题描述

我已经构建了一个程序(使用Django 1.9)来跟踪锦标赛。每场锦标赛由一系列比赛组成,每一场比赛都有两个人(拳击手)与之关联。

锦标赛有一个"COMPARATANT_POOL",它包含所有参赛对象的子集。该界面当前允许我从‘STARATANT_POOL’中添加/删除战斗人员。

问题出在允许我记录比赛的屏幕上。我可以毫无问题地查看/添加/删除/修改回合,但输入回合的问题在于,允许我选择‘COMPATANT_1’和‘COMPATANT_2’的下拉列表允许我从数据库中的任何选手中进行选择,我需要的是只列出‘COMPATANT_POOL’中的选手。

我查看了许多与此问题相关的其他论坛,但似乎没有一个论坛能帮助我解决此问题。

class combatant(models.Model):
    first_name = models.CharField(max_length=100)

class tournament(models.Model):
    combatant_pool = models.ManyToManyField(combatant, blank=True)

class bout(models.Model):
    parent_tournament = models.ForeignKey(tournament, on_delete=models.CASCADE)
    combatant_1 = models.ForeignKey(combatant, on_delete=models.CASCADE, related_name='combatant1')
    combatant_2 = models.ForeignKey(combatant, on_delete=models.CASCADE, related_name='combatant2')
    outcome = models.CharField(max_length=10)   


def BoutsView(request, pk):
    ThisTournament = tournament.objects.get(id=pk)

    BoutInlineFormSet = inlineformset_factory(tournament, bout, fields=('combatant_1', 'outcome', 'combatant_2'), formset=BaseInlineFormSet)

    if request.method == "POST":
        formset = BoutInlineFormSet(request.POST, request.FILES, instance=ThisTournament)
        if formset.is_valid():
            formset.save()
            # Do something.
            return HttpResponseRedirect('/TournamentTracker/' + str(pk) + '/bouts')
    else:
        formset = BoutInlineFormSet(instance=ThisTournament)

    return render(request, 'tournament_bouts_update_form.html', {'formset': formset, 'pk': pk})

解决方案

啊,终于找到了可行的解决方案。

我只需将此方法添加到视图:

    def get_field_qs(field, **kwargs):
        if field.name in ['combatant_1', 'combatant_2']:
            return forms.ModelChoiceField(queryset=Tournament.objects.get(id=pk).combatant_pool)
        return field.formfield(**kwargs)

然后添加参数

formfield_callback=get_field_qs

到inlineformset_Factory调用。

相关文章