python - 填充枚举数组

Question

我无法从我的 alembic 迁移中将枚举放入我的表中。

此处为 MVCE:https://pastebin.com/ng3XcKLf

(SQLAlchemy 1.2、psycopg2 2.7.3.2 和 postgres 10.1 - 第 15 行需要使用您的 postgres URI 进行修改)

我阅读了有关 SQLAlchemy/Postgres 和枚举数组的问题，但根据我在问题跟踪器中找到的内容，该问题已在 1.1 中得到解决。

有人能指出我正确的方向吗？

变体 1: 尝试使用 postgres 枚举类型的属性

op.bulk_insert(permission, [{
    'name': 'ViewContent',
    'contexts': [pgpermissioncontexts.resourceType]
}])

这失败了:AttributeError: 'Enum' object has no attribute 'resourceType'

变体 2: 尝试使用底层 python 枚举的属性

op.bulk_insert(permission, [{
    'name': 'ViewContent',
    'contexts': [PermissionContexts.resourceType]
}])

这失败了 sqlalchemy.exc.ProgrammingError: (psycopg2.ProgrammingError) cols of type permissioncontexts[] but expression is of type text[]

变体 3: 将字符串数组转换为枚举数组

op.bulk_insert(permission, [{
    'name': 'ViewContent',
    'contexts': sa.cast([PermissionContexts.resourceType], sa.ARRAY(pgenum))
}])

这可能有效也可能无效 - python 进程膨胀到使用 4GB 内存，并一直坐在那里直到终止。

变体 4: 插入空数组

op.bulk_insert(permission, [{
    'name': 'ViewContent',
    'contexts': []
}])

这有效，但显然没有任何值(value)。

Answer 1

不幸的是，枚举数组不能开箱即用，但有一个 documented workaround ,这也在 this answer 中进行了描述.使用 ArrayOfEnum 而不是 ARRAY，您的变体 2 有效:

op.bulk_insert(permission, [{
    'name': 'ViewContent',
    'contexts': [PermissionContexts.resourceType],
}])

转换为 ARRAY(permissioncontexts) 也应该有效，并且在不使用 bulk_insert() 时也有效

op.execute(permission.insert().values({
    'name': 'ViewContent',
    'contexts': sa.cast([PermissionContexts.resourceType], ARRAY(pgpermissioncontexts)),
}))

或者当使用 bulk_insert(multiinsert=False) 时:

op.bulk_insert(permission, [{
    'name': 'ViewContent',
    'contexts': sa.cast([PermissionContexts.resourceType], ARRAY(pgpermissioncontexts)),
}], multiinsert=False)

似乎 alembic 或 sqlalchemy 对多参数的处理存在错误。