class JsonQuery(object):
def __init__(self, data):
super(JsonQuery, self).__init__()
self.data = data
def get(self, path, default=None):
if self.data is None or path == '':
return self.data
query_keys = path.split("/")
key = query_keys[0]
left_path = '/'.join(query_keys[1:])
if isinstance(self.data, list):
return [JsonQuery(v).get(path, default) if v else None for v in self.data]
else:
return JsonQuery(dict.get(self.data, key, default)).get(left_path, default)
运行效果:
d = {"main":[{"x":[{"x1":1,"x2":2},{"x1":11,"x2": 12}],"y":2},{"x":[{"x1":3,"x2":4}],"y":4}]}
>>> info = JsonQuery(d)
>>> info.get('main')
[{'y': 2, 'x': [{'x2': 2, 'x1': 1}, {'x2': 12, 'x1': 11}]}, {'y': 4, 'x': [{'x2': 4, 'x1': 3}]}]
>>> info.get('main/x')
[[{'x2': 2, 'x1': 1}, {'x2': 12, 'x1': 11}], [{'x2': 4, 'x1': 3}]]
>>> info.get('main/x/x1')
[[1, 11], [3]]
转载于:https://www.cnblogs.com/tuzkee/p/9564625.html