Django社交应用：优化关注/取关功能的ManyToManyField实践

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

class User(AbstractUser):
    # 'follows' 字段表示当前用户关注了哪些其他用户。
    # 'symmetrical=False' 明确指出这是一种非对称关系（A关注B，B不一定关注A）。
    # 'related_name="followers"' 允许我们从被关注的用户（例如 user_B）
    # 反向查询到关注了 user_B 的所有用户（即 user_B 的粉丝），
    # 通过 user_B.followers.all() 来实现。
    follows = models.ManyToManyField(
        'self',
        symmetrical=False,
        related_name='followers', # 允许通过 user.followers 访问关注该用户的用户列表
        blank=True
    )

    def __str__(self):
        return self.username

    # 辅助方法，方便在模板中获取关注数和粉丝数
    @property
    def following_count(self):
        return self.follows.count() # 我关注了多少人

    @property
    def followers_count(self):
        return self.followers.count() # 有多少人关注了我

from django.shortcuts import render, redirect
from django.contrib.auth import get_user_model
from django.views.decorators.http import require_POST

User = get_user_model()

@require_POST # 确保只接受 POST 请求
def toggle_follow(request):
    """
    处理用户的关注/取关请求。
    """
    # 获取表单提交的动作 ('follow' 或 'unfollow') 和目标用户名
    action = request.POST.get('accion')
    target_username = request.POST.get('usuario')

    # 获取当前登录用户和目标用户对象
    logged_in_user = request.user
    try:
        target_user = User.objects.get(username=target_username)
    except User.DoesNotExist:
        # 如果目标用户不存在，可以重定向到错误页面或返回错误信息
        return redirect('index') # 假设 'index' 是主页URL

    # 检查当前用户是否已关注目标用户
    is_following = logged_in_user.follows.filter(username=target_username).exists()

    if action == 'follow':
        # 如果当前用户尚未关注目标用户，则添加关注关系
        if not is_following:
            logged_in_user.follows.add(target_user)
            print(f"{logged_in_user.username} 关注了 {target_user.username}")
    elif action == 'unfollow':
        # 如果当前用户已关注目标用户，则移除关注关系
        if is_following:
            logged_in_user.follows.remove(target_user)
            print(f"{logged_in_user.username} 取关了 {target_user.username}")
    else:
        # 处理未知动作，可以记录日志或返回错误
        print(f"未知动作: {action}")

    # 重定向