ModelViewSet更多使用和ModelViewSet源码分析

ModelViewSet更多使用

class UsersViewSet(ModelViewSet):
    # 指定查询集
    queryset = models.Users.objects.all()
    # 指定序列化器
    serializer_class = rbac_serializers.UserSerializer

    def perform_create(self, serializer):
        serializer.save()
        req_data = self.request.data
        roles = req_data[‘roles‘]
        # 添加多对多对象
        obj = models.Users.objects.get(id=serializer.data[‘id‘])
        for i in roles:
            role_obj = models.Roles.objects.get(id=i)
            obj.roles.add(role_obj)
            #把角色信息加入到序列化数据中
            serializer.data[‘roles‘].append({
   
   ‘id‘:role_obj.id,‘title‘:role_obj.title,‘label‘:role_obj.label})
        return  serializer.data

    def perform_update(self, serializer):
        req_data = self.request.data
        roles = req_data[‘roles‘]
        user_id = req_data[‘id‘]
        obj = models