接口自动化多层嵌套的json数据处理

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

最近在做接口自动化测试，响应的内容大多数是多层嵌套的json数据，在对响应数据进行校验的时候，可以通过（key1.key2.key3）形式获取嵌套字典值的方法获取响应值，再和预期值比较

def get_dict_value(date, keys, default=None): #default=None，在key值不存在的情况下，返回None keys_list = keys.split('.') #以“.”为间隔，将字符串分裂为多个字符串，其实字符串为字典的键，保存在列表keys_list里 if isinstance(date,dict): #如果传入的数据为字典 dictionary = dict(date) #初始化字典 for i in keys_list: #按照keys_list顺序循环键值 try: if dictionary.get(i) != None: dict_values = dictionary.get(i) #如果键对应的值不为空，返回对应的值 elif dictionary.get(i) == None: dict_values = dictionary.get(int(i)) #如果键对应的值为空，将字符串型的键转换为整数型，返回对应的值 except: return default #如果字符串型的键转换整数型错误，返回None dictionary = dict_values return dictionary else: #如果传入的数据为非字典 try: dictionary = dict(eval(date)) #如果传入的字符串数据格式为字典格式，转字典类型，不然返回None if isinstance(dictionary,dict): for i in keys_list: #按照keys_list顺序循环键值 try: if dictionary.get(i) != None: dict_values = dictionary.get(i) #如果键对应的值不为空，返回对应的值 elif dictionary.get(i) == None: dict_values = dictionary.get(int(i)) #如果键对应的值为空，将字符串型的键转换为整数型，返回对应的值 except: return default #如果字符串型的键转换整数型错误，返回None dictionary = dict_values return dictionary except: return default

运行结果：
一：字典类型数据。

二：字典类型数据，包含的键为数字。

三：json类型的字符串。

=============================================================

有个朋友分享过这段代码，大家可以试试。

class obj(object): def __init__(self, d): for a, b in d.items(): if isinstance(b, (list, tuple)): setattr(self, a, [obj(x) if isinstance(x, dict) else x for x in b]) else: setattr(self, a, obj(b) if isinstance(b, dict) else b)d = {'a':1, 'b':{'c':2}, 'd':[{'e':1}]}res = obj(d) print res.a print res.b.c print res.d[0].e