我正在尝试使用基于Djangos类的通用视图编写CRUD应用程序。下面是我为在数据库中创建新用户而编写的代码。

  from django.views.generic import CreateView
  from django.contrib.auth.decorators import login_required
  from django.contrib import messages

  class UserCreateView(CreateView):
  """
  Display and accept a new user to be created in db
  """
    form_class = ProfileForm
    template_name = 'userdb/profile_form.html'
    success_url = '/organization/users/'

    def post(self, request, *args, **kwargs):
      messages.success(request, "Success", extra_tags='msg')
      return super(UserCreateView, self).post(request, *args, **kwargs)

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

请注意,要添加要向用户显示的成功消息,我必须扩展post函数。我知道这不是一个很好的方法,因为当调用这个函数时,还不能决定提交的表单是否包含有效的数据。所以我的问题是,是否有推荐的方法将Djangos消息传递框架与基于类的通用视图结合起来?

最佳答案

答案取决于您具体希望如何处理消息传递框架。如果需要为每个get请求调用它,您自然需要将其放入get方法中(即没有合适的地方放置此代码)。
不管怎样,听起来你在找一个只有在表单有效时才会触发的地方。
CreateView使用实现ModelFormMixin方法的form_valid方法,该方法仅在成功保存表单时激发。很完美!

def form_valid(self, form):
    messages.success(self.request, "Success", extra_tags='msg')
    return super(UserCreateView, self).form_valid(form)
    # ModelFormMixin will now save
    # FormMixin will now redirect to success_url()
    # override above behavior if you need to do something with the object

10-07 19:57
查看更多