我正在使用 DRF 在單頁應用程式中創建 API。我有一個客戶用戶類，我只添加了一個is_manager標志和一個managerEntity模型，其中is_manager標志為 True 的用戶可以創建 managerEntities 成為它們的所有者。問題是我似乎無法弄清楚如何在創建方法之前驗證來自序列化程式的資料以檢查是否設定了is_manager。如果設定，則應該創建managerEntity，否則，引發例外。
class DeepmetricsUser(AbstractUser):
is_manager = models.BooleanField(default=False)
class managerEntity(models.Model):
id = models.AutoField(primary_key=True)
name = models.CharField(max_length=200)
owner = models.ForeignKey(get_user_model(), on_delete=models.CASCADE)
team = models.ManyToManyField(get_user_model(), blank=True)
視圖.py
class managersEntityViewSet(viewsets.ModelViewSet):
queryset = managerEntity.objects.all()
serializer_class = managerEntityModelSerializer
permission_classes = [permissions.IsAuthenticated]
def get_queryset(self):
return self.queryset.filter(Q(owner = self.request.user) | Q(team=self.request.user.id))
def create(self, request, *args, **kwargs):
serializer = managerEntitySerializer(data=request.data, context={"request": self.request})
serializer.is_valid(raise_exception=True)
res = serializer.save()
data = managerEntityModelSerializer(res).data
return Response(data, status=status.HTTP_201_CREATED)
序列化程式.py
class managerEntitySerializer(serializers.Serializer):
name = serializers.CharField(max_length=255)
owner = serializers.HiddenField(default=serializers.CurrentUserDefault())
def create(self, data):
res = managerEntity.objects.create(**data)
return res
uj5u.com熱心網友回復：
您需要覆寫validateSerializer 中的方法
def validate(self, attrs):
if not self.context["request"].user.is_manager:
serializers.ValidationError("Validation error")
return attrs
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/378543.html