我正在尝试使用基于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