8

假设我有一个简单的模型：

class Contact(models.Model):
    owner = models.ForeignKey(User, editable=False)
    first_name = models.CharField(max_length=255,)
    last_name = models.CharField(max_length=255,)
    email = models.EmailField()

我想在创建对象时自动为其设置所有者（request.user，登录用户）。我搜索了很多不同的选项，但其中大多数与您在管理方面的操作方式有关，而其他选项对我不起作用。我试过这个，例如http://blog.jvc26.org/2011/07/09/django-automatically-populate-request-user然后我尝试了很多方法来覆盖保存方法或某种 pre_save 信号的东西。似乎没有什么能解决问题，我只是得到一个错误

IntegrityError at /new
null value in column "owner_id" violates not-null constraint

这样做的正确方法是什么？我知道这很简单，但我就是找不到方法。

...编辑...我的创建视图如下所示：

class CreateContactView(LoginRequiredMixin, ContactOwnerMixin, CreateWithInlinesView):
    model = models.Contact
    template_name = 'contacts/edit_contact.html'
    form_class = forms.ContactForm
    inlines = [forms.ContactAddressFormSet]

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

    def get_success_url(self):
        return reverse('contacts-list')

    def get_context_data(self, **kwargs):
        context = super(CreateContactView, self).get_context_data(**kwargs)
        context['action'] = reverse('contacts-new')
        return context

到目前为止，这只是我试图解决该问题的一种方法。我从http://blog.jvc26.org/2011/07/09/django-automatically-populate-request-user找到了该解决方案

4

3 回答 3

6

假设您正在使用ContactForm ModelForm：

def contact(request):
    if request.method == 'POST':
        form = ContactForm(request.POST)
        if form.is_valid():
            contact = form.save(commit=False)
            contact.owner = request.user
            contact.save()
            return HttpResponseRedirect('/thanks/')
    else:
         # do stuff
于 2014-03-03T16:26:16.773 回答
0

请发布您尝试过的确切代码。

如果您的视图要求用户登录，请确保已强制执行。这可以通过使用@login_required装饰器来完成

如果您在视图中，并使用 aModelForm创建Contact，则将commit=Falsekwarg 传递给 save 方法（如您发布的链接中的示例）。这将阻止创建联系人，直到您分配owner = request.user.

由于登录用户仅在请求的上下文中可用，因此只需确保owner在创建新的视图时将其设置为视图Contact

于 2014-03-03T16:09:57.700 回答
0

问题是该form_valid方法的默认实现 sets self.object，然后用于get_success_url确定重定向到的位置。

如果你用 替换你的局部obj变量self.object，你应该没问题：

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

我在Classy Class-Based Views网站上快速检查了副作用的原始实现，或者GitHub 上的 Django 源代码对于发现我需要在子类实现中重现的任何副作用很有用。

于 2016-04-04T21:07:08.483 回答