将函数装饰器转换为类 Mixin django

问题描述

我尝试将 cutoms 装饰器从函数转换为类 Mixin，因为我使用 CVB 并且我想从这个 Mixin 继承来检查某些页面的用户状态。我有这个装饰器，它检查用户是否经过身份验证，如果用户经过身份验证并尝试访问具有此装饰器的页面，它将被重定向到仪表板，如果不是，则将有权访问该页面。我将这个类编写为装饰器的类版本，如果用户已登录，则它可以正常工作，但如果不是并尝试访问该页面，则会出现此错误：

在 /auth/login/ 处配置不当 UserLoginView 缺少 permission_required 属性。定义 UserLoginView.permission_required，或覆盖 UserLoginView.get_permission_required()。

这是装饰器：

def is_authenticated(view_func):
    def wrapper_func(request,*args,**kwargs):
        if request.user.is_authenticated:
            return redirect('dashboard')
        else:
            return view_func(request,**kwargs)
    return wrapper_func

类版本：

class IsAuthenticatedMixin(PermissionrequiredMixin):
    def dispatch(self,request,**kwargs):
        if request.user.is_authenticated:
            return redirect('dashboard')
        return super().dispatch(request,**kwargs)

继承这个mixin的视图

class IndexFormView(IsAuthenticatedMixin,CreateView):
    permission_required = 'view'
    template_name = 'home/index.html'
    form_class = NewsletterForm

    def post(self,**kwargs):
        email = request.POST['email']
        if Newsletter.objects.filter(email=email).exists():
            messages.warning(
                request,'This email is already subscribed in our system')
        else:
            Newsletter.objects.create(
                email=email)
            messages.success(request,'Your email was subscribed in our system,you\'ll hear from us as soon as possible !')
        return super().post(request,**kwargs)

    def get_context_data(self,**kwargs):
        context = super().get_context_data(**kwargs)
        context['testimonials'] = Testimonial.objects.order_by('-created_date')[:9]
        return context


class AboutTemplateView(IsAuthenticatedMixin,TemplateView):
    template_name = 'home/about.html'

    def get_context_data(self,**kwargs):
        context = super().get_context_data(**kwargs)
        context['banner_page_title'] = 'About Us'
        context['page_location'] = 'home / about'
        return context

class UserResgistrationCreateView(IsAuthenticatedMixin,CreateView):
    template_name = 'home/auth/register.html'
    form_class = UserRegistrationForm
    success_url = reverse_lazy('login')


class UserLoginView(IsAuthenticatedMixin,LoginView):
    template_name = 'home/auth/login.html'

解决方法

发生这种情况的原因是您从 PermissionRequiredMixin [Django-doc] 子类化，这意味着在使用 IsAuthenticatedMixin 的所有视图中，您需要提供 permission_required 类属性。这里的错误是关于 UserLoginView:

class UserLoginView(IsAuthenticatedMixin,LoginView):
    template_name = 'home/auth/login.html'
    permission_required = …

因此您需要为 … 输入一个值。但是在其他基于类的视图中也会出现同样的问题，在这些视图中您从 IsAuthenticatedMixin 子类，并且您没有为 permission_required 提供值。

然而，这就是您在这里编写自定义 mixin 的原因。您正在执行的操作是由 LoginRequiredMixin [Django-doc] 实现的。唯一不同的是重定向 url，但您可以指定：

from django.contrib.auth.mixins import LoginRequiredMixin
from django.urls import reverse_lazy

class IsAuthenticatedMixin(LoginRequiredMixin):
    login_url = reverse_lazy('dashboard')
    redirect_field_name = None

在需要使用权限的视图中，还可以将PermissionRequiredMixin子类化：

from django.contrib.auth.mixins import PermissionRequiredMixin

class IndexFormView(IsAuthenticatedMixin,PermissionRequiredMixin,CreateView):
    permission_required = 'view'
    template_name = 'home/index.html'
    form_class = NewsletterForm

decorator decorator django django mixins mixins python