在基于类的通用视图CreateView中访问request.user,以便在Django中设置FK字段

时间:2011-04-26 03:53:41

标签: django django-generic-views django-class-based-views

所以我的模型包括:

class Place(models.Model):
    ....
    created_by = models.ForeignKey(User)

我的观点是这样的:

class PlaceFormView(CreateView):
    form_class = PlaceForm

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

我有办法访问request.user并将created_by设置为该用户吗?我查看了文档,但似乎无法找到任何暗示。

`

3 个答案:

答案 0 :(得分:27)

覆盖保存表单的form_valid怎么样?自己保存,随心所欲,然后进行重定向。

class PlaceFormView(CreateView):
    form_class = PlaceForm

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

    def form_valid(self, form):
        obj = form.save(commit=False)
        obj.created_by = self.request.user
        obj.save()        
        return http.HttpResponseRedirect(self.get_success_url())

答案 1 :(得分:13)

我知道这是旧的,但对于有这个问题的其他人:

有一种更简单的方法 - 因为多次保存表单将始终使用相同的模型实例,您也可以这样做:

def form_valid(self, form):
    obj = form.save(commit=False)
    obj.created_by = self.request.user
    return super(PlaceFormView, self).form_valid(form)

这样,你就可以获得超级电话的所有好处 - 看到你真的只是添加这两行代码,而且你不必重复自己,这是微不足道的通过复制重定向逻辑。

答案 2 :(得分:2)

另一种方法是让用户通过覆盖CreateView中的get_initial()方法,并修改PlaceForm类中的save方法以保存用户:

class PlaceForm(forms.ModelForm):
    ...
    ...
    ...

    def __init__(self, *args, **kwargs):
        self.created_by = kwargs['initial']['created_by']
        super(PlaceForm, self).__init__(*args, **kwargs)

    def save(self, commit=True):
        obj = super(PlaceForm, self).save(False)
        obj.created_by = self.created_by
        commit and obj.save()
        return obj

class PlaceFormView(CreateView):
    ...
    ...
    form_class = PlaceForm

    def get_initial(self):
        self.initial.update({ 'created_by': self.request.user })
        return self.initial

这样,保存逻辑仍然封装在表单类中。

相关问题