如何仅当day表为空时使用Alembic的bulk_insert函数插入初始数据

我太懂你的痛点了——每次升级都删表重建不仅麻烦，还得维护两处表结构定义，简直是给自己挖坑。其实我们可以通过先检查表是否为空，再决定是否插入初始数据的方式来解决，完全不用删表，还能复用你独立文件里的表定义，一举两得。

具体实现步骤 #

1. 复用独立表定义：直接从你的day_table.py导入表结构，后续修改表结构只需要改这一处，不用再动Alembic的脚本模板。
2. 判断表是否为空：通过SQLAlchemy执行简单的计数查询，只有当表中没有数据时才执行批量插入操作。

完整代码示例 #

首先确保你正确导入表定义：

# 替换成你项目实际的导入路径
from your_project.models.day_table import day_table
import sqlalchemy as sa

然后修改你的upgrade函数：

def upgrade():
    ${upgrades if upgrades else "pass"}
    
    # 获取当前数据库连接
    conn = op.get_bind()
    
    # 查询day表的记录总数
    record_count = conn.execute(sa.select(sa.func.count()).select_from(day_table)).scalar()
    
    # 只有表为空时才插入初始数据
    if record_count == 0:
        op.bulk_insert(day_table, [
            {'week_day': None,'label': 'day'},
            {'week_day': None,'label': 'workday'},
            {'week_day': 0,'label': 'monday'},
            {'week_day': 1,'label': 'tuesday'},
            {'week_day': 2,'label': 'wednesday'},
            {'week_day': 3,'label': 'thursday'},
            {'week_day': 4,'label': 'friday'},
            {'week_day': 5,'label': 'saturday'},
            {'week_day': 6,'label': 'sunday'},
        ])

这个方案的优势 #

• 维护成本骤降：完全复用day_table.py的表定义，后续加字段、改约束只需要改这一处，不用同步修改Alembic脚本。
• 彻底避免重复数据：不管你执行多少次upgrade，只有第一次表为空时才会插入初始数据，不会出现重复条目。
• 安全不丢数据：不用删除表，哪怕后续表中已有业务数据，也不会被误删。

内容的提问来源于stack exchange，提问作者michael holstein