python - SQLAlchemy 根据 JSONB 中的嵌套键进行过滤

Question

我有一个有时有嵌套键的 JSONB 字段。例子：

{"nested_field": {"another URL": "foo", "a simple text": "text"},
 "first_metadata": "plain string",
 "another_metadata": "foobar"}

如果我做

.filter(TestMetadata.metadata_item.has_key(nested_field))

我得到这个记录。

如何搜索嵌套键的存在？( "a simple text")

Answer 1

使用 SQLAlchemy，以下内容应该适用于您的测试字符串：

class TestMetadata(Base):
    id = Column(Integer, primary_key=True)
    name = Column(String)
    metadata_item = Column(JSONB)

根据（搜索路径索引操作示例）的 SQLAlchemy 文档：JSONB

expr = TestMetadata.metadata_item[("nested_field", "a simple text")]
q = (session.query(TestMetadata.id, expr.label("deep_value"))
     .filter(expr != None)
     .all())

这应该生成SQL以下内容：

SELECT  testmetadata.id AS testmetadata_id, 
        testmetadata.metadata_item #> %(metadata_item_1)s AS deep_value
FROM    testmetadata
WHERE  (testmetadata.metadata_item #> %(metadata_item_1)s) IS NOT NULL
-- @params: {'metadata_item_1': u'{nested_field, a simple text}'}

Answer 2

在使用运算符提取嵌套 JSON 对象后，此查询使用?运算符->测试嵌套字段是否存在：

SELECT EXISTS (
   SELECT 1 
   FROM   testmetadata
   WHERE  metadata_item->'nested_field' ? 'a simple text'
   );

请注意，普通 GIN 索引不支持此查询。你需要一个表达式索引来metadata_item->'nested_field'加快速度。

CREATE INDEX testmetadata_special_idx ON testmetadata
USING gin ((metadata_item->'nested_field'));

手册中有一个类似案例的示例。