Flask-SQLAlchemy和alembic结合（步骤）

飞熊 • 2023年3月5日上午10:27 • Python • 阅读 277

在之前的数据库操作中，我们新增一个字段是不是每次都得删除数据库表，然后再重新将新创建的数据库表映射到数据库中。这样操作是不是很蛋疼？是吧？于是sqlalchemy作者拜尔为了解决这一问题，开发了alembic这一迁移工具。

步骤一：安装alembic

pip install alembic

步骤二：配置好数据库连接文件如config.py

#配置数据库变量
HOSTNAME = '127.0.0.1'
PORT = '3306'
DATABASE = 'test1'
USERNAME = 'root'
PASSWORD = 'root'
DB_URI = 'mysql+pymysql://{}:{}@{}:{}/{}?charset=utf8mb4'.format(USERNAME,PASSWORD,HOSTNAME,PORT,DATABASE)

#把连接数据库的参数设置到app中
SQLALCHEMY_DATABASE_URI= DB_URI
SQLALCHEMY_TRACK_MODIFICATIONS = False

步骤三：注册config.py文件到Flask项目，

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

步骤四：创建模型类

class User(db.Model):
    __tablename__='t_user'
    id=db.Column(db.Integer,name='编号',primary_key=True,autoincrement=True)
    name=db.Column(type_=db.String(10),name='name')
    pwd=db.Column(name='pwd',type_=db.String(10))
    age=db.Column(name='age',type_=db.Integer)

模型类创建好了之后就需要生成数据表了

步骤五：创建一个仓库

alembic init [仓库的名字]

注意：先进入虚拟环境，然后cd到当前项目中

执行命令：

alembic init my_alembic

步骤六：修改配置文件alembic.ini和env.py

alembic.ini文件中修改这行代码：

修改完成后：

env.py文件中：
修改这行代码：

target_metadata=None

修改完后：

import os
import sys
# 把当前项目路径加入到path中
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
#app:项目主程序的名字
import app

#模型在哪？在app.py文件中
#只要是继承了db.Model的模型类都可以去找
target_metadata = app.db.Model.metadata

target_metadata = app.db.Model.metadata：代码意思
模型在哪？在app.py文件中；
只要是继承了db.Model的模型类都可以去找

步骤七：自动生成迁移文件

将当前模型中的状态生成迁移文件。

alembic revision --autogenerate -m "提示信息"

如果都是INFO,说明生成迁移文件成功

在versions文件中可以看到生成的迁移文件

步骤八：映射到数据库中

使用alembic upgrade head将刚刚生成的迁移文件，真正映射到数据库中。

alembic upgrade head

步骤九：以后如果修改了模型，重复6、7步骤

比如：模型类中新增了一个字段address

class User(db.Model):
    __tablename__='t_user'
    id=db.Column(db.Integer,name='编号',primary_key=True,autoincrement=True)
    name=db.Column(type_=db.String(10),name='name')
    pwd=db.Column(name='pwd',type_=db.String(20))
    age=db.Column(name='age',type_=db.Integer)
    address=db.Column(name='address',type_=db.String(20))

重新执行这2行代码

alembic revision --autogenerate -m "提示信息"

alembic upgrade head

文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/123396.html