python xpath方法解析json

现在移动互联网飞速发展，在测试过程中通常需要进行HTTP接口测试，而大多数http请求返回的是json数据。

然而python对josn数据的解析，就是对字典的解析，以下面这段json为例：

{
"ver": "6.8",
"dcid": "477",
"head": {
"cid": "",
"ctok": "",
"cver": "1.0",
"lang": "01",
"sid": "1",
"syscode": "09",
"auth": "",
"extension": []
},
"contentType": "json"
}

如果我们要获取sid中的数据，需要这样解析

第一种：jsonstr['head']['sid']

第二种：jsonstr.get('head').get('sid')

写完后发现这种方法是不是太low，而且当json层级太深时，代码可视性极差，而且难于维护。

json属于结构化数据，就像html一样，于是考虑到用xpath方法去解析。下面直接上代码：

import json
import xmltodict

# 解析json字符串
class jsonprase(object):
def __init__(self, json_value):
try:
eval(json_value)
self.json_value = json.loads(json_value)
except Exception, e :
raise ValueError('must be a json str value')

def find_json_node_by_xpath(self, xpath):
elem = self.json_value
nodes = xpath.strip("/").split("/")
for x in range(len(nodes)):
try:
elem = elem.get(nodes[x])
except AttributeError:
elem = [y.get(nodes[x]) for y in elem]
return elem

def datalength(self, xpath="/"):
return len(self.find_json_node_by_xpath(xpath))

@property
def json_to_xml(self):
try:
root = {"root": self.json_value}
xml = xmltodict.unparse(root, pretty=True)
except ArithmeticError, e:
pyapilog().error(e)
return xml

然后测试一把

a ='''{
"ver": "6.8",
"dcid": "477",
"head": {
"cid": "",
"ctok": "",
"cver": "1.0",
"lang": "01",
"sid": "1",
"syscode": "09",
"auth": "",
"extension": []
},
"contentType": "json"
}'''
print type(a)
d = jsonprase(a).find_json_node_by_xpath('/head/sid')
print d