Flask框架之Flask-Migrate数据库迁移

原创

样子2018 2023-04-01 03:43:51 ©著作权

文章标签 flask python 版本号 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者样子2018的原创作品，请联系作者获取转载授权，否则将追究法律责任

注意：flask_script与flask2.0不再兼容

一、安装 flask_migrate

pip install flask-migrate==2.5.3

二、示例代码

flask：1.0.2

flask-migrate：2.5.3

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

import os

class Config(object):
    ''' sqlalchemy 参数配置 '''
    # 配置数据库
    SQLALCHEMY_DATABASE_URI = 'mysql://test01:123456@127.0.0.1:3306/test01?charset=utf8'
    # 自动会追踪对象的修改并且发送信号
    SQLALCHEMY_TRACK_MODIFICATIONS = True
    # 输出sql语句
    SQLALCHEMY_ECHO = True

    SECRET_KEY = os.urandom(24)

app = Flask(__name__)
app.config.from_object(Config)

db = SQLAlchemy(app)

migrate = Migrate(app=app, db=db)

manager = Manager(app)
manager.add_command("db", MigrateCommand)

#定义角色
class Role(db.Model):
    # 定义表名
    __tablename__ = 'roles'
    # 定义列对象
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True)
    email = db.Column(db.String(64), unique=True)
    age = db.Column(db.String(64), unique=True)
    users = db.relationship("User")


#定义用户
class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), unique=True, index=True)
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))


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

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