存储PostgreSQL中的ENUM值数组

Question

存储PostgreSQL中的ENUM值数组

4

我有一个表格，其中可能会有状态：

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实现，以查看是否可以获取某种本地enum类型而无需转换，但我没有看到任何内容。我正处于无从下手的境地。

谢谢你的帮助 :)

- WilliamMayor

我建议您使用由sqlalchemy的作者编写的枚举类型。 - lrnzcig

我也遇到了这个问题，并将其提交为issue 3467到SQLAlchemy的问题跟踪器。 - Wichert Akkerman

我也遇到了这个问题。我基本上也尝试了你所做的一切，但是没有成功。 - J-bob

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- J-bob · Accepted Answer

截至1.3.17版本，不再需要任何解决方法。

以下答案作为文档被归档到ENUM数组。该文档页面现在如下所示：

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

为了历史目的，以下是旧答案：

我看了Wichert Akkerman发布的Issue 3467，并发布了这个解决方法。Mike Bayer得到了信用。在您的代码中声明以下类（当然需要必要的导入）：

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']