Django 如何使Django的“用户”系统具有“好友”功能

在本文中，我们将介绍如何使用Django框架扩展Django的用户系统，实现用户的好友功能。这个功能对于社交媒体、网络社区等应用程序非常重要，因为用户之间的互动是用户体验和社交性的关键。

Django的用户系统简介

Django框架提供了一个强大的用户系统，可以满足大多数Web应用程序的需求。该系统包括用户注册、登录、权限管理等功能。但是，原始的用户系统并没有提供好友功能，因此我们需要进行扩展来实现这个功能。

扩展用户模型

我们可以通过扩展Django的用户模型来添加好友功能。首先，我们需要创建一个新的模型来表示用户之间的关系。这个模型可以包含两个外键，分别指向用户表格中的两个用户。以下是一个示例：

from django.contrib.auth.models import User
from django.db import models

class Friend(models.Model):
    user = models.ForeignKey(User, related_name='friends', on_delete=models.CASCADE)
    friend = models.ForeignKey(User, related_name='friend_of', on_delete=models.CASCADE)

在这个示例中，我们创建了一个Friend模型，其中包含两个外键：user和friend。这两个外键都指向User模型，并指定了related_name属性，使得我们可以通过用户对象访问他们的好友列表。

添加好友关系

为了创建用户之间的好友关系，我们需要添加一个视图函数和一个URL模式，用于处理添加好友的请求。以下是一个示例：

from django.contrib.auth.decorators import login_required
from django.http import HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse

@login_required
def add_friend(request, friend_id):
    user = request.user
    friend = User.objects.get(id=friend_id)
    user.friends.add(friend)
    user.save()
    return HttpResponseRedirect(reverse('profile'))

在这个示例中，我们使用login_required装饰器来确保只有已登录的用户才能添加好友。这个视图函数接收一个friend_id参数，表示要添加的好友的ID。然后，我们通过User.objects.get方法获取要添加的好友对象。接下来，我们把好友对象添加到用户的好友列表中，并保存用户对象。最后，我们重定向到用户的个人资料页面。

需要注意的是，我们还需要定义一个URL模式，将这个视图函数和URL路径绑定起来：

from django.urls import path
from .views import add_friend

urlpatterns = [
    # other URL patterns
    path('add_friend/<int:friend_id>/', add_friend, name='add_friend'),
]

显示好友列表

要显示用户的好友列表，我们需要在用户的个人资料页面中添加相应的代码。以下是一个示例：

from django.contrib.auth.decorators import login_required
from django.shortcuts import render

@login_required
def profile(request):
    user = request.user
    friends = user.friends.all()
    context = {'user': user, 'friends': friends}
    return render(request, 'profile.html', context)

在这个示例中，我们使用login_required装饰器来确保只有已登录的用户才能访问个人资料页面。在视图函数中，我们获取当前用户和他们的所有好友。然后，我们将用户和好友列表封装到一个上下文字典中，传递给渲染模板。

在个人资料页面的模板文件（profile.html）中，我们可以使用Django的模板语言来迭代和显示好友列表：

{% for friend in friends %}
    <p>{{ friend.username }}</p>
{% endfor %}

这个模板代码将迭代用户的好友列表，并显示每个好友的用户名。

移除好友关系

如果用户想要移除某个好友，我们也需要提供相应的功能。我们可以创建一个视图函数和URL模式，处理移除好友的请求：

from django.contrib.auth.decorators import login_required
from django.http import HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse

@login_required
def remove_friend(request, friend_id):
    user = request.user
    friend = User.objects.get(id=friend_id)
    user.friends.remove(friend)
    user.save()
    return HttpResponseRedirect(reverse('profile'))

这个视图函数类似于添加好友的视图函数，但是我们调用的是remove方法来移除好友关系。

同样地，我们需要定义一个URL模式，将这个视图函数和URL路径绑定起来：

from django.urls import path
from .views import add_friend, remove_friend

urlpatterns = [
    # other URL patterns
    path('add_friend/<int:friend_id>/', add_friend, name='add_friend'),
    path('remove_friend/<int:friend_id>/', remove_friend, name='remove_friend'),
]