Django多角色登录后按角色渲染对应页面的高效实现方案

• 硬编码if/elif分支写角色跳转，角色规模到30个以上时，代码冗余度高，每新增一个角色都要改动登录视图核心逻辑，维护成本极高
• 存在逻辑匹配bug：Profile模型的ROLE_CHOICES存储值全为大写（如SALES/MARKETING），但分支判断写的是pro.role == 'Sales'，永远无法匹配销售角色的跳转规则
• 未做权限校验：前端提交的role参数完全没在后端逻辑里使用，不具备对应角色的账号也能绕过选择的角色直接跳转，存在越权登录风险
• 原有代码存在缩进语法错误，会导致密码校验失败后的错误提示逻辑无法正常执行

核心思路 #

把「角色值 -> 对应登录跳转路由名」的映射关系从登录视图中抽离，采用配置驱动的模式，彻底移除所有if/elif判断，新增角色仅需修改配置项，无需改动登录核心代码。

1. 定义全局角色路由映射配置 #

在项目根目录新建独立配置文件（如role_config.py），完全对齐模型里的角色存储值维护映射关系：

# 项目根目录 / role_config.py
ROLE_DASHBOARD_MAP = {
    "SALES": "sales:sales-main-dashboard",
    "MARKETING": "marketing:marketing-main-dashboard",
    "HR": "hr:hr-main-dashboard",
    "PROJECT MANAGER": "projectmanager:pm-main-dashboard",
    "SUPERVISOR": "supervisor:supervisor-main-dashboard",
    # 后续新增30个角色仅需在此处追加键值对即可，无需改动登录视图
}

2. 重构登录视图逻辑 #

替换原有分支判断代码，补全安全校验和bug修复：

from django.shortcuts import redirect
from django.contrib.auth import login
from django.contrib.auth.hashers import check_password
from django.urls import reverse
from role_config import ROLE_DASHBOARD_MAP

class SignView(View):
    def get(self, request):
        return render(request, 'login/login.html')

    def post(self, request):
        email = request.POST.get('email')
        password = request.POST.get('password')
        selected_role = request.POST.get('role')

        # 校验用户是否存在
        user = User.objects.filter(email=email).first()
        if not user:
            messages.error(request, '账号或密码错误')
            return render(request, 'login/login.html')

        # 校验密码正确性
        if not check_password(password, user.password):
            messages.error(request, '账号或密码错误')
            return render(request, 'login/login.html')

        # 查询用户关联的Profile信息
        try:
            user_profile = Profile.objects.get(user=user)
        except Profile.DoesNotExist:
            messages.error(request, '用户权限配置异常，请联系管理员')
            return render(request, 'login/login.html')

        # 校验前端选择的角色和用户实际绑定角色是否一致，拦截越权请求
        if user_profile.role != selected_role:
            messages.error(request, '选择的角色和账号权限不匹配')
            return render(request, 'login/login.html')

        # 从映射表直接获取跳转路由，无任何分支判断
        target_route = ROLE_DASHBOARD_MAP.get(user_profile.role)
        if not target_route:
            messages.error(request, '当前角色未配置登录跳转权限')
            return render(request, 'login/login.html')

        login(request, user)
        return redirect(reverse(target_route))

3. 规模化扩展适配 #

当后续角色规模到30个以上时，可以把映射关系迁移到数据库表中，配套后台管理页面实现角色跳转规则的可视化配置，完全脱离代码修改就能完成新角色的接入。

优化提示：所有角色的dashboard路由可以统一遵循命名规范（如app_name为角色小写标识、路由名固定为{role_prefix}-main-dashboard），后续甚至可以直接通过规则拼接路由名，进一步减少配置维护量。

内容的提问来源于stack exchange，提问作者RUTUJA KADAM