python – 存储ENUM值的PostgreSQL ARRAY

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 存储ENUM值的PostgreSQL ARRAY，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2423字，纯文字阅读大概需要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实现,看看我是否可以在没有强制转换的情况下获得某种本机枚举类型,但我看不到任何东西.我正在抓稻草.

谢谢你的帮助 ：)

解决方法:

我查看了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))

现在在您的代码中,您可以使用列表为状态指定值,并在保存时执行正确的转换：

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

内容总结

以上是互联网集市为您收集整理的python – 存储ENUM值的PostgreSQL ARRAY全部内容，希望文章能够帮你解决python – 存储ENUM值的PostgreSQL ARRAY所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端