最近在做接口自動化測試,響應的內容大多數是多層嵌套的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