django-rest-framework Instagram like follower系统。

问题描述 投票:0回答:1

所以,事情是这样的:当经过认证的用户(我们称其为用户1)关注另一个用户(我们称其为用户2)时,用户2会被添加到用户1的关注列表中,而用户1则会被自动添加到用户2的关注列表中,这就是我想实现的。谁能帮我解决这个逻辑?

模型.py

class Follower(models.Model):
    user = models.OneToOneField(UserModel,related_name="followers", verbose_name=_("User"), on_delete=models.CASCADE)
    follower_user = models.ManyToManyField(UserModel, verbose_name=_("Follower"),related_name='follower_user')


class Following(models.Model):
    user = models.OneToOneField(UserModel, related_name="following",unique=False, verbose_name=_("User"), on_delete=models.CASCADE)
    following_user = models.ManyToManyField(UserModel, verbose_name=_("Following"), related_name='following_user')

serializers.py

class FollowerSerializer(ModelSerializer):

    class Meta:
        model = Follower
        fields = ('user','follower_user')
        read_only_fields = ()


class FollowingSerializer(ModelSerializer):

    class Meta:
        model = Following
        fields = ('user','following_user')
        read_only_fields = () 

views.py

class FollowerView(ListAPIView):
    queryset = Follower.objects.all()
    serializer_class = FollowerSerializer
    permission_classes = [IsAuthenticated]


class FollowingView(ListCreateAPIView):
    queryset = Following.objects.all()
    serializer_class = FollowingSerializer
    permission_classes = [IsAuthenticated]
python django-models django-rest-framework django-views django-serializer
1个回答
1
投票

你需要在FollowingSerializer中添加一个字段来捕捉跟随者ID,因为你每次只创建一个跟随者。

class FollowingSerializer(ModelSerializer):
    new_following = serializers.PrimaryKeyRelatedField(queryset=UserModel.objects.all(), required=True, write_only=True)
    class Meta:
        model = Following
        fields = ('user', 'following_user', 'new_following')
        read_only_fields = ('following_user')

    def create(self, validated_data):
        user = validated_data['user']
        new_follwoing = validated_data['new_following']
        user.following.following_user.add(new_follwoing)
        new_follwoing.followers.following_user.add(user)

        return user.following
© www.soinside.com 2019 - 2024. All rights reserved.