使用Django.contrib.auth.models.User实现用户角色分配

在Django中，可以使用django.contrib.auth.models.User来实现用户角色分配。User是Django内置的用户模型，用于管理用户的认证和授权。

首先，我们需要在Django项目中创建一个Role模型来定义不同的角色。可以在项目的models.py文件中添加以下代码：

from django.db import models

class Role(models.Model):
    name = models.CharField(max_length=50)

    def __str__(self):
        return self.name

以上代码定义了一个简单的Role模型，其中包含一个名为name的CharField字段用于存储角色名称。

接下来，我们需要修改django.contrib.auth.models.User模型，添加一个外键字段，将用户与角色关联起来。可以在项目的models.py文件中添加以下代码：

from django.contrib.auth.models import AbstractUser
from django.db import models

class User(AbstractUser):
    role = models.ForeignKey(Role, on_delete=models.CASCADE, null=True)

    def __str__(self):
        return self.username

以上代码使用ForeignKey字段添加了一个名为role的外键字段，它关联到Role模型，并设置了on_delete=models.CASCADE表示当角色被删除时，用户也会被删除。null=True表示该字段可以为空。

然后，我们需要迁移数据库以应用以上更改。在命令行中运行以下命令：

python manage.py makemigrations
python manage.py migrate

现在，我们可以在视图和模板中使用这些用户角色信息。

在视图中，我们可以通过request.user.role来获取当前用户的角色信息。例如，假设我们有一个需要管理员权限的视图：

from django.contrib.auth.decorators import login_required
from django.http import HttpResponseForbidden

@login_required
def admin_view(request):
    if not request.user.role or request.user.role.name != 'admin':
        return HttpResponseForbidden("You do not have permission to access this page.")
    # 当前用户有管理员权限，执行相应操作

以上代码使用了login_required装饰器来确保只有登录用户能够访问该视图。然后，通过检查当前用户的角色信息，判断是否具有管理员权限。

在模板中，我们可以使用{% if %}模板标签来根据用户角色显示或隐藏相应的内容。例如，在HTML模板中：

{% if request.user.role and request.user.role.name == 'admin' %}
    <h1>Welcome, Admin!</h1>
{% endif %}

以上代码根据当前用户的角色信息，如果用户具有管理员角色，则显示欢迎信息。

最后，我们需要为用户分配角色。可以通过Django的管理界面或在视图中编写相应代码来实现。以下是一个简单的例子，展示了如何在视图中为用户分配角色：

from django.contrib.auth.decorators import login_required
from django.shortcuts import render
from .models import User, Role

@login_required
def assign_role(request):
    if request.method == 'POST':
        user_id = request.POST.get('user_id')
        role_id = request.POST.get('role_id')
        user = User.objects.get(id=user_id)
        role = Role.objects.get(id=role_id)
        user.role = role
        user.save()
    users = User.objects.all()
    roles = Role.objects.all()
    return render(request, 'assign_role.html', {'users': users, 'roles': roles})

以上代码定义了一个assign_role视图，它接收POST请求，并从请求参数中获取要分配角色的用户ID和角色ID。然后，通过User.objects.get()和Role.objects.get()来获取相应的用户和角色对象，并将角色分配给用户的role字段。最后，我们渲染一个模板，显示所有用户和角色，以便在界面上进行选择。

以上是使用django.contrib.auth.models.User实现用户角色分配的示例，它可以帮助你实现用户角色的管理和授权功能。你可以根据自己的需求进行相应的修改和扩展。