在接口测试或其他测试场景中，我们经常需要获取JSON中的某路径下的值进行校验，即从JSON中抽取指定信息。

本文主要介绍满足该需求的Python jmespath库，除此之外还有Jsonpath，有兴趣的可以自行了解。

安装

pip install jmespath

代码示例

我们以下面test_json 这个较为复杂的Json对象为示例。

test_json = {
  "code": 200,
  "desc": {
    "info":"说明",
    "update":"2021年3月6日"
   },
  "data": [
     {
      "apps": {
        "app_op_seq": [
           {
            "action": "点击",
            "module_name": "聚划算",
            "module_type": "resource"
           }
         ]
       },
      "content": {
        "des": {
          "company_name": "耐克",
          "intent": [
            "full"
           ]
         },
        "rel": [
           {
            "des": {
              "person_name": "欧阳玖林",
              "political_status": "金牌会员"
             },
            "ont": [
               {
                "name":"Person",
                "id":1
               },
               {
                "name":"Company",
                "id":2
               },
               {
                "name":"Car",
                "id":3
               }
             ],
            "relIdx": [
              0,
              "8-9"
             ],
            "relName": "欧阳",
            "segs": [
              "耐克篮球鞋"
             ]
           }
         ],
        "segs": [
          "耐克篮球鞋"
         ]
       },
      "content_op": "查询"
     }
   ]
}

我们先用之前文章中介绍的自研 《Python 实现 JSON、字典数据结构的递归解析》 代码，解析上述 JSON对象，解析结果如下：

1 JsonPath:code  Value:200 
2 JsonPath:desc.info  Value:说明 
3 JsonPath:desc.update  Value:2021年3月6日 
4 JsonPath:data[0].apps.app_op_seq[0].action  Value:点击 
5 JsonPath:data[0].apps.app_op_seq[0].module_name  Value:聚划算 
6 JsonPath:data[0].apps.app_op_seq[0].module_type  Value:resource 
7 JsonPath:data[0].content.des.company_name  Value:耐克 
8 JsonPath:data[0].content.des.intent[0]  Value:full 
9 JsonPath:data[0].content.rel[0].des.person_name  Value:欧阳玖林 
10 JsonPath:data[0].content.rel[0].des.political_status  Value:金牌会员 
11 JsonPath:data[0].content.rel[0].ont[0].name  Value:Person 
12 JsonPath:data[0].content.rel[0].ont[0].id  Value:1 
13 JsonPath:data[0].content.rel[0].ont[1].name  Value:Company 
14 JsonPath:data[0].content.rel[0].ont[1].id  Value:2 
15 JsonPath:data[0].content.rel[0].ont[2].name  Value:Car 
16 JsonPath:data[0].content.rel[0].ont[2].id  Value:3 
17 JsonPath:data[0].content.rel[0].relIdx[0]  Value:0 
18 JsonPath:data[0].content.rel[0].relIdx[1]  Value:8-9 
19 JsonPath:data[0].content.rel[0].relName  Value:欧阳 
20 JsonPath:data[0].content.rel[0].segs[0]  Value:耐克篮球鞋 
21 JsonPath:data[0].content.segs[0]  Value:耐克篮球鞋 
22 JsonPath:data[0].content_op  Value:查询 

Process finished with exit code 0

Demo1 查询某个key的值

import jmespath

result = jmespath.search("code",test_json)
print(result)

# 执行上述代码，输出结果如下:
200

Demo2 层级查询某个key的值

import jmespath

result = jmespath.search("desc.info",test_json)
print(result)

# 执行上述代码，输出结果如下:
说明

Demo3 通过索引查询Json中List 元素

import jmespath

result = jmespath.search("data[0]",test_json)
print(result)

# 执行上述代码，输出结果如下:
{'apps': {'app_op_seq': [{'action': '点击', 'module_name': '聚划算', 'module_type': 'resource'}]}, 'content': {'des': {'company_name': '耐克', 'intent': ['full']}, 'rel': [{'des': {'person_name': '欧阳玖林', 'political_status': '金牌会员'}, 'ont': [{'name': 'Person', 'id': 1}, {'name': 'Company', 'id': 2}, {'name': 'Car', 'id': 3}], 'relIdx': [0, '8-9'], 'relName': '欧阳', 'segs': ['耐克篮球鞋']}], 'segs': ['耐克篮球鞋']}, 'content_op': '查询'}

Demo4 复杂层级查询某个key的值

import jmespath

result = jmespath.search("data[0].apps.app_op_seq[0].action",test_json)
print(result)

# 执行上述代码，输出结果如下:
点击

通过上述查询方式，我们基本也能看出，在取列表值时用的 “[]”，取字典值时用的 “.”，来表示路径层级 。

Demo5 对Json中List进行切片查询

import jmespath
# 获取relIdx下第0、1个元素 
result = jmespath.search("data[0].content.rel[0].relIdx[0:2]",test_json)
print(result)

# 获取relIdx下全部元素 relIdx[*]
result1 = jmespath.search("data[0].content.rel[0].relIdx[*]",test_json)
print(result1)

# 执行上述代码，输出结果如下:
[0, '8-9']
[0, '8-9']

Demo6 其他用法

import jmespath
# 批量获取Json中List的字典元素的某个Key的值
result = jmespath.search("data[0].content.rel[0].ont[0:3].name",test_json)
print(result)

# 批量获取Json中List的字典元素的多个Key的值
result1 = jmespath.search("data[0].content.rel[0].ont[0:3].[name,id]",test_json)
print(result2)


# 执行上述代码，输出结果如下:
['Person', 'Company', 'Car']
[['Person', 1], ['Company', 2], ['Car', 3]]