Python字典嵌套(源碼分析/自定義)

今天在寫監控腳本的時候遇到一個問題,就是我執行每一個監控模塊(腳本)的時候,例如CPU、內存、磁盤腳本,都會返回一個字典格式的數據,但是我需要將這三個字典,組合成一個大字典,然後通過requests模塊發送給api接口,so,我就在網上找了一些方法,然後總結,寫成這編博文。

1、首先定義三個字典(不需要考慮字典的具體內容)

>>> cpu_dict = {'cpu_count':8,'cpu_ratio':3.5}
>>> memory_dict = {'memory_count':16,'memory_ration':10}
>>> disk_dict = {'disk_read':200,'disk_write':120,'tps':340}

2、將上面三個字典合併成一個字典

>>> data_dict = dict(**cpu_dict,**memory_dict,**disk_dict)
>>> data_dict
{'cpu_count': 8, 'cpu_ratio': 3.5, 'memory_count': 16, 'memory_ration': 10, 'disk_read': 200, 'disk_write': 120, 'tps': 340}

注意:需要被合併的字典的key是不能出現重複的,否則python會直接報錯

3、從源碼分析dict這個類是如何實現字典拼接

def __init__(self, seq=None, **kwargs):
    """
    忽略官方註釋..
    """
    pass

首先我們調用 dict() 這個類在括號裏傳入值,會執行init構造方法
第一位形參:self 是對象即文中的data_dict,
第二位形參:seq 排序作用(當前文中我們忽略它)
第三位形參:**kwargs 接收所有以字典形式保存的數據

那麼我們在執行下面這條命令的時候發生了什麼?

data_dict = dict(**cpu_dict,**memory_dict,**disk_dict)

首先cpu_dict、memory_dict、disk_dict 本身就是字典,在它們前面加個了 * 後就是把這些字典傳給 dict類裏的__init__構造函數的第三個位置形參來 *kwargs 來接收字典,最後在返回結果給 data_dict 這個對象

4、kwargs實現方法**

def func(**kwargs):           # 位置形參
    print(kwargs)

func(**{'k1':'v1','k2':'v2'})  # 位置實參

# 執行結果:
{'k1': 'v1', 'k2': 'v2'}

現在我們知道了,在位置形參裏的 **kwargs 是可以接收所有字典形式的數據,那如果我把位置實參裏傳一個字典形式存儲的變量,會如何?

t1 = {'k1':'v1','k2':'v2'}
def func(**kwargs):
    print(kwargs)

func(**t1)

# 執行結果
{'k1': 'v1', 'k2': 'v2'}

so,我們可以看到直接執行func函數,通過*t1 可以直接將存儲字典的變量傳入給func函數,當然在func函數裏也可以傳入多個字典,我再次聲明下再位置形參裏的*kwargs,是可以接收N個從位置實參傳入的字典數據
那麼,有朋友會問,位置形參裏的 *kwargs 必須是固定這麼寫嗎?,不然,* 這個關鍵字是必須的,但是* 後面的變量名可以隨意,我們也可以寫成 *abc,但是大部分人都是按照開發標準統一寫成 **kwargs,因爲python源碼也是這麼寫的,我們就遵循規範就行

4、利用神奇的雙星號 來自定義類似dict的合併字典功能**

cpu_dict = {'cpu_count':8,'cpu_ratio':3.5}
memory_dict = {'memory_count':16,'memory_ration':10}
disk_dict = {'disk_read':200,'disk_write':120,'tps':340}

class task():
    def __init__(self,*args,**kwargs):
        self.kwargs = kwargs

    def __call__(self, *args, **kwargs):
        print(kwargs)

my_dict = task()
my_dict(**cpu_dict,**memory_dict,**disk_dict)

最終執行的結果:

{'cpu_count': 8, 'cpu_ratio': 3.5, 'memory_count': 16, 'memory_ration': 10, 'disk_read': 200, 'disk_write': 120, 'tps': 340}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章