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

 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM