如何将好友系统添加到 Django

Question

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

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)

如果需要个人资料.html请告诉我：(

Answer 1

request.user 没有错误指示的好友记录。您可以简单地将 get 操作更改为filter&first操作。Friend.objects.filter(current_user=request.user).first()不要忘记检查好友实例是否存在。所以你的视图应该是这样的：

def profile(request, username=None):
friend = Friend.objects.filter(current_user=request.user).first()
friends = []
if friend:   
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)