我有一个表格,其中包含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: []
预期结果:
1 3
2 5
3 2
4 0
有这样的事情:
session.query(Post.id, len(Post.items))
注意:JSON字段包含复杂对象的数组,我在这里仅使用整数来说明。
找到了解决方案
创建一个像这样扩展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
一个更简单的解决方案,使用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