在 django vanilla CreateView 上设置当前用户 [英] Setting current user on django vanilla CreateView

问题描述

我想用当前登录的用户更新我的模型.我正在使用 django-vanilla-views.为了存储新记录，我正在尝试使用 CreateView.我不想在表单上显示用户，只需自动更新即可.

I would like to update my model with the currently logged in user. I am using django-vanilla-views. To store a new record I am trying to use CreateView. I don't want to display user on the form, just update it automatically.

这是我的模型:

class Measurement(models.Model):
    date = models.DateField()
    user = models.ForeignKey(User)

这是我的观点:

class CreateMeasurement(CreateView):
    model = Measurement
    fields = ['date']
    success_url = reverse_lazy('list_measurements')

    def get_form(self, data=None, files=None, **kwargs):
        kwargs['user'] = self.request.user
        return super(CreateMeasurement, self).get_form(data=data, files=files, **kwargs)

不幸的是，在访问视图时出现以下异常:

Unfortunately when accessing the view I get the following exception:

TypeError: __init__() got an unexpected keyword argument 'user'

我还尝试为我的模型创建一个 ModelForm，但得到了完全相同的错误.任何想法我可能做错了什么?

I also tried to create a ModelForm for my model but got exactly the same error. Any ideas what I might be doing wrong?

推荐答案

您不需要将用户传递给表单，因此不要覆盖 get_form 方法.您已经通过在视图中设置 fields 从模型表单中排除了用户 field，因此您也不应该需要自定义模型表单.

You don't need to pass the user to the form, so don't override the get_form method. You have already excluded the user field from the model form by setting fields in your view, so you shouldn't need a custom model form either.

重写form_valid方法应该就够了，并在保存表单时设置用户.

It should be enough to override the form_valid method, and set the user when the form is saved.

from django.http import HttpResponseRedirect

class CreateMeasurement(CreateView):
    model = Measurement
    fields = ['date']
    success_url = reverse_lazy('list_measurements')

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

这篇关于在 django vanilla CreateView 上设置当前用户的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！