flask-sqlalchemy 迁移数据(生成数据库表)与 查询数据

时间:2021-08-23 03:04:41

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,迁移数据,  添加

当数据库升级,如增加字段,修改字段类型等,(删除外键等)

flask-sqlalchemy 迁移数据(生成数据库表)与 查询数据

flask-sqlalchemy 迁移数据(生成数据库表)与 查询数据flask-sqlalchemy 迁移数据(生成数据库表)与 查询数据
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