flask 更新数据库 --

更新书库库表

模型类（表）不是一成不变的，当你添加了新的模型类，或是在模型类中添加了新的字段，甚至是修改了字段的名称或类型，都需要更新表。在前面我们把数据库表比成盛放货物的货架，这些货架是固定生成的。当我们在操控程序（DBMS/ORM）变更了货架的结构时，仓库的货架也要根据变化相应进行调整。而且，当货架的结构产生变动时，我们还需要考虑如何处理货架上的货物（数据）。

当你在数据库的模型中添加了一个新的字段后，比如在Note模型里添加了一个存储笔记创建时间的timestamp字段。

class Note(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    timeStamp = db.Column(db.String(70), unique=True)
    body = db.Column(db.Text)
    def __repr__(self):
        # %r是用repr()方法处理对象，返回类型本身，而不进行类型转化
        return '<Note %r>' % self.body

此时模型类的定义中，加入了一个新的列，在命令行界面初始化数据时，只给了body赋了值，此时访问这个表的数据时，这条数据有一个字段没有初始化，就是表的结构变化了，需要更新这个表结构，才能正常访问。

这时在命令行里查看note表的字段时，会看到报错：

>>> Note.query.all()
sqlalchemy.exc.OperationalError: (sqlite3.OperationalError) no such column: note.timeStamp
[SQL: SELECT note.id AS note_id, note."timeStamp" AS "note_timeStamp", note.body AS note_body
FROM note]

 

这段错误信息支出note表中没有timestamp列，并在中括号里给出了查询所对应的SQL原语，之所以会出错，是因为数据库并不会随着模型的修改而自动更新。就像之前关于仓库的比喻，仓库里来了一批新类型的货物，可我们还没为他们安排响应的货架，这当然要出凑了，下面学习如果更新数据库。

重新生成表

重新调用create_all()方法并不会起到更新表或重新创建表的作用。如果你并不在意表中的数据，最简单的方法是使用drop_all()方法删除表以及其中的数据，然后再使用create_all()方法重新创建：

>>> db.drop_all()

>>> db.create_all()

这会清楚数据库里的原有数据，请勿在生产环境下使用。

为了方便开发，我们修改initdb命令函数的内容，为其增加一个—drop选项来支持删除表和数据库后进行重建，如下所示：

app.py: 支持删除表后重建

 

@app.cli.command()
@click.option('--drop', is_flag=True, help='Create after drop')
def initdb(drop):
    """Initialize the database."""
    if drop:
        click.confirm('This operation will delete the database, do you want to continue?', abort = True)
        db.drop_all()
        click.echo('Drop tables.')
    db.create_all()
    click.echo('Initialized database.')

在这个命令函数前，我们使用click提供的option装饰器为命令添加了一个—drop选项，将is_flag参数设为True可以将这个选项声明为布尔值标志（boolean flag）。--drop选项的值作为drop参数传入命令函数，如果提供了这个选项，那么drop的值将是True，否则为False。因为添加—drop选项会直接清空数据库内容，如果需要，也可以通过click.confirm()函数添加一个确认提示，这样只有输入y或yes才会继续执行操作。

下面来用这个命令重建数据库和表：

(Lenovo-ezd1lI9Y) D:\flask\FLASK_PRACTICE\DataBase>flask initdb --drop
This operation will delete the database, do you want to continue? [y/N]: y
Drop tables.
Initialized database.

 

当使用SQLite时，直接删除data.db文件和调用drop_all()方法效果相同，而且更直接，不容易出错。