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]]