在Django表单中获取当前用户

36

可能是重复问题:
如何在Django表单中获取请求数据

这是我的Guest模型的一部分:

class Guest(models.Model):
    event = models.ForeignKey(Event, related_name='guests')
    user = models.ForeignKey(User, unique=True, related_name='guests')
    ...

获取来自客人响应的表单:

class RSVPForm(forms.Form):
    attending_d= forms.ChoiceField(choices=VISIBLE_ATTENDING_CHOICES, initial='yes', widget=forms.RadioSelect)
    attending_b = forms.ChoiceField(choices=VISIBLE_ATTENDING_CHOICES, initial='yes', widget=forms.RadioSelect)
    number_of_guests = forms.IntegerField(initial=0)
    comment = forms.CharField(max_length=255, required=False, widget=forms.Textarea)

    ....    

    def save(self):
        guest = self.guest_class.objects.get(user=1)
        guest.attending_status_d = self.cleaned_data['attending_d']
        guest.attending_status_b = self.cleaned_data['attending_b']
        guest.number_of_guests = self.cleaned_data['number_of_guests']
        guest.comment = self.cleaned_data['comment']
        guest.save()
        return guest

问题出在save方法中。我怎样能将来宾与当前登录的用户相关联?

guest = self.Guest.objects.get(user=1)

我需要使用当前已登录用户的ID,而不是user=1

谢谢!

1个回答

57

我找到了方法:)

  1. 在表单上编写一个__init__方法:
    def __init__(self, user, *args, **kwargs):
        self.user = user
        super(RSVPForm, self).__init__(*args, **kwargs)
  1. 修改视图函数,并将request.user传递给表单
def event_view(request, slug, model_class=Event, form_class=RSVPForm,
               template_name='rsvp/event_view.html'):
    event = get_object_or_404(model_class, slug=slug)

    if request.POST:
        form = form_class(request.user, request.POST)

    if form.is_valid():
        guest = form.save()
        return HttpResponseRedirect(reverse('rsvp_event_thanks',
                                            kwargs={'slug': slug, 'guest_id': guest.id}))
    else:
        form = form_class(request.user)

    return render_to_response(template_name, {
        'event': event,
        'form': form,
    }, context_instance=RequestContext(request))
  1. 现在save()方法的代码应该是这样的:
guest = self.guest_class.objects.get(user=self.user)

14
注意,你违反了Liskov替换原则:https://en.wikipedia.org/wiki/Liskov_substitution_principle。请查看此答案以添加用户参数而不违反Liskov原则:https://dev59.com/Cmw05IYBdhLWcg3wVwc4#7300076 - bux

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接