存储 ENUM 值的 PostgreSQL 数组

我有一个可以有状态的表：

statuses = ['unmoderated', 'nominee', 'finalist', 'winner']
status = db.Enum(
    *statuses, name='enum_nomination_status', metadata=db.metadata)


class Nomination(db.Model):
    status = db.Column(status, default='unmoderated')

我现在想要一个表，其中有一列可以包含多个状态：

class Judge(db.Model):
    statuses = db.Column(ARRAY(status, dimensions=1))

然而，上述方法导致我出现此错误：

ProgrammingError: (psycopg2.ProgrammingError) column "statuses" is of type enum_nomination_status[] but expression is of type text[]
LINE 1: ...4, 'Name', ARRAY['unm...
                  ^
HINT:  You will need to rewrite or cast the expression.

所以我尝试创建一个自定义类型来转换为枚举类型：

class STATUS_ARRAY(TypeDecorator):
    impl = ARRAY(status, dimensions=1)

    def process_bind_param(self, value, dialect):
        if value is None:
            return value
        else:
            return cast(array(value), ARRAY(status, dimensions=1))

但这会导致段错误。

我也尝试过铸造各个项目：

class STATUS_ARRAY(TypeDecorator):
    impl = ARRAY(status, dimensions=1)

    def process_bind_param(self, value, dialect):
        if value is None:
            return value
        else:
            return array(cast(s, status) for s in value)

但我得到：

ProgrammingError: (psycopg2.ProgrammingError) can't adapt type 'Cast' [SQL: 'INSERT INTO judge (statuses) VALUES (%(statuses)s)'] [parameters: {'statuses': [<sqlalchemy.sql.elements.Cast object at 0x7fc8bb69c710>]}]

我承认我主要是在尝试不同的铸造组合，但并不真正了解幕后发生的事情。我尝试查看底层 ENUM 实现，看看是否可以在不进行转换的情况下获得某种本机枚举类型，但我什么也看不到。我正在抓住救命稻草。

感谢您的帮助 ：）

自 1.3.17 起，不再需要解决方法

下面的答案最终出现在文档中枚举数组。该文档页面现在显示：

1.3.17 版本更改：ENUM 和 ARRAY 的组合现在由 SQLAlchemy 的实现直接处理，无需任何解决方法。

出于历史目的的旧答案：

我在看问题 3467由 Wichert Akkerman 发布，并发布了此解决方法。感谢迈克·拜尔。在代码中声明以下类（当然还有必要的导入）：

from sqlalchemy.dialects.postgresql import ARRAY
from sqlalchemy import cast

class ArrayOfEnum(ARRAY):
    def bind_expression(self, bindvalue):
        return cast(bindvalue, self)

    def result_processor(self, dialect, coltype):
        super_rp = super(ArrayOfEnum, self).result_processor(dialect, coltype)

        def handle_raw_string(value):
            if value==None:
                return []
            inner = re.match(r"^{(.*)}$", value).group(1)
            return inner.split(",")

        def process(value):
            return super_rp(handle_raw_string(value))
        return process

ArrayOfEnum现在是模型定义中使用的特殊列类型。

所以而不是

class Judge(db.Model):
    statuses = db.Column(ARRAY(status))

现在你可以这样做：

class Judge(db.Model):
    statuses = db.Column(ArrayOfEnum(status))

现在在您的代码中您可以将值分配给statuses带有列表，它将在保存时进行正确的转换：

my_judge_object.status = ['unmoderated', 'nominee']