查看用户是否在列表中添加了书签

Question

我在 Django 中有一个 ListView，我在其中列出了所有对象。一些对象由经过身份验证的用户添加书签。

每个对象都有一个关系bookmarks = GenericRelation(Bookmark)。

如何检查看到ListView 的用户是否为每个特定对象添加了书签？

我认为如果我在模板中检查它会花费很多 sql 查询。

编辑

我的Bookmark 模特：

class Bookmark(models.Model):
    content_type = models.ForeignKey(ContentType)
    object_id = models.PositiveIntegerField()
    content_object = generic.GenericForeignKey()
    user = models.ForeignKey(settings.AUTH_USER_MODEL)

我的Post 模特：

class Post(models.Model):
    title = models.CharField(max_length=100)
    user = models.ForeignKey(settings.AUTH_USER_MODEL)
    bookmarks = generic.GenericRelation(Bookmark)

我的列表视图：

class PostListView(ListView):
    model = Post
    template_name = 'resource/post_list.html'

我的猜测是我需要创建一个方法get_queryset()，在该方法中我为每个对象检查是否存在带有user=self.request.user 的书签对象。

编辑 2

我不想过滤任何东西。我想要模型中的所有对象，但是当我在模板中循环我的查询集时，我想确定经过身份验证的用户是否为特定对象添加了书签。所以我想我需要用一个额外的字段来填充查询集中的每个对象，告诉用户是否为特定对象添加了书签。我只是不知道该怎么做。

Answer 1

查看the documentation for Django's ListView class，我发现这个类继承自django.views.generic.list.MultipleObjectMixin 等。这意味着你的直觉是正确的。您将需要覆盖 ListView 类从 MultipleObjectMixin 类继承的 get_queryset() 方法。

From the documentation about get_queryset():

获取此视图的项目列表。这必须是一个可迭代的并且可能是一个查询集（其中将启用特定于查询集的行为）。

这只是意味着您应该重写此方法以返回包含您希望 ListView 向用户显示的对象的查询集。您的预期行为是过滤 ListView 中显示的 Post 对象，以仅显示用户具有现有书签的 Post 对象。

还有一个额外的问题是如何获取用户对象进行过滤。 get_queryset() 方法似乎没有接收到request 对象，因此我们无法在该方法中直接访问request.user。在 Django ListView 的文档中，有一个简短的流程图，解释了为此视图调用了哪些方法，以及它的顺序。

根据此流程图，首先运行dispatch() 方法。更重要的是，它在get_queryset() 方法之前运行，因此我们可以存储来自dispatch 方法的数据，以供以后在get_queryset 方法中使用。您可以在视图的dispatch() 方法中获取用户的副本（documentation here）。

首先，重写dispatch() 方法，以便您可以保留request.user 的副本以供以后在get_queryset() 中使用。像这样：

def dispatch(self, request, *args, **kwargs):
    self.user = request.user
    return super(PostListView, self).dispatch(request, *args, **kwargs)

这是一个示例，说明如何覆盖 get_queryset() 以仅过滤具有与当前用户对象关联的书签的 Post 对象：

def get_queryset(self):
    queryset_with_all_posts = super(PostListView, self).get_queryset()
    queryset_for_only_this_user = queryset_with_all_posts.filter(bookmarks__user=self.user)
    return queryset_for_only_this_user

Answer 2

1) 使用 django extra https://docs.djangoproject.com/en/dev/ref/models/querysets/#extra:

def get_queryset(self):
    qs = super(PostListView, self).get_queryset()
    annotated_qs = qs.extra(select=OrderedDict([('bookmarked', 'SELECT CASE WHEN user_id = %s THEN "true" ELSE NULL END FROM resource_bookmark WHERE object_id = resource_post.id AND content_type_id = %s'), ('has_voted', 'provide_some_query_that_returns_one_column_and_one_row'), ]), select_params=[self.request.user.id, ContentType.objects.get_for_model(self.model).id])

    return annotated_qs

主要思想是，您可以在额外的选择语句中访问每行中的 some_app_post.bookmarks_id，使用该 ID 选择 bookmarks 对象并根据条件向该行添加额外的列 bookmarked 值bookmarks__user=self.request.user

Answer 3

我有一个更简单的设置，其中对象可以拥有粉丝（用户）作为多对多关系。我标记当前用户“喜欢”哪些项目的解决方案是在上下文中添加一个包含模型 ID 的列表，然后在模板中检查这个列表。这可能不是最有效的方法，但对我来说已经足够快了。

您可能需要对其进行调整以使其适应您的通用关系，但代码如下所示：

class PostListView(ListView):
    model = Post
    template_name = 'resource/post_list.html'

    def get_context_data(self, **kwargs):
        context = super(PostListView, self).get_context_data(**kwargs)
        if self.request.user.is_authenticated():
            context['user_favs'] = Post.objects.filter(
                bookmarks__username=self.request.user.username).values_list(
                'id', flat=True)
        return context

模板：

{% if item.id in user_favs %}
... Your special code here ...
{% endif %}