使用Django Rest Framework API删除相关模型

4 投票
1 回答
2872 浏览
提问于 2025-04-17 20:05

当我使用Django Rest Framework删除一个用户时,和这个用户相关联的用户资料(UserProfile)对象也会被删除。我希望反过来也是这样,也就是说,如果删除用户资料,用户也能被删除。请问我该怎么做呢?

我有一个Django模型,用来表示用户的资料。

class UserProfile(models.Model):
    user = models.OneToOneField(User, related_name='profile')
    preferred_name = models.CharField(max_lengths=100)
    #other fields here

以下是我的视图代码:

class UserDetail(generics.RetrieveUpdateDestroyAPIView):
    """ 
    API endpoint that represents a single user.
    """
    model = User
    serializer_class = UserSerializer

class UserProfileDetail(generics.RetrieveUpdateDestroyAPIView):
    """ 
    API endpoint that represents a single UserProfile
    """
    model = UserProfile
    serializer_class = UserProfileSerializer

还有序列化器的代码:

class UserSerializer(serializers.HyperlinkedModelSerializer):
    profile = serializers.HyperlinkedRelatedField(view_name = 'userprofile-detail')

    class Meta:
        model = User
        fields = ('url', 'username', 'email', 'profile')

class UserProfileSerializer(serializers.HyperlinkedModelSerializer):

    class Meta:
        model = UserProfile
        fields = ('url', 'preferred_name', 'user')

1 个回答

2

你可以在你的用户资料类里重写删除方法,像这样:

class UserProfile(models.Model):
    user = models.OneToOneField(User, related_name='profile')
    preferred_name = models.CharField(max_lengths=100)
    #other fields here
    def delete(self, *args, **kwargs):
        self.user.delete()
        super(UserProfile, self).delete(*args, **kwargs)

撰写回答