Django REST 查询：检索用户

Question

我希望检索当前用户（已给出 ID）的所有员工（用户）。员工是用户的多对多字段。目前我的查询检索当前用户。而 user.employees 只是返回所有员工的 id。

是否可以立即进行查询以检索当前用户的所有员工？或者我只是应该为每个用户发送更多的 API 调用（从前端），我通过 id 检索用户的数据？

如果有人能引导我朝着正确的方向前进，那就太棒了。 :)

views.py

# display all your employees
class EmployeeViewSet(viewsets.ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

    def list(self, request):
        queryset = User.objects.filter(pk=request.user.pk) #get current user
        serializer = UserSerializer(queryset, many=True) 
        return Response(serializer.data)

models.py

class UserProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    employees = models.ManyToManyField(User, related_name='employees')

serializers.py

class UserSerializer(UserDetailsSerializer):

class Meta(UserDetailsSerializer.Meta):
    fields = UserDetailsSerializer.Meta.fields + ('employees')

def update(self, instance, validated_data):
    profile_data = validated_data.pop('userprofile', {})
    employees = profile_data.get('employees')

    instance = super(UserSerializer, self).update(instance, validated_data)

    # get and update user profile
    profile = instance.userprofile
    if profile_data:
        if employees:
            profile.employees = employees
        profile.save()
    return instance

rest_auth/serializers.py（依赖）

class UserDetailsSerializer(serializers.ModelSerializer):
    """
    User model w/o password
    """
    class Meta:
        model = UserModel
        fields = ('pk', 'username', 'email', 'first_name', 'last_name')
        read_only_fields = ('email', )

示例用户

Answer 1

class EmployeeViewSet(viewsets.ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

    def list(self, request):
        queryset = self.queryset.filter(id__in=request.user.employees.all())
        serializer = UserSerializer(queryset, many=True) 
        return Response(serializer.data)

此外，在序列化程序中，您可能必须执行以下操作才能获得所需的响应结构。

class UserSerializer(UserDetailsSerializer):
    employees = serializers.SerializerMethodField() 

    class Meta(UserDetailsSerializer.Meta):
        fields = UserDetailsSerializer.Meta.fields + ('employees')

    def get_employees(self, obj):
        return obj.userprofile.employees.all()

Answer 2

您可以使用如下过滤器： request.user.userprofile.employees.all() 获取特定用户的员工。

class EmployeeViewSet(viewsets.ModelViewSet):
    queryset = User.objects.all()
    serializer_class = UserSerializer

    def list(self, request):
        queryset = request.user.userprofile.employees.all() #get all employees for current user
        serializer = UserSerializer(queryset, many=True) 
        return Response(serializer.data)

This documention 关于一对一的关系也会很有用。