Question

我有一个带有JSON类型字段的表。 JSON字段包含带有数组的JSON对象。如何查询每行的数组长度？

class Post(Base):
    __tablename__ = 'posts'
    id = Column(Integer, primary_key=True)
    items = Column(JSON)

此表中的示例行：

id: 1, items: [2,3,5]
id: 2, items: [2,3,5, 8, 12]
id: 3, items: [2,5]
id: 4, items: []

预期结果：

有这样的事情：

session.query(Post.id, len(Post.items))

注意：JSON字段包含一个复杂对象数组，我在这里仅使用整数来说明。

Answer 1

找到解决方案

创建一个像这样扩展FunctionElement的类

from sqlalchemy.sql.expression import FunctionElement
from sqlalchemy.ext.compiler import compiles
class json_array_length(FunctionElement):
    name = 'json_array_len'

@compiles(json_array_length)
def compile(element, compiler, **kw):
    return "json_array_length(%s)" % compiler.process(element.clauses)

并像这样使用

session.query(Post.id, json_array_length(Post.items))

注意：这将适用于postgres，因为函数'json_array_length'已在其中定义。如果您希望将此功能用于需要更改功能的其他DB。请参阅http://docs.sqlalchemy.org/en/rel_0_9/core/compiler.html#further-examples

Answer 2

使用sqlalchemy.sql.expression.func对象的简单解决方案：

session.query(Post.id, func.json_array_length(Post.items))

从文档中：

func是一个特殊的对象实例，它基于基于名称的属性生成SQL函数

...

可以给func指定任何名称。如果SQLAlchemy不知道该函数名，则将完全按原样呈现。

来源：http://docs.sqlalchemy.org/en/latest/core/sqlelement.html#sqlalchemy.sql.expression.func

SqlAlchemy：查询具有数组的长度为json的字段

2 个答案: