python - 我想向 Django DetailView 添加条件

Question

我正在使用 Django 通用 View DetailView。 但我想阻止用户访问尚未通过电子邮件确认的详细信息。 我在用户模型中有一个 email_confirmed 字段。

我的代码是:

@method_decorator(login_required(login_url='/login/'), name='dispatch')
class RecruitView(generic.DetailView):
    model = Recruit
    template_name = 'recruit.html'

我想补充:

    if not request.user.email_confirmed:
        error_message = "you did not confirmed yet. please check your email."
        return render(request, 'info.html', {'error_message': error_message})
    else: pass

如何将此条件添加到 DetailView？

(我试图覆盖“as_view”，但我不知道该怎么做)

Answer 1

我会使用 PermissionRequiredMixin .有了这个，您可以指定用户需要拥有的特定权限或覆盖 has_permission方法。

from django.contrib.auth.mixins import PermissionRequiredMixin

class RecruitView(PermissionRequiredMixin, generic.DetailView):
    ...

    login_url = '/login/'
    permission_denied_message = 'you did not confirmed yet. please check your email.'

    def has_permission(self):
         return self.request.user.email_confirmed

这会将没有 email_confirmed 的用户重定向到可以显示错误消息的 login_url。为了使用 index.html 模板，您可能需要覆盖 handle_no_permission方法。