python函數預設引數使用避坑指南

引言

阿刁是一個自動化測試用例，從一出生他就被賦予終生使命，去測試一個叫登入的過程是否合理。他一直就被關在一個小黑屋裡面，從來也沒有出去過，小黑屋裡還被關著其他的同胞，他們身上都捆著兩個小袋子。

小黑屋裡很難受，他們都想跑出去，可怎麼也跑不出去。Python 是他們的總司令，有一次，python 告訴他們，你們就不要想著跑出去了，你們已經夠幸運了，只有 8 個人用這個屋子，別的屋子都擠著 30 多個人呢！

“這裡還有其他的屋子？” 一個用例頓時感到很欣喜。

“有，這樣的屋子這裡有 200 多個。每個屋子都有門牌號，你們這個門牌是 test_login，你們這個小鎮住的人都很危險，所以通常不允許出去逛。這是你們的小鎮地圖。‘’

“我們哪裡危險了。。。” 阿刁很不滿。

Python 微微一笑：“你別不服，看到你們身後的袋子了嗎？這裡面裝了炸彈，裡面有兩種火藥配方，一個叫 ‘’ 使用者名稱”，一個叫 “密碼”，你們每個人的配方都不一樣，因此威力也不一樣。你們的工作就是去測試 login 這個堡壘的堅固程度，這樣堡壘真正投入使用的時候，就不怕外面的攻擊了。”

“可我們每天都被關在屋子裡，哪知道怎麼去攻擊城堡。”

“這個是個好問題。你們每個人的名字下面我都掛了個錦囊，裡面有指令，按照指令做就行了。”

阿刁低頭一看，還真有指令，他大聲的唸了出來：第一條指令是叫 verify 過來幫忙，第二條指令是把炸彈裝到 verify 身上，第三條指令是檢視 verify 豎起的旗幟，看是否和自己身上的胎記一樣，如果一樣，就可以下班了；如果不一樣，那就報告 Python。

“咦，我的指令也是一樣的。我的胎記上有一行字，密碼為空。你們的呢？”

“我的也是。”

"我的是使用者不能為空。"

“我的跟你一樣啊”

阿刁低頭去看自己的胎記 “登入成功”。這給了他希望，不過他對自己的工作還有些疑惑，他得乘總司令還在的時候問問他：“老大，你的指令寫得非常明確，可我並不認識什麼 verify，萬一他不過來怎麼辦？”，其他人一聽到這個頓時都焦慮起來，是啊，萬一 verify 不來，這炸彈炸到自己了怎麼辦？

verify 炸彈

Python 指揮官給了阿刁一個讚賞的眼神，開始講：“大家不要慌，你們看到牆上的按鈕了嗎 “

所有人都紛紛望向牆面，上面有很多按鈕，其中一個印著：

from castles.login import verify

” 雖然你們通常不允許出去，但是別人是可以進出的，你們看地圖，verify 是 logi

列表是一種經常使用的資料型別。在函數的定義中，常常會使用列表作為引數。

測試介面的資料

比如，要測試一個介面的資料，介面返回的資料格式如下：

{
  "code": "20000", 
  "data": ["孫悟空","李白","甄姬"], 
  "msg": "success", 
  "status": 0
}

要測試的內容是：返回的 data 資料是否跟需求符合。在測試之前，需要對資料進一步處理，比如要增加 "王昭君" 這個元素進去，需要寫一個函數：

def add_element(data=["孫悟空","李白","甄姬"]):
    data.append('王昭君')
    return data

print(add_element())
print(add_element())
print(add_element())

在函數定義的時候經常會給引數設定預設值，在這個例子中，將 data 引數設定了預設值，函數定義以後，後面會被頻繁的呼叫，期望值應該是列印如下：

["孫悟空","李白","甄姬","王昭君"]
["孫悟空","李白","甄姬","王昭君"]
["孫悟空","李白","甄姬","王昭君"]

實際結果為：

["孫悟空","李白","甄姬","王昭君"]
["孫悟空","李白","甄姬","王昭君","王昭君"]
["孫悟空","李白","甄姬","王昭君","王昭君","王昭君"]

原因

當定義函數時，會儲存函數中預設引數 data 的值，也就是 ["孫悟空","李白","甄姬"],在每次呼叫的時候如果傳遞了新的實參，則使用傳遞的引數；沒有傳遞，使用定義函數時儲存的預設引數。

上面兩次呼叫中，都沒有傳遞新的實參，程式會呼叫定義函數時儲存的預設引數，因為 append() , 在第一次呼叫以後，預設引數已經由 ["孫悟空","李白","甄姬"] 改變為 ["孫悟空","李白","甄姬","王昭君"],再次執行 append() 之後，就變成了 ["孫悟空","李白","甄姬","王昭君","王昭君"]；同理，第三次又改變了。

可以使用 id() 函數來定位問題：

def add_element(data=["孫悟空","李白","甄姬"]):
    # id 來表示是不是同一個物件
  print(id(data))
    data.append('王昭君')
    return data

print(add_element())
print(add_element())
print(add_element())

列印出來的 id(data) 為同一個物件，也就是預設引數。如果我們改變 第二個 print(add_element())為 print(add_element(["孫悟空","李白","甄姬"]))，那麼第 2 個 id(data) 就會發生變化，因為它不在是預設值，而是新傳進來的實參，實際結果也將變成：

2543416926792
['孫悟空', '李白', '甄姬', '王昭君']
2543418907848
["孫悟空","李白","甄姬", '王昭君']
2543416926792
['孫悟空', '李白', '甄姬', '王昭君', '王昭君']

改進方案

• 如果引數中有列表，儘量不要用它做預設引數
• 如果使用了列表作為預設引數，函數呼叫時傳入實參，而不是省略
• 可以在函數體中另外定義一個變數接收預設引數

def add_element(data=["孫悟空","李白","甄姬"]):
    if data == ["孫悟空","李白","甄姬"]:
        data = ["孫悟空","李白","甄姬"]
    data.append('王昭君')
    return data

以上就是python函數預設引數使用避坑指南的詳細內容，更多關於python函數預設引數的資料請關注it145.com其它相關文章！