1. 程式人生 > >flask中如何生成遷移檔案

flask中如何生成遷移檔案

  在flask網站開發中,如果直接對資料庫進行修改的話,風險比較高,最好的是由遷移檔案生成,這樣確保了資料的誤操作。  

  在Flask中可以使用Flask-Migrate擴充套件,來實現資料遷移。並且整合到Flask-Script中,所有操作通過命令就能完成。為了匯出資料庫遷移命令,Flask-Migrate提供了一個MigrateCommand類,可以附加到flask-script的manager物件上。

  注意這個模組需要安裝

pip install flask-migrate

  下面是實現程式碼,且檔名為zengsf.py:

#coding=utf-8
from flask import Flask from flask_sqlalchemy import SQLAlchemy from flask_migrate import Migrate,MigrateCommand from flask_script import Manager app = Flask(__name__) manager = Manager(app) app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:[email protected]:3306/zengsf' app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN
'] = True app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True db = SQLAlchemy(app) #第一個引數是Flask的例項,第二個引數是Sqlalchemy資料庫例項 Migrate(app,db) #manager是Flask-Script的例項,這條語句在flask-Script中新增一個db命令 manager.add_command('db',MigrateCommand) #定義模型Role class Role(db.Model): # 定義表名 __tablename__ = 'roles
' # 定義列物件 id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(64), unique=True) def __repr__(self): return 'Role:'.format(self.name) #定義使用者 class User(db.Model): __tablename__ = 'users' id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(64), unique=True, index=True) def __repr__(self): return 'User:'.format(self.username) if __name__ == '__main__': manager.run()

然後在終端輸入:

python zengsf.py db init   初始化資料庫,會生成一個叫migrations資料夾。

python zengsf.py db migrate    生成遷移檔案,但不改變資料庫。

python zengsf.py db upgrade    更新資料庫。

python zengsf.py db history     檢視歷史記錄

python zengsf.py db downgrade 版本號     回退到該版本