对于我的项目,我使用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。