如何使用ModelForm过滤Django表单中的值？

Question

我正在尝试使用ModelForm添加数据。它工作正常，除了ForeignKey下拉列表显示所有值，我只希望它显示与已登录用户相关的值。

这是我要添加的记录ExcludedDate的模型：

class ExcludedDate(models.Model):
    date = models.DateTimeField()
    reason = models.CharField(max_length=50)
    user = models.ForeignKey(User)
    category = models.ForeignKey(Category)
    recurring = models.ForeignKey(RecurringExclusion)

    def __unicode__(self):
        return self.reason

这是类别的模型，该表是包含我想由用户限制的关系的表：

class Category(models.Model):
    name = models.CharField(max_length=50)
    user = models.ForeignKey(User, unique=False)

    def __unicode__(self):
        return self.name

最后，表单代码：

class ExcludedDateForm(ModelForm):

    class Meta:
        model = models.ExcludedDate
        exclude = ('user', 'recurring',)

如何获取仅显示类别子集的表单，其中category.user等于登录用户？

Answer 1

您可以在 init中* 自定义表格 *

class ExcludedDateForm(ModelForm):
    class Meta:
        model = models.ExcludedDate
        exclude = ('user', 'recurring',)
    def __init__(self, user=None, **kwargs):
        super(ExcludedDateForm, self).__init__(**kwargs)
        if user:
            self.fields['category'].queryset = models.Category.objects.filter(user=user)

在视图中，在构建表单时，除了标准表单参数外，您还将指定当前用户：

form = ExcludedDateForm(user=request.user)