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

时间:2016-01-04 04:26:49

标签: python django django-models django-queryset

我已经构建了一个程序(使用Django 1.9)来跟踪锦标赛。每场比赛都包含一系列比赛,每场比赛都有两个人(战斗员)与之相关。

锦标赛有一个'combatant_pool',其中包含所有战斗对象的子集。该界面目前允许我从'combatant_pool'添加/删除战斗员。

问题在于允许我记录发作的屏幕。我可以毫无问题地查看/添加/删除/修改回合,但输入回合的问题是允许我选择'combatant_1'和'combatant_2'的下拉菜单允许我从数据库中的任何战斗员中选择,以及我需要什么是'battleant_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})

1 个答案:

答案 0 :(得分:4)

啊,终于找到了一个有效的解决方案。

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

    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调用。