对枚举列应用array_agg将返回" " { " is not among the defined enum values"



对于我的项目,我使用Postgres DB和SQLAlchemy作为ORM。我已经定义了几个我们在不同的模型中使用的枚举。

我定义了一个枚举

class TestEnum(MultiValuedEnum):
STATE1 = 'state-1', (
'State 1', 'Waiting state 2')
STATE2 = 'state-2', (
'State 2', 'In state 2')
STATE3 = 'state-3', (
'State 3', 'In state 3')
@property
def display_name(self):
return self.metadata[0]
@property
def description(self):
return self.metadata[1]

此枚举用于类

class ExampleClass(db.Model):
id = db.Column(db.BigInteger, primary_key=True, nullable=False)
product_id = db.Column(db.BigInteger, nullable=False)
state = db.Column(db.Enum(constants.TestEnum), nullable=False)

现在,如果我编写一个查询

query = session.query(
ExampleClass.product_code,
func.array_agg(ExampleClass.state)
).group_by(ExampleClass.product_code).all()

我收到以下错误

"{"不在定义的枚举值中

如果我在 dbeaver 中运行生成的查询(使用sqltap捕获(,我会得到正确的结果,它看起来像这样

+---+------------+----------------+
|   |product_code|      State     |
+---+------------+----------------+
| 1 | A          |{STATE1,STATE2} |
| 2 | B          |{STATE2,STATE3} |
| 3 | C          |{STATE1,STATE3} |
+---+------------+----------------+

似乎,SQL Alchemy 正在尝试将状态列表解析为State对象,但由于开始时{将其表示为我想的列表而失败。如果它不将其转换为状态列表,为什么我会收到此错误以及修复或解决方法。

array_agg似乎在项目中的其他字段上运行良好,并且我能够遍历返回的数组,但对于 ENUM,它似乎失败了。 谢谢

所以,在尝试不同的事情时,我尝试了json_agg,它似乎有效。所以现在我的查询看起来像

query = session.query(
ExampleClass.product_code,
func.json_agg(ExampleClass.state)
).group_by(ExampleClass.product_code).all()

它似乎以列表的形式返回给我 ENUMS。

相关内容

  • 没有找到相关文章

最新更新