使用Flask-Migrate拓展數(shù)據(jù)庫(kù)表結(jié)構(gòu)的方法

這篇文章給大家分享的是有關(guān)使用Flask-Migrate拓展數(shù)據(jù)庫(kù)表結(jié)構(gòu)的方法的內(nèi)容。小編覺(jué)得挺實(shí)用的，因此分享給大家做個(gè)參考，一起跟隨小編過(guò)來(lái)看看吧。

前言

在我們用 sqlchemy 模塊創(chuàng)建完幾個(gè)表時(shí)，如果在實(shí)際生產(chǎn)環(huán)境中，需要對(duì)表結(jié)構(gòu)進(jìn)行更改，應(yīng)該怎么辦呢？總不能把表刪除了吧，這樣數(shù)據(jù)就會(huì)丟失了。

更好的解決辦法是使用數(shù)據(jù)庫(kù)遷移框架，它可以追蹤數(shù)據(jù)庫(kù)模式的變化，然后把變動(dòng)應(yīng)用到數(shù)據(jù)庫(kù)中。

在Flask中可以使用Flask-Migrate擴(kuò)展，來(lái)實(shí)現(xiàn)數(shù)據(jù)遷移。并且集成到Flask-Script中，所有操作通過(guò)命令就能完成。

操作示例：

1.首先需要安裝 flask-migrate 模塊，當(dāng)然還有 flask-script 模塊。

pip3 install flask-migrate
pip3 install flask-script

2.代碼：

#coding=utf-8
from flask import Flask

from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate,MigrateCommand
from flask_script import Shell,Manager

app = Flask(__name__)
manager = Manager(app)

app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/Flask_test'
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
db = SQLAlchemy(app)

#第一個(gè)參數(shù)是Flask的實(shí)例，第二個(gè)參數(shù)是Sqlalchemy數(shù)據(jù)庫(kù)實(shí)例
migrate = Migrate(app,db) 

#manager是Flask-Script的實(shí)例，這條語(yǔ)句在flask-Script中添加一個(gè)db命令
manager.add_command('db',MigrateCommand)

#定義模型Role
class Role(db.Model):
  # 定義表名
  __tablename__ = 'roles'
  # 定義列對(duì)象
  id = db.Column(db.Integer, primary_key=True)
  name = db.Column(db.String(64), unique=True)
  user = db.relationship('User', backref='role')

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

#定義用戶
class User(db.Model):
  __talbe__ = 'users'
  id = db.Column(db.Integer, primary_key=True)
  username = db.Column(db.String(64), unique=True, index=True)
  #設(shè)置外鍵
  role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

  def __repr__(self):
    return 'User:'.format(self.username)


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

主要注意第8行-20行，48行。要把原來(lái)的代碼改成這樣。

3.命令行的操作

創(chuàng)建遷移倉(cāng)庫(kù)

這個(gè)命令會(huì)創(chuàng)建 migirations 文件夾，所有遷移文件都放在里面

python3 xxx.py db init

注：db 是20行決定了，可以更改

自動(dòng)創(chuàng)建遷移腳本

python3 xxx.py db migrate -m"版本名(注釋)"

更新數(shù)據(jù)庫(kù)

python3 xxx.py db upgrade

當(dāng)我們需要修改表結(jié)構(gòu)時(shí)，直接在 xxx.py 里直接增刪相應(yīng)的代碼

修改完成后，繼續(xù)創(chuàng)建新的遷移腳本

python 文件 db migrate -m"新版本名(注釋)"

更新數(shù)據(jù)庫(kù)

python3 xxx.py db upgrade

更新完之后，其實(shí)就是提交操作，類似于 git 添加一個(gè)新的版本。

但是，如果我們想返回歷史的版本，應(yīng)該怎么操作呢？

先查看版本號(hào)

python xxx.py db history

然后記住想要返回的版本號(hào)。

返回指定的版本

python xxx.py db downgrade(upgrade) 版本號(hào)

然后打開(kāi)你的代碼，可以發(fā)現(xiàn)他自動(dòng)復(fù)原了！

感謝各位的閱讀！關(guān)于“使用Flask-Migrate拓展數(shù)據(jù)庫(kù)表結(jié)構(gòu)的方法”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，讓大家可以學(xué)到更多知識(shí)，如果覺(jué)得文章不錯(cuò)，可以把它分享出去讓更多的人看到吧！