实战项目十三：构建社交网络（一）：用户关系-Django框架入门指南

当前位置:　首页>> 技术小册>> Django框架入门指南

### 实战项目十三：构建社交网络（一）：用户关系

在前面的章节中，我们已经通过Django框架搭建了基础的Web应用，学习了模型(Models)、视图(Views)、模板(Templates)和表单(Forms)等核心概念，并实现了用户注册、登录、个人信息管理等基本功能。本章节，我们将正式踏入社交网络的构建之旅，首先聚焦于“用户关系”这一核心部分，它是任何社交网络不可或缺的基础架构。

#### 引言

社交网络的核心在于用户之间的连接与互动。用户关系通常包括好友关系、关注关系、群组关系等，它们构成了社交图谱的基础。在本项目中，我们将从最简单的“好友关系”入手，实现用户之间的添加好友、删除好友、查看好友列表等功能。

#### 1. 设计用户关系模型

在开始编码之前，我们首先需要设计用户关系的数据库模型。Django的ORM（对象关系映射）允许我们以Python类的方式定义数据库模型，极大简化了数据库操作。

##### 1.1 定义用户模型（如果尚未定义）

通常，Django自带的`User`模型已经足够用于大多数场景，但如果你需要额外的用户信息（如昵称、头像等），可以通过继承`AbstractUser`来自定义用户模型。这里我们假设已经有一个包含基本信息的用户模型`User`。

##### 1.2 创建好友关系模型

好友关系可以通过多种方式建模，常见的方法包括使用多对多字段、中间模型（通过自定义模型表示关系）等。为了保持关系的灵活性和可扩展性（比如未来可能需要添加好友关系的创建时间、备注等信息），我们选择使用中间模型。

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

class Friendship(models.Model):
    """
    用户间的好友关系模型
    """
    from_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='friendships_from')
    to_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='friendships_to')
    created_at = models.DateTimeField(auto_now_add=True)
    # 可以添加更多字段，如备注、关系状态等

class Meta:
        unique_together = (('from_user', 'to_user'),)  # 确保一对用户之间只存在一种关系

def __str__(self):
        return f"{self.from_user} -> {self.to_user}"
```

这里，`Friendship`模型包含了两个`ForeignKey`字段，分别指向`User`模型，表示关系的发起者和接收者。`unique_together`元组确保了在数据库中不会存在重复的（from_user, to_user）对，即每个用户只能对另一个用户发起一次好友请求（尽管在实际应用中，我们可能还需要处理请求接受与否的状态）。

#### 2. 实现好友功能

有了模型之后，我们就可以开始实现具体的好友功能了。

##### 2.1 添加好友

添加好友功能通常包括发送好友请求和接受好友请求两个步骤。在这里，我们先实现发送好友请求的功能。

```python
from django.shortcuts import redirect, render
from .models import Friendship

def add_friend(request, user_id):
    if request.method == 'POST':
        current_user = request.user
        target_user = User.objects.get(id=user_id)
        
        # 检查当前用户是否已经向目标用户发送了好友请求或已经是好友
        if Friendship.objects.filter(from_user=current_user, to_user=target_user).exists():
            return render(request, 'friends/already_friends.html', {'target_user': target_user})
        
        # 发送好友请求
        Friendship.objects.create(from_user=current_user, to_user=target_user)
        return redirect('user_profile', user_id=target_user.id)  # 假设有一个用户资料页的视图

# 如果是GET请求或其他情况，则重定向到用户资料页或其他页面
    return redirect('user_profile', user_id=user_id)
```

注意：在实际应用中，你可能需要为好友请求添加更多的处理逻辑，比如通知机制（通知目标用户有新的好友请求）、好友请求的过期时间等。

##### 2.2 好友请求的接受与拒绝

好友请求的接受与拒绝可以通过更新`Friendship`模型的状态（虽然在这个简单的模型中我们没有显式地定义状态字段）或创建反向的`Friendship`记录来实现。这里，我们采用后者作为示例，即当用户接受好友请求时，在数据库中为两个用户之间各自创建一个`Friendship`记录。

```python
def accept_friend_request(request, friendship_id):
    if request.method == 'POST':
        friendship = Friendship.objects.get(id=friendship_id)
        # 创建反向的Friendship记录以表示双向好友关系
        Friendship.objects.create(from_user=friendship.to_user, to_user=friendship.from_user)
        # 可选：删除原始的好友请求记录，如果不再需要
        # friendship.delete()
        return redirect('user_profile', user_id=friendship.from_user.id)
    # 错误处理或重定向逻辑
    return HttpResponseBadRequest("Invalid request")
```

注意：在这个示例中，我们没有直接删除原始的好友请求记录，因为有时你可能想要保留历史记录以供将来分析或审计。然而，根据你的应用需求，你可能想要这样做。

##### 2.3 查看好友列表

查看好友列表功能相对简单，只需从当前用户出发，查询所有与之关联的`Friendship`记录，并获取这些记录的`to_user`字段即可。

```python
def view_friends(request):
    if request.user.is_authenticated:
        friends = User.objects.filter(friendships_to__from_user=request.user)
        return render(request, 'friends/friends_list.html', {'friends': friends})
    else:
        return redirect('login')  # 未登录用户重定向到登录页面
```

这里，我们使用了Django的`__`查询语法来跨关系查询。`friendships_to__from_user=request.user`意味着查找所有`Friendship`记录，其中`to_user`字段对应的`User`的`friendships_to`（即接收到的友谊）中的`from_user`是当前用户。

#### 3. 视图与模板的整合

上述代码片段展示了如何在Django中处理用户关系的基本逻辑。然而，要让这些功能对用户可见并可用，你还需要创建相应的视图和模板。视图负责处理HTTP请求并调用业务逻辑，模板则负责渲染数据并展示给用户。

在视图中，你将调用上述定义的函数或方法，并处理可能的异常和错误。在模板中，你将使用Django模板语言来展示数据，如好友列表、好友请求状态等。

#### 4. 扩展与考虑

虽然我们已经实现了用户关系的基本功能，但还有许多可以扩展和改进的地方：

- **好友请求状态**：为`Friendship`模型添加状态字段（如`pending`, `accepted`, `rejected`），以更精细地控制好友请求的流程。
- **通知系统**：实现一个通知系统，当有新的好友请求、好友接受请求等事件发生时，及时通知用户。
- **性能优化**：考虑使用缓存来减少数据库查询次数，提高应用的响应速度。
- **安全性**：确保所有用户操作都经过身份验证和授权，防止未授权访问和篡改。
- **UI/UX**：设计直观易用的用户界面，提升用户体验。

通过这些扩展和改进，你的社交网络应用将变得更加健壮、灵活和易用。

该分类下的相关小册推荐：

Django零基础入门

Django快速开发实战