如何在Django REST Framework视图集中创建带默认用户组的用户？

嘿，这个需求实现起来挺简单的，我给你两种常用的方案，你可以根据自己的代码结构和职责划分偏好来选：

方案1：在序列化器的create方法中添加组分配逻辑 #

这种方式适合把用户创建和组绑定的逻辑放在一起，修改你现有的UserSerializer就行。首先要导入Django的Group模型，然后在创建用户后自动关联默认组：

from django.contrib.auth.models import Group
from rest_framework import serializers
from django.contrib.auth.models import User

class UserSerializer(serializers.ModelSerializer):
    password = serializers.CharField(write_only=True)
    
    class Meta:
        model = User
        fields = ('username', 'first_name', 'last_name', 'email', 'password')
    
    def create(self, validated_data):
        user = super(UserSerializer, self).create(validated_data)
        user.set_password(validated_data['password'])
        user.save()
        
        # 分配默认用户组：这里用get_or_create确保组存在，避免报错
        # 把'普通用户'改成你实际需要的默认组名称
        default_group, _ = Group.objects.get_or_create(name='普通用户')
        user.groups.add(default_group)
        
        return user

小说明： #

• 用get_or_create是为了防止你的默认组还没在系统中创建的情况，如果已经手动在后台添加过目标组，也可以直接用Group.objects.get(name='你的组名')，但前者更安全。
• 如果需要给用户分配多个组，可以多次调用user.groups.add()，或者传入组对象的列表，比如user.groups.add(group1, group2)。

方案2：在视图集的perform_create方法中处理 #

这种方式把组分配的逻辑放在视图层，让序列化器只专注于用户数据的序列化和基础创建，职责划分更清晰：

from django.contrib.auth.models import Group
from rest_framework import viewsets
from django.contrib.auth.models import User
from .serializers import UserSerializer

class UserView(viewsets.ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer
    
    def perform_create(self, serializer):
        # 先让序列化器完成用户创建
        user = serializer.save()
        # 分配默认组
        default_group, _ = Group.objects.get_or_create(name='普通用户')
        user.groups.add(default_group)

小说明： #

• DRF的ModelViewSet中，perform_create方法是用来处理创建对象的额外逻辑的，重写它不会影响序列化器的原有逻辑，适合后续可能需要在不同视图中给用户分配不同组的场景。

不管选哪种方案，只要你的用户模型（不管是Django自带的User还是自定义的继承AbstractUser的模型）带有groups多对多字段，就能正常工作啦。

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