flask中如何生成遷移檔案
阿新 • • 發佈:2018-12-13
在flask網站開發中,如果直接對資料庫進行修改的話,風險比較高,最好的是由遷移檔案生成,這樣確保了資料的誤操作。
在Flask中可以使用Flask-Migrate擴充套件,來實現資料遷移。並且整合到Flask-Script中,所有操作通過命令就能完成。為了匯出資料庫遷移命令,Flask-Migrate提供了一個MigrateCommand類,可以附加到flask-script的manager物件上。
注意這個模組需要安裝
pip install flask-migrate
下面是實現程式碼,且檔名為zengsf.py:
#coding=utf-8from 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 版本號 回退到該版本