如何让通用ListView只显示用户的列表?
我刚开始学习Django,这次是第一次使用基于类的视图。我想用通用的ListView来显示一个用户拥有的“表格”列表。目前我已经让它显示了数据库中的所有表格,但我只想显示当前登录用户的表格。
这是我的视图代码:
from django.shortcuts import render
from django.http import HttpResponse
from django.views import generic
from vtables.models import Vtable
class TableListView(generic.ListView):
model = Vtable
context_object_name = 'table_list'
def get_context_data(self, **kwargs):
context = super(TableListView, self).get_context_data(**kwargs)
return context
这是我的模型代码:
from django.db import models
from django.contrib.auth.models import User
class Vtable(models.Model):
user = models.ForeignKey(User)
table_name = models.CharField(max_length=200)
added_date = models.DateTimeField('date added')
class Vdata(models.Model):
table_id = models.ForeignKey(Vtable)
table_pk = models.IntegerField()
column_1 = models.CharField(max_length=200)
column_2 = models.CharField(max_length=200)
added_date = models.DateTimeField('date added')
我承认,我不太明白这一行代码的作用:
context = super(TableListView, self).get_context_data(**kwargs)
但我怀疑我需要修改这个部分?还是说我需要在我的模板里加个if语句?可能像这样:{% if request.user == vtable.user %}
?
编辑:这是我的模板代码:
{% extends "base.html" %}
{% load staticfiles %}
{% block content %}
{% if request.user.is_authenticated %}
<img src="{{ request.user.profile.profile_image_url }}"/>
<a href="/accounts/logout/" class="pull-right">Logout</a>
{% if request.user.first_name or request.user.last_name %}
{{ request.user.first_name }} {{ request.user.last_name }}
{% else %}
{{ request.user.username }}
{% endif %}
{% if request.user.profile.account_verified %}
(verified)
{% else %}
(unverified)
{% endif %}
<h1>Tables</h1>
<ul>
{% for vtable in table_list %}
<li>{{ vtable.user }}, {{ vtable.table_name }}</li>
{% endfor %}
</ul>
{% else %}
<a href="/accounts/login/" class="pull-right">Login</a>
{% endif %}
{% endblock %}
1 个回答
6
也许,可以重写 get_queryset() 方法:
def get_queryset(self):
return self.model.objects.filter(user=self.request.user)