要实现 Django 中的选择角色登录功能，我们可以采用自定义用户模型和 Django 内置的认证系统。首先，我们需要创建一个自定义用户模型，然后扩展 Django 的认证后端，以支持根据角色进行登录。
以下是完整的可运行代码：
首先，创建一个新的 Django 项目，并创建一个新的应用程序：
# 创建Django项目
django-admin startproject role_login_project
# 进入项目目录
cd role_login_project
# 创建应用程序
python manage.py startapp role_login
接下来，我们定义自定义用户模型和角色模型：
在 role_login/models.py 文件中：
from django.contrib.auth.models import AbstractUser
from django.db import models
# 自定义用户模型
class CustomUser(AbstractUser):
# 添加一个角色字段，用于区分不同角色
ROLE_CHOICES = (
('user', 'User'),
('admin', 'Admin'),
)
role = models.CharField(max_length=10, choices=ROLE_CHOICES, default='user')
然后，我们需要更新 Django 的配置，使用我们的自定义用户模型：
在 role_login/settings.py 文件中，添加以下内容：
AUTH_USER_MODEL = 'role_login.CustomUser'
接下来，我们要扩展 Django 的认证后端，以支持根据角色进行登录。在 role_login/auth_backends.py 文件中创建一个自定义的认证后端：
from django.contrib.auth import get_user_model
CustomUser = get_user_model()
class RoleBackend:
def authenticate(self, request, username=None, password=None, **kwargs):
try:
user = CustomUser.objects.get(username=username)
except CustomUser.DoesNotExist:
return None
if user.check_password(password) and user.role in kwargs.get('allowed_roles', []):
return user
def get_user(self, user_id):
try:
return CustomUser.objects.get(pk=user_id)
except CustomUser.DoesNotExist:
return None
现在，我们需要将自定义认证后端添加到 Django 的认证后端列表中。在 role_login/settings.py 文件中，添加以下内容：
AUTHENTICATION_BACKENDS = ['role_login.auth_backends.RoleBackend']
接下来，创建视图和模板来处理登录：
在 role_login/views.py 文件中：
from django.contrib.auth import authenticate, login
from django.shortcuts import render, redirect
def select_role(request):
if request.method == 'POST':
role = request.POST.get('role')
if role in ['user', 'admin']:
user = authenticate(request, username=request.user.username, password=request.user.password, allowed_roles=[role])
if user:
login(request, user)
return redirect('home') # 修改为实际的主页URL
return render(request, 'role_login/select_role.html')
在 role_login/templates/role_login 文件夹中创建 select_role.html 模板：
<!DOCTYPE html>
<html>
<head>
<title>Select Role</title>
</head>
<body>
<h2>Select your role:</h2>
<form method="post">
{% csrf_token %}
<label for="user">User</label>
<input type="radio" name="role" value="user" required>
<label for="admin">Admin</label>
<input type="radio" name="role" value="admin" required>
<br>
<button type="submit">Login</button>
</form>
</body>
</html>
最后，配置 URL 路由，让我们可以访问 select_role 视图：
在 role_login/urls.py 文件中：
from django.urls import path
from . import views
urlpatterns = [
path('select_role/', views.select_role, name='select_role'),
# 添加其他视图和URL配置
]
现在，我们已经完成了选择角色的登录功能。当用户登录时，他们将被重定向到选择角色的页面，选择合适的角色后，将被重定向到主页或其他相关页面。
请注意，在实际的项目中，你可能需要根据你的需求进行进一步的调整和改进。此示例代码中，我们使用了简单的角色选择机制（用户或管理员），你可以根据需要进行扩展。
解释说明：
上述代码通过自定义用户模型（CustomUser）和自定义认证后端（RoleBackend ）来实现选择角色的登录功能。首先，我们定义了一个 CustomUser 模型，它继承自 Django 的 AbstractUser，并添加了一个 role 字段，用于标识用户的角色。
然后，我们创建了一个自定义的认证后端 RoleBackend，它继承自 Django 的 ModelBackend。在认证过程中，我们根据传入的角色信息判断用户是否具有登录权限。这样，我们可以根据用户选择的角色来限制登录的权限，例如，只有管理员角色可以登录到管理后台。
接着，我们创建了视图 select_role，用于渲染选择角色的页面和处理用户的角色选择请求。当用户提交角色选择表单后，我们在视图中使用 authenticate() 方法来验证用户凭据和角色，如果通过认证，我们使用 login() 方法将用户登录到系统中。
最后，我们配置了 URL 路由，使得用户可以通过访问 select_role/ 路径来进行角色选择登录。
总之，以上代码提供了一个简单的选择角色登录功能的示例，你可以根据项目需求进行扩展和优化。请注意，对于更复杂的角色和权限管理，你可能需要使用 Django 的权限系统或其他相关扩展来实现。
