如何添加额外的上下文& Django通用视图中的queryset字段?

时间:2015-10-04 09:38:45

标签: django django-views django-queryset django-generic-views django-context

我正在构建Django应用程序,提交链接和投票功能。

我想显示用户在用户详细信息页面中投票的所有链接。我可以使用以下命令在python shell中检索它们:

Link.objects.filter(votes__voter=user)

但我不知道如何将此作为额外的上下文字段添加到视图中。以下是代码:

models.py

class Link(models.Model):
    title       = models.CharField(max_length=200)
    submitter   = models.ForeignKey(User)
    submit_date = models.DateTimeField(auto_now_add=True)
    up_votes    = models.IntegerField(default=0, blank=True, db_index=True)
    ...

class UserProfile(models.Model):
    user        = models.OneToOneField(User, unique=True)
    ...

class Vote(models.Model):
    voter = models.ForeignKey(User)
    link = models.ForeignKey(Link, related_name='votes')
    ...

views.py

class UserProfileDetailView(DetailView):
    model       = get_user_model()
    slug_field  = "username"
    template_name = "user_detail.html"


    def get_object(self, queryset=None):
        user    = super(UserProfileDetailView, self).get_object(queryset)s
        UserProfile.objects.get_or_create(user=user)
        return user

user_detail.html

...
{% if object == request.user and request.user.is_authenticated %}
    <p><a href='{% url "edit_profile" %}'>Edit My Profile</a></p>
{% endif %}

{% if link in voted_list %}
    <p><a href='{% url "link_detail" %}'>{{ link.title }}</a></p>
{% endif %}
...

我无法弄清楚如何根据this链接添加额外字段。我不知道我需要定义新的查询集,定义新对象或只指向新的上下文。如果你能为我点亮这一点,我会更感激。

1 个答案:

答案 0 :(得分:9)

您只需覆盖get_context_data()中的UserProfileDetailView功能即可添加额外的上下文。

在这里,我们将添加一个额外的变量voted_links,其中包含当前用户投票的所有链接。

class UserProfileDetailView(DetailView):     

    def get_context_data(self, **kwargs):
        context = super(UserProfileDetailView, self).get_context_data(**kwargs) # get the default context data
        context['voted_links'] = Link.objects.filter(votes__voter=self.request.user) # add extra field to the context
        return context