Django中如何限制仅Premium用户访问销售页 Trial用户跳转升级页
Django 限制Trial用户访问销售页面实现方案
方案1:单视图直接重写dispatch方法(适合仅当前页面需要校验的场景)
首先导入依赖:
from django.contrib.auth.mixins import LoginRequiredMixin from django.shortcuts import render
修改你的SalesView视图代码如下:
# 注意LoginRequiredMixin要放在继承列表最左侧,确保先校验登录状态 class SalesView(LoginRequiredMixin, TemplateView, SingleTableView): model = Product table_class = ProductTable template_name = 'src/dashboard/sales.html' def dispatch(self, request, *args, **kwargs): # 校验用户类型,Trial用户直接返回升级页面 if request.user.user_type == 'Trial': # 此处upgrade.html路径替换为你实际的模板路径即可 return render(request, 'src/dashboard/upgrade.html') # 高级用户正常走原有视图逻辑 return super().dispatch(request, *args, **kwargs) def get_queryset(self): ... ... def get_context_data(self, **kwargs): # my contexts goes here
路由配置不需要做任何修改,原有配置可以直接使用。如果你的业务允许未登录用户访问该页面,仅限制Trial类型的登录用户,删除LoginRequiredMixin继承即可。
方案2:通用权限混入类(适合多页面需要Premium权限校验的场景)
如果后续有多个页面都需要限制仅Premium用户访问,可以封装通用的混入类避免重复代码:
# 可以放在项目的utils/views.py等公共文件中 class PremiumRequiredMixin: def dispatch(self, request, *args, **kwargs): # 可按需保留/删除登录校验逻辑 if not request.user.is_authenticated: return self.handle_no_permission() if request.user.user_type == 'Trial': return render(request, 'src/dashboard/upgrade.html') return super().dispatch(request, *args, **kwargs)
之后所有需要Premium权限的视图都直接继承该混入类即可:
class SalesView(PremiumRequiredMixin, TemplateView, SingleTableView): # 原有视图逻辑不需要做任何修改 model = Product table_class = ProductTable template_name = 'src/dashboard/sales.html' def get_queryset(self): ... ... def get_context_data(self, **kwargs): # my contexts goes here
内容的提问来源于stack exchange,提问作者Sammy
相关产品推荐
相关产品推荐

