枚举输入无效问题：SQLAlchemy更新数据库模型报错

核心问题 #

报错的根本原因是：你试图设置的FAILED_TO_SEND既不在数据库的recipientstatus枚举定义中，也没有在Python的RecipientStatus枚举类里声明，导致数据库无法识别这个值。

解决步骤 #

1. 更新Python枚举类 #

先在RecipientStatus里添加新的枚举成员：

class RecipientStatus(enum.Enum):
    WAITING_TO_SEND = "WAITING_TO_SEND"
    PENDING_TO_SEND = "PENDING_TO_SEND"
    FAILED_TO_SEND = "FAILED_TO_SEND"  # 新增枚举项

2. 生成Alembic迁移脚本更新数据库枚举 #

PostgreSQL不允许直接修改现有枚举类型的成员，需要通过SQL命令添加新值。创建新的迁移脚本：

def upgrade():
    # 给数据库中的recipientstatus枚举添加新值
    op.execute("ALTER TYPE recipientstatus ADD VALUE 'FAILED_TO_SEND';")

def downgrade():
    # 注意：PostgreSQL不支持直接删除枚举值，如需回滚需手动重建枚举类型
    # 操作风险较高，建议提前备份数据或忽略降级逻辑
    pass

执行迁移：alembic upgrade head

3. 修正枚举赋值代码 #

SQLAlchemy的Enum列绑定Python枚举类时，直接赋值枚举成员即可，无需使用.value：

for recipient in recipients:
    recipient.status = RecipientStatus.FAILED_TO_SEND
    recipient.status_changed_at = datetime.now(timezone.utc)
db.session.commit()

关键注意事项 #

• 必须先更新数据库枚举定义，再在代码中使用新枚举值，否则数据库会拒绝无效输入。
• 直接使用枚举成员而非字符串，能避免拼写错误，同时利用Python的类型检查提前发现问题。

内容的提问来源于stack exchange，提问作者Sammy Lam