python 分享一個經過 (key1.key2.key3) 形式獲取嵌套字典值的方法

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