如何将朋友系统添加到django

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

我一直试图在django中添加一个朋友系统,用户可以在其中添加和删除朋友,在此过程中,我发现Friend匹配查询不存在错误,并显示这部分代码是错误的:

    friend = Friend.objects.get(current_user=request.user) 

现在我将保留完整的代码。

views.py

    def profile(request, username=None):
        friend = Friend.objects.get(current_user=request.user)
        friends = friend.users.all()
        if username:
          post_owner = get_object_or_404(User, username=username)
          user_posts=Post.objects.filter(user_id=post_owner) 
        else:
          post_owner = request.user
          user_posts=Post.objects.filter(user=request.user)
        args1 = {
          'post_owner': post_owner,
          'user_posts': user_posts,
          'friends': friends,
        }
        return render(request, 'profile.html', args1)

    def change_friends(request, operation, pk):
        friend = User.objects.get(pk=pk)
        if operation == 'add':
          Friend.make_friends(request.user, friend)
        elif operation == 'remove':
          Friend.lose_friends(request.user, friend)
        return redirect('profile')

models.py

    class Friend(models.Model):
        users = models.ManyToManyField(User, default='users', blank=True, related_name='users')
        current_user = models.ForeignKey(User, related_name='owner', on_delete=models.CASCADE, null=True)

        @classmethod
        def make_friend(cls, current_user, new_friend):
            friend, created = cls.objects.get_or_create(
                current_user=current_user
            )
            friend.users.add(new_friend)

        @classmethod
        def lose_friend(cls, current_user, new_friend):
            friend, created = cls.objects.get_or_create(
                current_user=current_user
            )
            friend.users.remove(new_friend)

如果需要profile.html,请告诉我:)

html django django-forms django-views django-queryset
1个回答
1
投票

request.user没有朋友记录,错误表明。您可以简单地将get操作更改为filterfirst操作。 Friend.objects.filter(current_user=request.user).first()然后不要忘记检查朋友实例是否存在。因此,您的视图应类似于:

© www.soinside.com 2019 - 2024. All rights reserved.