flask-sqlalchemy 迁移数据（生成数据库表）与 查询数据教程

1, 生成表

db.Model主要用于数据库的增删改查操作， 构建表交给db.Table完成

安装 pip install flask-migrate

from datetime import datetime
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.dialects.mysql import TINYINT, BIGINT, VARCHAR, CHAR, DATETIME, INTEGER
# 使用原生sqlalchemy创建字段
app = create_app()
# 创建数据库连接对象
db = SQLAlchemy(app)
# 构建表
t_user = db.Table('user_basic',
                  db.Column('user_id', BIGINT(10, unsigned=True), nullable=False, primary_key=True, autoincrement=True, comment='主键'),
                  db.Column('status', TINYINT(1), nullable=False, default=1, comment="状态"),
                  # db.Column('status1', TINYINT(1), nullable=False, default=1, comment="状态1"),
                  db.Column('mobile', CHAR(11), nullable=False, unique=True, comment='手机号'),
                  db.Column('create_time', DATETIME, nullable=False, default=datetime.now, comment='创建时间'),
                  db.Column('update_time', DATETIME, nullable=False, default=datetime.now, onupdate=datetime.now, comment='更新时间'),
                  # 注意: 如果有外键, 定义方式和普通字段一样, 可以添加索引提高性能
                  # db.Column('leader_id', BIGINT(10, unsigned=True), default=0, comment='上级的id', index=True),
                  mysql_engine='MyISAM',
                  mysql_charset='utf8mb4')


if __name__ == '__main__':
    db.drop_all()
    db.create_all() #调用create_all()能找到对应的db.Tabke然后生成表

注释：逻辑外键添加index = True，生成外键，查询速度快

2，数据增删查改的模型：

from datetime import datetime

from flask_sqlalchemy import SQLAlchemy

def create_app(): #在其他地方封装的
    app = Flask(__name__)
    app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:[email protected]:3306/sqlalchemy_test?charset=utf8'
    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

    return app

app = create_app()

# 为了避免和创建表的db产生冲突, 创建专门用于数据操作的SQLAlchemy对象
model_db = SQLAlchemy(app)


class User(model_db.Model):  # db.Model主要用于数据的增删改查, 构建表交给db.Table去完成
    __tablename__ = 'user_basic'
    # 由于模型不用于建表, 所以类型不需要设置的很严谨, 并可以省略大部分字段细节(除了default参数)
    user_id = model_db.Column(model_db.Integer, primary_key=True)
    status = model_db.Column(model_db.Integer, default=1)
    mobile = model_db.Column(model_db.String(11))
    create_time = model_db.Column(model_db.DateTime, default=datetime.now)
    update_time = model_db.Column(model_db.DateTime, default=datetime.now)


@app.route('/')
def index():
    user1 = User()
    user1.mobile = '18912341234'
    model_db.session.add(user1)
    model_db.session.commit()
    return 'index'


if __name__ == '__main__':
    app.run(debug=True)

注释：建表时如果字段中有default字段时，查询等操作的模型类也要写，不写会报错

2，迁移数据, 添加

当数据库升级，如增加字段，修改字段类型等，（删除外键等）

from datetime import datetime
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.dialects.mysql import TINYINT, BIGINT, VARCHAR, CHAR, DATETIME, INTEGER
from tool import create_app
from flask_migrate import Migrate

app = create_app()
# 创建数据库连接对象
db = SQLAlchemy(app)

# 初始化迁移器
Migrate(app, db)

# 构建表
t_user = db.Table('user_basic',
                  db.Column('user_id', BIGINT(10, unsigned=True), nullable=False, primary_key=True, autoincrement=True, comment='主键'),
                  db.Column('status', TINYINT(1), nullable=False, default=1, comment="状态"),
                  # db.Column('status1', TINYINT(1), nullable=False, default=1, comment="状态1"),
                  db.Column('mobile', CHAR(11), nullable=False, unique=True, comment='手机号'),
                  db.Column('create_time', DATETIME, nullable=False, default=datetime.now, comment='创建时间'),
                  db.Column('update_time', DATETIME, nullable=False, default=datetime.now, onupdate=datetime.now, comment='更新时间'),
                  # 注意: 如果有外键, 定义方式和普通字段一样, 可以添加索引提高性能
                  # db.Column('leader_id', BIGINT(10, unsigned=True), default=0, comment='上级的id', index=True),
                  mysql_engine='MyISAM',
                  mysql_charset='utf8mb4')


if __name__ == '__main__':
    # db.drop_all()
    # db.create_all()
    pass

View Code

修改好后执行步骤，如果是已经改过，可以只执行最后两步

1，export FLASK\_APP = init\_db.py #指定文件夹

2，flask db init #生成migrations 文件 （迁移文件）

3，flask db migrate 生成版本文件 查看是否有变化 versions 多了一个文件 （修改的地方）

4，flask db upgrade