在Python中实现函数重载的示例代码
假设你有一个函数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)
我们运行一下这段代码，大家看看根据参数的不同，有什么样的不同效果：
可以看到，我们调用的函数，始终都是 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}')
同时，还有一个需要注意的点，就是只有第一个参数的不同类型会被重载。后面的参数的类型变化会被自动忽略。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Matplotlib实现各种条形图绘制，文章通过利用 plt.bar 方法实现各种条形图绘制，内容详细具有一定的参考价值，需要的小伙伴可以参考一下2022-03-03
今天小编就为大家分享一篇关于Python魔法方法详解，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2019-02-02
这篇文章主要介绍了Python中import机制的相关资料，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下2020-09-09
下面小编就为大家分享一篇Python向MySQL批量插数据的实例讲解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-03-03
这篇文章主要介绍了Python3实现的回文数判断及罗马数字转整数算法,涉及Python数值运算、转换等相关操作技巧,需要的朋友可以参考下2019-03-03
logging是python语言中的一个日志模块，专门用来写日志的，日志级别通常分为debug、info、warning、error、critical几个级别，一般情况下，默认的日志级别为warning，在调试或者测试阶段，下面就快速体验一下logging模块写日志的用法，感兴趣的朋友跟随小编一起看看吧2022-12-12
这篇文章主要介绍了python判断字符串的前两个字母是否是”id",使用 Python 的字符串切片来判断一个字符串的前两个字母是否是 "id"，本文结合示例代码给大家介绍的非常详细，需要的朋友可以参考下2023-04-04
这篇文章主要为大家详细介绍了python实现windows壁纸定期更换功能，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-01-01
这篇文章主要介绍了python中常见进制之间的转换方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
Python3连接MySQL（pymysql）模拟转账实现代码
这篇文章主要介绍了Python3连接MySQL（pymysql）模拟转账实现代码，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2016-05-05
最新评论