从 Django 中基于类的通用视图将 request.user 对象发送到 ModelForm

Question

因此，我的目标是能够过滤 ModelForm 中的 ModelChoiceField 查询集，以仅包含 request.user 创建的地点。

我的 ModelForm 很简单：

class PlaceEventForm(models.ModelForm):
    class Meta:
        model = Event

我希望能够添加类似的内容：

def __init__(self, *args, **kwargs):
    super(PlaceEventForm, self).__init__(*args, **kwargs)
    self.fields['place'].queryset = Place.objects.filter(created_by=request.user)

但是，我似乎找不到访问 ModelForm 中的请求的方法。

我的视图是这样的：

class PlaceEventFormView(CreateView):
    form_class = PlaceEventForm
    template_name = 'events/event_create.html'

    @method_decorator(login_required)
    def dispatch(self, *args, **kwargs):
        return super(PlaceEventFormView, self).dispatch(*args, **kwargs)

我不确定这是否接近我应该做的，但我尝试过：

def get_form_kwargs(self):
    kwargs = super(PlaceEventFormView, self).get_form_kwargs()
    kwargs.update({'place_user': self.request.user})
    return kwargs

但我收到错误： init() 得到了意外的关键字参数 'place_user'

对这个有什么想法吗？或者任何人都可以想出一种方法来过滤视图中的 ModelChoiceField 而无需将我的请求传递给 ModelForm？

Answer 1

您需要在 PlaceEventForm.__init__() 方法中从 kwargs 中弹出关键 user，以防止其转到 ModelForm.__init__( ) 方法：

views.py:

class PlaceEventFormView(CreateView):
    form_class = PlaceEventForm
    template_name = 'events/event_create.html'

    @method_decorator(login_required)
    def dispatch(self, *args, **kwargs):
        return super(PlaceEventFormView, self).dispatch(*args, **kwargs)

    def get_form_kwargs(self):
        kwargs = super(PlaceEventFormView, self).get_form_kwargs()
        kwargs.update({'place_user': self.request.user})
        return kwargs

forms.py:

class PlaceEventForm(models.ModelForm):
    class Meta:
        model = Event

    def __init__(self, *args, **kwargs):
        user = kwargs.pop('place_user')
        # now kwargs doesn't contain 'place_user', so we can safely pass it to the base class method
        super(PlaceEventForm, self).__init__(*args, **kwargs)
        self.fields['place'].queryset = Place.objects.filter(created_by=user)

Answer 2

我使用的是 iPhone，但这样做：

def get_form(self, form_class):
     form = super(MyView, self).get_form(form_class)
     form.fields['place'].querset = Place....
     return form

哇，这太难了！不支持缩进！

Answer 3

要更新 Yuji 对 Django 1.10+（包括 Django 2.0+）的答案，请参阅下面的示例（注意更新的方法签名）。 Yuji 建议的方法将查询集与其他业务逻辑一起保留在视图中，并有助于保持扩展 models.ModelForm 的任何表单类简洁明了。

def get_form(self, form_class=None):
    if form_class is None:
        form_class = self.get_form_class()
    form = super(MyView, self).get_form()
    form.fields['place'].queryset = Place.objects.filter(created_by=self.request.user)
    return form

较短：

def get_form(self, form_class=None):
    form = super(MyView, self).get_form(form_class)
    form.fields['place'].queryset = Place.objects.filter(created_by=self.request.user)
    return form