Flask学习——数据库迁移

0x01

前言

用过Django的小伙伴都知道，Django的ORM是自带的，比较特殊，而且集成了很多功能，比如数据库迁移…
何为ORM，个人之见解，简化sql语句的书写，将关系型数据库的一张张表转化为了python的类，最大的好处是简化了学习成本，不会sql语句的程序员也能平滑的使用数据库，并且天生防sql注入。flask可以使用SQLAlchemy，包名为flask-sqlalchemy.至于具体用法翻译的官方文档讲的也足够清楚。大家直接谷歌跟着文档就能搞定，本篇主要讲如何进行数据库迁移。

flask的小工具

大家从Django转至Flask最明显的一个差异就是，flask直接运行就可以启动服务，Django是使用了runserver来进行运行。至于好坏大家自行斟酌。这里提一个插件。flask-script。pip安装即可。

from flask_script import Manager
......
manager = Manager(app)
......
#app.run()  #注释这句
manager.run()

这样，启动的方式变为 python main.py runserver
这似乎有点画蛇添足，实则不然，这和我们下面要讲的数据库迁移有很大的联系。

数据库迁移

需要用的的插件 flask-migrate
安装之

main.py

from flask import Flask
from flask_script import Manager
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate,MigrateCommand

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:opt/db.sqlite' #使用sqlite3数据库
db = SQLAlchemy(app)  #ORM

migrate = Migrate(app,db)
manager = Manager(app)
manager.add_comman('db',MigrateCommand) #添加db 命令（runserver的用法）

#一个用户表
class User(db.Model):
...
...
省略

if __name__ == '__main__':
    manager.run()

这样，准备工作已经就绪。
运行python main.py db init创建数据表。并且会在你项目下生成migrations/目录，保存你数据库每次变更的内容。

修改User类。
python main.py db migrate 提交修改
python main.py db upgrade 执行修改
再看User表，已经改变。
python main.py db downgrade 回退修改

有一点注意的：SQLite3不能删除有值的列…所以,大家就不要删除列了…