flask-migrate數據庫遷移

數據庫遷移

---

• 在Flask中可以使用Flask-Migrate擴展，來實現數據遷移。並且集成到Flask-Script中，所有操作通過命令就能完成。
• 為了導出數據庫遷移命令，Flask-Migrate提供了一個MigrateCommand類，可以附加到flask-script的manager對象上。

 

• 創建遷移倉庫

python3 database.py db init

• 創建遷移腳本

 

• 自動創建遷移腳本有兩個函數
  • upgrade()：函數把遷移中的改動應用到數據庫中。
  • downgrade()：函數則將改動刪除。
• 自動創建的遷移腳本會根據模型定義和數據庫當前狀態的差異，生成upgrade()和downgrade()函數的內容。
• 對比不一定完全正確，有可能會遺漏一些細節，需要進行檢查

python3 database.py db migrate -m 'initial migration'

• 更新數據庫

  # 更新
  python3 database.py db upgrade
  
  # 查看歷史版本
  python app.py db history
  
  # 回退
  python3 app.py db downgrade 版本號
  
  '''
  實際操作順序:
  1.python 文件 db init
  2.python 文件 db migrate -m"版本名(注釋)"
  3.python 文件 db upgrade 然后觀察表結構
  4.根據需求修改模型
  5.python 文件 db migrate -m"新版本名(注釋)"
  6.python 文件 db upgrade 然后觀察表結構
  7.若返回版本,則利用 python 文件 db history查看版本號
  8.python 文件 db downgrade(upgrade) 版本號
  
  '''

   

  ---

   

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

'''
MigrateCommand:遷移命令

第一步初始化：python3 demo6_數據庫遷移.py db init
第二步生成遷移版本文件：python3 demo6_數據庫遷移.py db migrate -m'initial_01'
第三步執行遷移：python3 demo6_數據庫遷移.py db upgrade|downgrade


'''

app = Flask(__name__)

#設置連接數據庫的URL
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/migratetest'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True

db = SQLAlchemy(app)

# 一、遷移初始化，第一個參數是flask實例， 第二個參數是數據庫實例
Migrate(app, db)
# 二、實例終端命令執行對象
manager = Manager(app)
# 三、為manager添加遷移命令
manager.add_command('db', MigrateCommand)




# 創建數據表模型----實例化后就是指向數據表中的一條記錄
class Role(db.Model):
    # 定義表名
    __tablename__ = 'roles'
    # 定義列對象
    id = db.Column(db.Integer, primary_key=True)
    nike_name = db.Column(db.String(64), unique=True)  # 修改字段名
    age = db.Column(db.Integer, nullable=True)  # 新增字段
    # height = db.Column(db.Integer)  # 新增字段, 刪除此字段
    user = db.relationship('User', backref='role')


    #repr()方法顯示一個可讀字符串
    def __repr__(self):
        return 'Role:%s'% self.name


class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True, index=True)
    email = db.Column(db.String(64),unique=True)
    password = db.Column(db.String(64))
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

    def __repr__(self):
        return 'User:%s'%self.name


@app.route('/')
def index():
    return "hello world"



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