Flask系列教程（8）——Flask-Migrate

Flask-Migrate

在实际的开发环境中，经常会发生数据库修改的行为。一般我们修改数据库不会直接手动的去修改，而是去修改ORM对应的模型，然后再把模型映射到数据库中。这时候如果有一个工具能专门做这种事情，就显得非常有用了，而flask-migrate就是做这个事情的。flask-migrate是基于Alembic进行的一个封装，并集成到Flask中，而所有的迁移操作其实都是Alembic做的，他能跟踪模型的变化，并将变化映射到数据库中。

使用Flask-Migrate需要安装，命令如下：

pip install flask-migrate

要让Flask-Migrate能够管理app中的数据库，需要使用Migrate(app,db)来绑定app和数据库。假如现在有以下app文件：

from flask import Flaskfrom flask_sqlalchemy import SQLAlchemyfrom constants import DB_URIfrom flask_migrate import Migrateapp = Flask(__name__)app.config['SQLALCHEMY_DATABASE_URI'] = DB_URIapp.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = Truedb = SQLAlchemy(app)# 绑定app和数据库migrate = Migrate(app,db)class User(db.Model):    id = db.Column(db.Integer,primary_key=True)    username = db.Column(db.String(20))    addresses = db.relationship('Address',backref='user')class Address(db.Model):    id = db.Column(db.Integer,primary_key=True)    email_address = db.Column(db.String(50))    user_id = db.Column(db.Integer,db.ForeignKey('user.id'))db.create_all()@app.route('/')def hello_world():    return 'Hello World!'if __name__ == '__main__':    app.run()

之后，就可以在命令行中映射ORM了。要操作当前的flask app，首先需要将当前的app导入到环境变量中：

# windows$env:FLASK_APP='your_app.py'#linux/unixexport FLASK_APP='your_app.py'

将当前的app导入到环境变量中后，接下来就是需要初始化一个迁移文件夹：

flask db init

然后再把当前的模型添加到迁移文件中：

flask db migrate

最后再把迁移文件中对应的数据库操作，真正的映射到数据库中：

flask db upgrade

以上是通过加载当前的app到环境变量的做法，这种做法有点麻烦，每次都要设置环境变量。还有一种方法，可以直接通过flask-script的方式进行调用。现在重构之前的项目，设置为以下的目录结构：
以下对各个文件的作用进行解释：

constants.py文件：常量文件，用来存放数据库配置。

# constants.pyHOSTNAME = '127.0.0.1'PORT = '3306'DATABASE = 'xt_flask_migrate'USERNAME = 'root'PASSWORD = 'root'DB_URI = 'mysql+mysqldb://{}:{}@{}:{}/{}'.format(USERNAME,PASSWORD,HOSTNAME,PORT,DATABASE)

ext.py文件：把db变量放到一个单独的文件，而不是放在主app文件。这样做的目的是为了在大型项目中如果db被多个模型文件引用的话，会造成from your_app import db这样的方式，但是往往也在your_app.py中也会引入模型文件定义的类，这就造成了循环引用。所以最好的办法是把它放在不依赖其他模块的独立文件中。

# ext.pyfrom flask_sqlalchemy import SQLAlchemydb = SQLAlchemy()

models.py文件：模型文件，用来存放所有的模型，并且注意，因为这里使用的是flask-script的方式进行模型和表的映射，因此不需要使用db.create_all()的方式创建数据库。

# models.pyfrom ext import dbclass User(db.Model):    id = db.Column(db.Integer,primary_key=True)    username = db.Column(db.String(50))    addresses = db.relationship('Address',backref='user')    def __init__(self,username):        self.username = usernameclass Address(db.Model):    id = db.Column(db.Integer,primary_key=True)    email_address = db.Column(db.String(50))    user_id = db.Column(db.Integer,db.ForeignKey('user.id'))    def __init__(self,email_address):        self.email_address = email_address

manage.py文件：这个文件用来存放映射数据库的命令，MigrateCommand是flask-migrate集成的一个命令，因此想要添加到脚本命令中，需要采用manager.add_command('db',MigrateCommand)的方式，以后运行python manage.py db xxx的命令，其实就是执行MigrateCommand。

# manage.pyfrom flask_migrate import Migrate,MigrateCommandfrom ext import dbfrom flask_script import Managerfrom flask import Flaskfrom constants import DB_URIimport modelsapp = Flask(__name__)app.config['SQLALCHEMY_DATABASE_URI'] = DB_URIapp.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = Truedb.init_app(app)migrate = Migrate(app,db)manager = Manager(app)manager.add_command('db',MigrateCommand)if __name__ == '__main__':    manager.run()

flaskmigrate.py文件：这个是主app文件，运行文件。并且因为db被放到另外一个文件中，所以使用db.init_app(app)的方式来绑定数据库。

# flaskmigrate.pyfrom flask import Flaskfrom ext import dbapp = Flask(__name__)db.init_app(app)@app.route('/')def hello_world():    return 'Hello World!'if __name__ == '__main__':    app.run()

之后运行命令来初始化迁移文件：

python manage.py db init

然后运行命令来将模型的映射添加到文件中：

python manage.py db migrate

最后添加将映射文件真正的映射到数据库中：

python manage.py db upgrade

对Python和Flask感兴趣，可以加群：482869582一起讨论和学习哦。