Flask Web Development —— 数据库（下）

九、数据库在视图函数中的使用

在前面章节描述的数据库操做是能够在视图函数中直接使用的。示例5-5展现的新版本主页实现了用户输入名字并保存到数据库中。

示例5-5. hello.py：数据库在视图函数中的使用

@app.route('/', methods=['GET', 'POST']) 
def index():
    form = NameForm()
    if form.validate_on_submit():
        user = User.query.filter_by(username=form.name.data).first() 
        if user is None:
            user = User(username = form.name.data)
            db.session.add(user)
            session['known'] = False
        else:
            session['known'] = True
        session['name'] = form.name.data 
        form.name.data = ''
        return redirect(url_for('index'))
    return render_template('index.html',
        form = form, name = session.get('name'), known = session.get('known', False))

在这个已修改的应用程序版本中，每次提交一个名字应用程序都要使用filter_by()查询过滤器到数据库中检查一遍。known变量被写入到用户会话中，这样在重定向后信息就能够发送到模板来定制问候语。注意为了应用程序能够工做，必须建立好以前在Python shell中展现的那些数据库表。

新版本的相关模板在示例5-6中展现。这个模板使用known参数用于区分是已知用户仍是新用户的问候。

{% extends "base.html" %}
{% import "bootstrap/wtf.html" as wtf %}

{% block title %}Flasky{% endblock %}

{% block page_content %}


<div class="page-header">
  <h1>Hello, {% if name %}{{ name }}{% else %}Stranger{% endif %}!</h1> 
  {% if not known %}
  <p>Pleased to meet you!</p>
  {% else %}
  <p>Happy to see you again!</p>
  {% endif %}
</div>


{{ wtf.quick_form(form) }}
{% endblock %}

建议：若是你有克隆在GitHub上的应用程序，你如今能够运行git checkout 5b来切换到这个版本的应用程序。

十、与python shell的集成

每次shell会话启动都必须导入数据库实例和模型是很是单调乏味的工做。为了不不断重复这些导入，能够配置Flask-Script的shell命令来自动导入特定的对象。

shell命令须要注册make_context回调函数来将对象添加到导入列表。如示例5-7所示。

示例5-7. hello.py：增长shell上下文

from flask.ext.script import Shell 

def make_shell_context():
    return dict(app=app, db=db, User=User, Role=Role)
manager.add_command("shell", Shell(make_context=make_shell_context))

make_shell_context()函数注册应用程序和数据库的实例及模型，这样就能够自动导入到shell中了：

$ python hello.py shell
>>> app
<Flask 'app'>
>>> db
<SQLAlchemy engine='sqlite:////home/flask/flasky/data.sqlite'> 
>>> User
<class 'app.User'>

建议：若是你有克隆在GitHub上的应用程序，你如今能够运行git checkout 5c来切换到这个版本的应用程序。

十一、使用Flask-Migrate迁移数据库

随着开发进度不断向前，你会发现你的数据库模型须要更改，而当这种状况发生时须要更新数据库。

Flask-SQLAlchemy只有当数据库表不存在了才从模型建立它们，因此更新表的惟一途径就是销毁旧的表，固然这将致使全部数据库中的数据丢失。

有个更好的解决方案就是使用数据库迁移框架。和源码版本控制工具跟踪更改源码文件同样，数据库迁移框架跟踪更改数据库模型，而后将增量变化应用到数据库中。

SQLAlchemy的主要开发人员写了一个Alembic迁移框架，但咱们不直接使用Alembic，Flask应用可使用Flask-Migrate扩展，一个集成了Flask-Script来提供全部操做命令的轻量级Alembic包。

11.一、建立迁移仓库

首先，Flask-Migrate必须已经安装到虚拟环境中：

(venv) $ pip install flask-migrate

示例5-8展现扩展如何初始化。

示例5-8. hello.py：Flask-Migrate配置

from flask.ext.migrate import Migrate, MigrateCommand 

# ...

migrate = Migrate(app, db)
manager.add_command('db', MigrateCommand)

为了可使用数据库迁移命令，Flask-Migrate提供MigrateCommand类来链接Flask-Script的manager对象。在这个示例中使用db来链接到命令。

在数据库迁移能够维护以前，必须经过init子命令来建立一个迁移库：

(venv) $ python hello.py db init
  Creating directory /home/flask/flasky/migrations...done
  Creating directory /home/flask/flasky/migrations/versions...done
  Generating /home/flask/flasky/migrations/alembic.ini...done
  Generating /home/flask/flasky/migrations/env.py...done
  Generating /home/flask/flasky/migrations/env.pyc...done
  Generating /home/flask/flasky/migrations/README...done
  Generating /home/flask/flasky/migrations/script.py.mako...done
  Please edit configuration/connection/logging settings in
  '/home/flask/flasky/migrations/alembic.ini' before proceeding.

这个命令建立一个migrations文件夹，里面存放了全部迁移脚本。

建议：若是你有克隆在GitHub上的应用程序，你如今能够运行git checkout 5c来切换到这个版本的应用程序。

11.二、建立迁移脚本

在Alembic，数据库迁移工做由迁移脚本完成。这个脚本有两个函数，分别叫作upgrade()和downgrade()。upgrade()函数实施数据库更改，是迁移的一部分，downgrade()函数则删除它们。经过添加和删除数据库变化的能力，Alembic能够从新配置数据库从历史记录中的任什么时候间点。

Alembic迁移能够分别使用revision和migrate命令手动或自动建立。手动迁移经过由开发人员使用Alembic的Operations对象指令实现的空upgrade()和downgrade()函数建立迁移框架脚本。另外一方面，自动迁移经过寻找模型定义和数据库当前状态间的不一样为upgrade()和downgrade()生成代码。

警告：自动迁移并不老是准确的，能够忽略一些细节。因此应该常常审查一下自动生成的迁移脚本。

migrate子命令建立自动迁移脚本：

(venv) $ python hello.py db migrate -m "initial migration"
INFO  [alembic.migration] Context impl SQLiteImpl.
INFO  [alembic.migration] Will assume non-transactional DDL.
INFO  [alembic.autogenerate] Detected added table 'roles'
INFO  [alembic.autogenerate] Detected added table 'users'
INFO  [alembic.autogenerate.compare] Detected added index
'ix_users_username' on '['username']'
  Generating /home/flask/flasky/migrations/versions/1bc
  594146bb5_initial_migration.py...done

建议：若是你有克隆在GitHub上的应用程序，你如今能够运行git checkout 5c来切换到这个版本的应用程序。注意，你不须要为这个应用生成migrations，全部的迁移脚本都包含在版本库中。

11.三、更新数据库

一旦迁移脚本被审查且接受，就可使用db upgrade命令更新到数据库中：

(venv) $ python hello.py db upgrade
INFO  [alembic.migration] Context impl SQLiteImpl.
INFO  [alembic.migration] Will assume non-transactional DDL.
INFO  [alembic.migration] Running upgrade None -> 1bc594146bb5, initial migration

第一次迁移实际上至关于调用db.create_all()，但在后续迁移中，upgrade命令对表实施更新操做但不影响表中的内容。

数据库的设计和使用是很是重要的，事实上整本书都是围绕这个主题来写的。你应该把本章做为一个概述来研究；更高级的主题将在之后的章节中讨论。下一章致力于讲解发送电子邮件。