flask数据库迁移保留数据库无关的表

我们开发过程中,每当使用flask-migrate 进行数据迁移时,总会将数据库原有的表全部删除。那是因为在 alembic的设定中:如果目标数据库有许多不属于元数据的表,自动生成过程通常会假定这些表是要删除的数据库中的无关表,并且它将为每个表执行 Operations.drop _ table ()操作。

为了防止这种情况,可以使用 EnvironmentContext.configure.include _ name 钩子进行过滤


```python
# ./migrations/env.py

from flask import current_app
'''
target_metadata 获取当前项目中所有的表数据
在 MetaData 对象的表集合中搜索每个名称,并确保不包含不存在的名称
'''
target_metadata = current_app.extensions['migrate'].db.metadata

def include_name(name, type_, parent_names):
    if type_ == "table":
        return name in target_metadata.tables
    else:
        return True

#默认会执行 run_migrations_online()函数,在context.configure配置以下几项
context.configure(
    # ...
    target_metadata = target_metadata,
    include_name = include_name,
    include_schemas = False
)

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐