蓝图和视图

视图是一个应用对请求进行响应的函数。 Flask 通过模型把进来的请求 URL 匹配到 对应的处理视图。视图返回数据， Flask 把数据变成出去的响应。 Flask 也可以反 过来，根据视图的名称和参数生成 URL 。

创建蓝图

Blueprint 是一种组织一组相关视图及其他代码的方式。与把视图及其他 代码直接注册到应用的方式不同，蓝图方式是把它们注册到蓝图，然后在工厂函数中 把蓝图注册到应用。

Flaskr 有两个蓝图，一个用于认证功能，另一个用于博客帖子管理。每个蓝图的代码 都在一个单独的模块中。使用博客首先需要认证，因此我们先写认证蓝图。

flaskr/auth.py

  1. import functools
  3. from flask import (
  4.     Blueprint, flash, g, redirect, render_template, request, session, url_for
  5. )
  6. from werkzeug.security import check_password_hash, generate_password_hash
  8. from flaskr.db import get_db
  10. bp = Blueprint('auth', __name__, url_prefix='/auth')

这里创建了一个名称为 'auth' 的 Blueprint 。和应用对象一样， 蓝图需要知道是在哪里定义的，因此把 __name__ 作为函数的第二个参数。 url_prefix 会添加到所有与该蓝图关联的 URL 前面。

使用 app.register_blueprint() 导入并注册 蓝图。新的代码放在工厂函数的尾部返回应用之前。

flaskr/__init__.py

  1. def create_app():
  2.     app = ...
  3.     # existing code omitted
  5.     from . import auth
  6.     app.register_blueprint(auth.bp)
  8.     return app

认证蓝图将包括注册新用户、登录和注销视图。

第一个视图：注册

当用访问 /auth/register URL 时， register 视图会返回用于填写注册 内容的表单的 HTML 。当用户提交表单时，视图会验证表单内容，然后要么再次 显示表单并显示一个出错信息，要么创建新用户并显示登录页面。

现在只是编写视图代码，在下一页会编写生成 HTML 表单的模板。

flaskr/auth.py

  1. @bp.route('/register', methods=('GET', 'POST'))
  2. def register():
  3.     if request.method == 'POST':
  4.         username = request.form['username']
  5.         password = request.form['password']
  6.         db = get_db()
  7.         error = None
  9.         if not username:
  10.             error = 'Username is required.'
  11.         elif not password:
  12.             error = 'Password is required.'
  14.         if error is None:
  15.             try:
  16.                 db.execute(
  17.                     "INSERT INTO user (username, password) VALUES (?, ?)",
  18.                     (username, generate_password_hash(password)),
  19.                 )
  20.                 db.commit()
  21.             except db.IntegrityError:
  22.                 error = f"User {username} is already registered."
  23.             else:
  24.                 return redirect(url_for("auth.login"))
  26.         flash(error)
  28.     return render_template('auth/register.html')

这个 register 视图做了以下工作：

  1. @bp.route 关联了 URL /register 和 register 视图函数。当 Flask 收到一个指向 /auth/register 的请求时就会调用 register 视图并把其返回值作为响应。

  2. 如果用户提交了表单，那么 request.method 将会是 'POST' 。这咱情况下 会开始验证用户的输入内容。

  3. request.form 是一个特殊类型的 dict ，其映射了提交表单的键和值。表单中，用户将会输入其 username 和 password 。

  4. 验证 username 和 password 不为空。

  5. 如果验证成功，就把新用户的数据插入数据库。

    • db.execute 使用了带有 ? 占位符的 SQL 查询语句。占位符可以代替后面的元组参数中相应的值。使 用占位符的好处是会自动帮你转义输入值，以抵御 SQL 注入攻击 。

    • 因为安全原因，不能把密码明文储存在数据库中。而是应当使用 generate_password_hash() 生成安全的哈希值， 再把哈希值储存到数据库中。因为查询修改了数据，所以要使用 meth:db.commit() <sqlite3.Connection.commit> 保存修改。

    • 如果用户名已存在，会产生一个 sqlite3.IntegrityError 错误， 应当将该错误作为一个验证错误显示给用户。

  6. 用户数据保存后将转到登录页面。 url_for() 根据登录视图的名称生成相应的 URL 。与写固定的 URL 相比， 这样做的好处是如果以后需要修改该视图相应的 URL ，那么不用修改所有涉及到 URL 的代码。 redirect() 为生成的 URL 生成一个重定向响应。

  7. 如果验证失败，那么会向用户显示一个出错信息。 flash() 用于储存在渲染模块时可以调用的信息。

  8. 当用户最初访问 auth/register 时，或者注册出错时，应用显示一个注册 表单。 render_template() 会渲染一个包含 HTML 的模板。你会在教程的下一节 学习如何写这个模板。

登录

这个视图和上述 register 视图原理相同。

flaskr/auth.py

  1. @bp.route('/login', methods=('GET', 'POST'))
  2. def login():
  3.     if request.method == 'POST':
  4.         username = request.form['username']
  5.         password = request.form['password']
  6.         db = get_db()
  7.         error = None
  8.         user = db.execute(
  9.             'SELECT * FROM user WHERE username = ?', (username,)
  10.         ).fetchone()
  12.         if user is None:
  13.             error = 'Incorrect username.'
  14.         elif not check_password_hash(user['password'], password):
  15.             error = 'Incorrect password.'
  17.         if error is None:
  18.             session.clear()
  19.             session['user_id'] = user['id']
  20.             return redirect(url_for('index'))
  22.         flash(error)
  24.     return render_template('auth/login.html')

与 register 有以下不同之处：

  1. 首先需要查询用户并存放在变量中，以备后用。

    fetchone() 根据查询返回一个记录行。 如果查询没有结果，则返回 None 。后面还用到 fetchall() ，它返回包括所有结果的列表。

  2. check_password_hash() 以相同的方式哈希提交的 密码并安全的比较哈希值。如果匹配成功，那么密码就是正确的。

  3. session 是一个 dict ，它用于储存横跨请求的值。当验证 成功后，用户的 id 被储存于一个新的会话中。会话数据被储存到一个 向浏览器发送的 cookie 中，在后继请求中，浏览器会返回它。 Flask 会安全对数据进行 签名 以防数据被篡改。

现在用户的 id 已被储存在 session 中，可以被后续的请求使用。 请每个请求的开头，如果用户已登录，那么其用户信息应当被载入，以使其可用于 其他视图。

flaskr/auth.py

  1. @bp.before_app_request
  2. def load_logged_in_user():
  3.     user_id = session.get('user_id')
  5.     if user_id is None:
  6.         g.user = None
  7.     else:
  8.         g.user = get_db().execute(
  9.             'SELECT * FROM user WHERE id = ?', (user_id,)
  10.         ).fetchone()

bp.before_app_request() 注册一个 在视图函数之前运行的函数，不论其 URL 是什么。 load_logged_in_user 检查用户 id 是否已经储存在 session 中，并从数据库中获取用户数据，然后储存在 g.user 中。 g.user 的持续时间比请求要长。 如果没有用户 id ，或者 id 不存在，那么 g.user 将会是 None 。

注销

注销的时候需要把用户 id 从 session 中移除。 然后 load_logged_in_user 就不会在后继请求中载入用户了。

flaskr/auth.py

  1. @bp.route('/logout')
  2. def logout():
  3.     session.clear()
  4.     return redirect(url_for('index'))

在其他视图中验证

用户登录以后才能创建、编辑和删除博客帖子。在每个视图中可以使用 装饰器 来完成这个工作。

flaskr/auth.py

  1. def login_required(view):
  2.     @functools.wraps(view)
  3.     def wrapped_view(**kwargs):
  4.         if g.user is None:
  5.             return redirect(url_for('auth.login'))
  7.         return view(**kwargs)
  9.     return wrapped_view

装饰器返回一个新的视图，该视图包含了传递给装饰器的原视图。新的函数检查用户 是否已载入。如果已载入，那么就继续正常执行原视图，否则就重定向到登录页面。 我们会在博客视图中使用这个装饰器。

端点和 URL

url_for() 函数根据视图名称和发生成 URL 。视图相关联的名称亦称为 端点 ，缺省情况下，端点名称与视图函数名称相同。

例如，前文被加入应用工厂的 hello() 视图端点为 'hello' ，可以使用 url_for('hello') 来连接。如果视图有参数，后文会看到，那么可使用 url_for('hello', who='World') 连接。

当使用蓝图的时候，蓝图的名称会添加到函数名称的前面。上面的 login 函数 的端点为 'auth.login' ，因为它已被加入 'auth' 蓝图中。

下面请阅读 模板 。