关于“Python”的核心知识点整理大全59

19.3.2 将数据关联到用户

现在，需要将数据关联到提交它们的用户。我们只需将最高层的数据关联到用户，这样更低 层的数据将自动关联到用户。例如，在项目“学习笔记”中，应用程序的最高层数据是主题，而 所有条目都与特定主题相关联。只要每个主题都归属于特定用户，我们就能确定数据库中每个条 目的所有者。

下面来修改模型Topic，在其中添加一个关联到用户的外键。这样做后，我们必须对数据库 进行迁移。最后，我们必须对有些视图进行修改，使其只显示与当前登录的用户相关联的数据。

1. 修改模型Topic

对models.py的修改只涉及两行代码：

models.py

from django.db import models
from django.contrib.auth.models import User
class Topic(models.Model):
 """用户要学习的主题"""
 text = models.CharField(max_length=200)
 date_added = models.DateTimeField(auto_now_add=True)
 owner = models.ForeignKey(User)
 def __str__(self):
 """返回模型的字符串表示"""
 return self.text
class Entry(models.Model):
 --snip--

我们首先导入了django.contrib.auth中的模型User，然后在Topic中添加了字段owner，它建 立到模型User的外键关系。

2. 确定当前有哪些用户

我们迁移数据库时，Django将对数据库进行修改，使其能够存储主题和用户之间的关联。为 执行迁移，Django需要知道该将各个既有主题关联到哪个用户。最简单的办法是，将既有主题都 关联到同一个用户，如超级用户。为此，我们需要知道该用户的ID。

下面来查看已创建的所有用户的ID。为此，启动一个Django shell会话，并执行如下命令：

(venv)learning_log$ python manage.py shell
1 >>> from django.contrib.auth.models import User
2 >>> User.objects.all()
[<User: ll_admin>, <User: eric>, <User: willie>]
3 >>> for user in User.objects.all():
... print(user.username, user.id)
...
ll_admin 1
eric 2
willie 3
>>>

在1处，我们在shell会话中导入了模型User。然后，我们查看到目前为止都创建了哪些用户 （见2）。输出中列出了三个用户：ll_admin、eric和willie。

在3处，我们遍历用户列表，并打印每位用户的用户名和ID。Django询问要将既有主题关联 到哪个用户时，我们将指定其中的一个ID值。

3. 迁移数据库

知道用户ID后，就可以迁移数据库了。

1 (venv)learning_log$ python manage.py makemigrations learning_logs
2 You are trying to add a non-nullable field 'owner' to topic without a default;
we can't do that (the database needs something to populate existing rows).
3 Please select a fix:
1) Provide a one-off default now (will be set on all existing rows)
2) Quit, and let me add a default in models.py
4 Select an option: 1
5 Please enter the default value now, as valid Python
The datetime and django.utils.timezone modules are available, so you can do e.g. timezone.now()
6 >>> 1
Migrations for 'learning_logs':
 0003_topic_owner.py:
 - Add field owner to topic

我们首先执行了命令makemigrations（见1）。在2处的输出中，Django指出我们试图给既有 模型Topic添加一个必不可少（不可为空）的字段，而该字段没有默认值。在3处，Django给我 们提供了两种选择：要么现在提供默认值，要么退出并在models.py中添加默认值。在4处，我们 选择了第一个选项，因此Django让我们输入默认值（见5）。

为将所有既有主题都关联到管理用户ll_admin，我输入了用户ID值1（见6）。并非必须使用 超级用户，而可使用已创建的任何用户的ID。接下来，Django使用这个值来迁移数据库，并生成 了迁移文件0003_topic_owner.py，它在模型Topic中添加字段owner。

现在可以执行迁移了。为此，在活动的虚拟环境中执行下面的命令：

(venv)learning_log$ python manage.py migrate
Operations to perform:
 Synchronize unmigrated apps: messages, staticfiles
 Apply all migrations: learning_logs, contenttypes, sessions, admin, auth
--snip--
Running migrations:
 Rendering model states... DONE
1 Applying learning_logs.0003_topic_owner... OK
(venv)learning_log$

Django应用新的迁移，结果一切顺利（见1）。

为验证迁移符合预期，可在shell会话中像下面这样做：

1 >>> from learning_logs.models import Topic
2 >>> for topic in Topic.objects.all():
... print(topic, topic.owner)
...
Chess ll_admin
Rock Climbing ll_admin
>>>

我们从learning_logs.models中导入Topic（见1），再遍历所有的既有主题，并打印每个主 题及其所属的用户（见2）。正如你看到的，现在每个主题都属于用户ll_admin。

注意

你可以重置数据库而不是迁移它，但如果这样做，既有的数据都将丢失。一种不错的做 法是，学习如何在迁移数据库的同时确保用户数据的完整性。如果你确实想要一个全新 的数据库，可执行命令python manage.py flush，这将重建数据库的结构。如果你这样做， 就必须重新创建超级用户，且原来的所有数据都将丢失。

19.3.3 只允许用户访问自己的主题

当前，不管你以哪个用户的身份登录，都能够看到所有的主题。我们来改变这种情况，只向 用户显示属于自己的主题。 在views.py中，对函数topics()做如下修改：

views.py

--snip--
@login_required
def topics(request):
 """显示所有的主题"""
 topics = Topic.objects.filter(owner=request.user).order_by('date_added')
 context = {'topics': topics}
 return render(request, 'learning_logs/topics.html', context)
--snip--

用户登录后，request对象将有一个user属性，这个属性存储了有关该用户的信息。代码 Topic.objects.filter(owner=request.user)让Django只从数据库中获取owner属性为当前用户的 Topic对象。由于我们没有修改主题的显示方式，因此无需对页面topics的模板做任何修改。 要查看结果，以所有既有主题关联到的用户的身份登录，并访问topics页面，你将看到所有 的主题。然后，注销并以另一个用户的身份登录，topics页面将不会列出任何主题。

19.3.4 保护用户的主题

我们还没有限制对显示单个主题的页面的访问，因此任何已登录的用户都可输入类似于 http://localhost:8000/topics/1/的URL，来访问显示相应主题的页面。

你自己试一试就明白了。以拥有所有主题的用户的身份登录，访问特定的主题，并复制该页 面的URL，或将其中的ID记录下来。然后，注销并以另一个用户的身份登录，再输入显示前述主 题的页面的URL。虽然你是以另一个用户登录的，但依然能够查看该主题中的条目。 为修复这种问题，我们在视图函数topic()获取请求的条目前执行检查：

views.py

from django.shortcuts import render
1 from django.http import HttpResponseRedirect, Http404
from django.core.urlresolvers import reverse
--snip--
@login_required
def topic(request, topic_id):
 """显示单个主题及其所有的条目"""
 topic = Topic.objects.get(id=topic_id)
 # 确认请求的主题属于当前用户
2 if topic.owner != request.user:
 raise Http404 
entries = topic.entry_set.order_by('-date_added')
 context = {'topic': topic, 'entries': entries}
 return render(request, 'learning_logs/topic.html', context)
--snip--

服务器上没有请求的资源时，标准的做法是返回404响应。在这里，我们导入了异常Http404 （见1），并在用户请求它不能查看的主题时引发这个异常。收到主题请求后，我们在渲染网页前 检查该主题是否属于当前登录的用户。如果请求的主题不归当前用户所有，我们就引发Http404 异常（见2），让Django返回一个404错误页面。

现在，如果你试图查看其他用户的主题条目，将看到Django发送的消息Page Not Found。在 第20章，我们将对这个项目进行配置，让用户看到更合适的错误页面。

19.3.5 保护页面 edit_entry

页面edit_entry的URL为http://localhost:8000/edit_entry/entry_id/，其中entry_id是一 个数字。下面来保护这个页面，禁止用户通过输入类似于前面的URL来访问其他用户的条目：

views.py

我们获取指定的条目以及与之相关联的主题，然后检查主题的所有者是否是当前登录的用 户，如果不是，就引发Http404异常。

19.3.6 将新主题关联到当前用户

当前，用于添加新主题的页面存在问题，因此它没有将新主题关联到特定用户。如果你尝试 添加新主题，将看到错误消息IntegrityError，指出learning_logs_topic.user_id不能为NULL。 Django的意思是说，创建新主题时，你必须指定其owner字段的值。

由于我们可以通过request对象获悉当前用户，因此存在一个修复这种问题的简单方案。请 添加下面的代码，将新主题关联到当前用户：

views.py