在Python中實現函數重載,60%的人都不會


假設你有一個函數connect,它有一個參數address,這個參數可能是一個字符串,也可能是一個元組。例如:

connect('123.45.32.18:8080') connect(('123.45.32.18', 8080)) 復制代碼

你想在代碼里面兼容這兩種寫法,於是你可能會這樣寫代碼:

def connect(address): if isinstance(address, str): ip, port = address.split(':') elif isinstance(address, tuple): ip, port = address else: print('地址格式不正確') 復制代碼

這種寫法簡單直接,但是如果參數的類型更多,那么你就需要寫很長的 if-elif-elif-...-else。代碼看起來就非常不美觀。

學習過 Java 的同學,應該對函數重載比較熟悉,可以定義幾個名字相同的函數,但是他們的參數類型或者數量不同,從而實現不同的代碼邏輯。

在 Python 里面,參數的數量不同可以使用默認參數來解決,不需要定義多個函數。那如果參數類型不同就實現不同的邏輯,除了上面的 if-else外,我們還可以使用functools模塊里面的singledispatch裝飾器實現函數重載。

我們來寫一段代碼:

from functools import singledispatch @singledispatch def connect(address): print(f' 傳輸參數類型為:{type(address)},不是有效類型') @connect.register def _(address: str): ip, port = address.split(':') print(f'參數為字符串,IP是:{ip}, 端口是:{port}') @connect.register def _(address: tuple): ip, port = address print(f'參數為元組,IP是:{ip}, 端口是:{port}') connect('123.45.32.18:8080') connect(('123.45.32.18', 8080)) connect(123) 復制代碼
  1. 我們運行一下這段代碼,大家看看根據參數的不同,有什么樣的不同效果:

 

 可以看到,我們調用的函數,始終都是connect,但是由於傳入參數的類型不同,它運行的結果也不一樣。

我們使用singledispatch裝飾一個函數,那么這個函數就是我們將會調用的函數。

這個函數在傳入參數不同時的具體實現,通過下面注冊的函數來實現。注冊的時候使用@我們定義的函數名.register來注冊。被注冊的函數名叫什么無關緊要,所以這里我都直接使用下划線代替。

被注冊的函數的第一個參數,通過類型標注來確定它應該使用什么類型。當我們調用我們定義的函數是,如果參數類型符合某個被注冊的函數,那么就會執行這個被注冊的函數。如果參數類型不滿足任何一個被注冊的函數,那么就會執行我們的原函數。

使用類型標注來指定參數類型是從 Python 3.7才引入的新特性。在 Python 3.6或之前的版本,我們需要通過@我們定義的函數名.register(類型)來指定類型,例如:

from functools import singledispatch @singledispatch def connect(address): print(f' 傳輸參數類型為:{type(address)},不是有效類型') @connect.register(str) def _(address): ip, port = address.split(':') print(f'參數為字符串,IP是:{ip}, 端口是:{port}') @connect.register(tuple) def _(address): ip, port = address print(f'參數為元組,IP是:{ip}, 端口是:{port}') 復制代碼

同時,還有一個需要注意的點,就是只有第一個參數的不同類型會被重載。后面的參數的類型變化會被自動忽略。

最后注意:不管你是為了Python就業還是興趣愛好,記住:項目開發經驗永遠是核心,如果你沒有2020最新python入門到高級實戰視頻教程,可以去小編的Python交流.裙 :七衣衣九七七巴而五(數字的諧音)轉換下可以找到了,里面很多新python教程項目,還可以跟老司機交流討教!
本文的文字及圖片來源於網絡加上自己的想法,僅供學習、交流使用,不具有任何商業用途,版權歸原作者所有,如有問題請及時聯系我們以作處理。


免責聲明!

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



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