使用SQLAlchemy&Alembic创建PostgreSQL ENUM类型后读写请求异常

核心问题点 #

1. 字段名不匹配
   错误日志里的INSERT语句使用了user_name和date_created字段，但你通过Alembic创建的表实际字段是username和timestamp_created，数据库中不存在请求里的字段，直接导致插入失败。
2. 主键重复定义
   你同时给id和uuid设置了primary_key=True，PostgreSQL不允许单表存在多个独立主键（复合主键是多字段组合，并非两个独立主键），这会导致表结构异常，连GET请求都无法正常执行。
3. ENUM重复实例化
   代码中先定义了gender_types枚举对象，但在gender字段中又重新实例化了Enum(...)，即便名称一致，SQLAlchemy也可能将其识别为新类型，引发映射冲突。

修复后的Alembic迁移代码 #

def upgrade() -> None:
    # 定义枚举类型并复用，避免重复实例化
    gender_types = ENUM('male', 'female', 'non_binary', name='gender_types', create_type=True)
    op.create_table(
        'users',
        Column('id', Integer, primary_key=True, autoincrement=True),
        # 将uuid设为唯一约束而非主键，避免主键冲突
        Column('uuid', UUID(as_uuid=True), default=uuid.uuid4, autoincrement=False, unique=True, nullable=False),
        Column('username', String(30), nullable=False, unique=True),
        Column('first_name', String(30), nullable=False),
        Column('last_name', String(30), nullable=False),
        Column('middle_name', String(30), nullable=True),
        Column('timestamp_created', TIMESTAMP(timezone=False), default=func.now()),
        # 复用预先定义的枚举对象
        Column('gender', gender_types, nullable=True)
    )

后续操作步骤 #

• 修正请求字段：POST请求中的字段需与表结构一致，将user_name改为username，date_created改为timestamp_created。
• 回滚并重新迁移：如果已执行过错误的迁移脚本，先执行alembic downgrade -1回滚，替换为修复后的脚本后再执行alembic upgrade head重新创建表。
• 验证表结构：在psql中执行\d users，确认字段名、主键约束、枚举类型均正确配置。

内容的提问来源于stack exchange，提问作者NITHINN MANICKAM G