Django自定义User模型后实现用户收藏多Feed的方法问询

嘿，这个需求其实挺常见的，咱们一步步拆解来实现～

因为一个用户可以收藏多个Feed，同时一个Feed也能被多个用户收藏，所以多对多（ManyToMany）关系是最适合的方案。我们分两种场景来实现：

方案一：基础收藏（仅记录关联关系） #

如果你的收藏功能只需要知道「用户收藏了哪些Feed」，不需要额外信息（比如收藏时间、备注），直接在模型里加ManyToManyField就行。

1. 修改自定义User模型 #

假设你已经有了自定义User和Feed模型，直接在CustomUser里添加关联字段：

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

class CustomUser(AbstractUser):
    # 你的自定义字段（比如昵称、头像）
    nickname = models.CharField(max_length=50, blank=True)
    # 关联Feed，related_name方便反向查询（比如查哪些用户收藏了某条Feed）
    favorite_feeds = models.ManyToManyField(
        'Feed', 
        related_name='favorited_by', 
        blank=True
    )

class Source(models.Model):
    name = models.CharField(max_length=100)
    # 其他Source字段...

class Feed(models.Model):
    source = models.ForeignKey(Source, on_delete=models.CASCADE, related_name='feeds')
    title = models.CharField(max_length=200)
    content = models.TextField()
    # 其他Feed字段...

2. 数据库迁移 #

添加字段后执行迁移命令：

python manage.py makemigrations
python manage.py migrate

3. 常用操作示例 #

• 收藏Feed：

user = CustomUser.objects.get(username='melissa')
target_feed = Feed.objects.get(id=1)
user.favorite_feeds.add(target_feed)

• 取消收藏：

user.favorite_feeds.remove(target_feed)

• 查询用户收藏的所有Feed：

user.favorite_feeds.all()

• 查询某条Feed的所有收藏用户：

target_feed.favorited_by.all()

方案二：带额外信息的收藏（比如收藏时间、备注） #

如果需要记录「用户什么时候收藏的」或者给收藏加备注，就得用through参数指定中间模型，来存储额外字段。

1. 定义中间模型 #

创建一个Favorite模型，用来保存用户、Feed和额外信息：

class Favorite(models.Model):
    user = models.ForeignKey(CustomUser, on_delete=models.CASCADE)
    feed = models.ForeignKey(Feed, on_delete=models.CASCADE)
    created_at = models.DateTimeField(auto_now_add=True)  # 自动记录收藏时间
    note = models.CharField(max_length=200, blank=True)  # 可选的备注字段

    # 确保一个用户只能收藏同一条Feed一次
    class Meta:
        unique_together = ('user', 'feed')

2. 修改User模型的多对多关联 #

指定through参数关联中间模型：

class CustomUser(AbstractUser):
    nickname = models.CharField(max_length=50, blank=True)
    favorite_feeds = models.ManyToManyField(
        'Feed',
        through='Favorite',  # 指定中间模型
        related_name='favorited_by',
        blank=True
    )

3. 数据库迁移 #

同样执行迁移命令：

python manage.py makemigrations
python manage.py migrate

4. 常用操作示例 #

• 添加带备注的收藏：

user = CustomUser.objects.get(username='melissa')
target_feed = Feed.objects.get(id=1)
# 方式一：直接创建中间模型
Favorite.objects.create(user=user, feed=target_feed, note='这个Feed干货满满！')
# 方式二：用User的关联字段添加，传入额外字段
user.favorite_feeds.add(target_feed, through_defaults={'note': '这个Feed干货满满！'})

• 取消收藏：

user.favorite_feeds.remove(target_feed)
# 或者通过中间模型删除
Favorite.objects.filter(user=user, feed=target_feed).delete()

• 查询用户收藏的Feed及额外信息：

# 直接查询中间模型，能拿到收藏时间、备注
user.favorite_set.all()
# 或者关联查询Feed详情
Favorite.objects.filter(user=user).select_related('feed')

额外实用提示 #

• 如果用Django REST Framework，序列化器里可以用PrimaryKeyRelatedField或嵌套序列化器来处理多对多关系；
• 模板中展示用户收藏的Feed，直接循环{{ user.favorite_feeds.all }}即可；
• 查询时用prefetch_related优化性能，避免N+1查询：

CustomUser.objects.prefetch_related('favorite_feeds').get(username='melissa')

内容的提问来源于stack exchange，提问作者Melissa Stewart