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.Users