本站资源收集于互联网,不提供软件存储服务,每天免费更新优质的软件以及学习资源!

Python中的重载函数

网络教程 app 1℃

Python中的重载函数

函数重载是指定义多个具有相同名称但不同签名的函数的能力,这意味着它们具有不同数量或类型的参数。编译器或解释器会根据函数调用期间传递的参数数量和类型自动选择函数的正确版本。

java 和 c++++ 等语言本身就支持此功能。

虽然 python 本身不支持函数重载,因为它是一种动态类型语言,但可以使用各种模块和实用程序来实现相同的功能。

这是我的重载实现。

执行

from __future__ import annotationsimport inspectimport typingbin: dict[str, overloadnamespace] = {}class overloadnamespace: overloads: dict[tuple[type, …], typing.callable[…, typing.any]] fallback: typing.callable[…, typing.any] def __init__(self, name: str) -> none: self.overloads = {} self.fallback = self._fallback bin[name] = self def __call__(self, *args: typing.any, **kwds: typing.any) -> typing.any: types = [type(arg) for arg in args] types.extend([type(kwrg) for kwrg in kwds]) try:return self.overloads[tuple(types)](*args, **kwds) except keyerror:return self.fallback(*args, **kwds) @staticmethod def _fallback(*_, **__) -> none: raise notimplementederror

overloadnamespace 类是一个可调用类,充当函数名称和调用签名之间的媒介。参数被传递到 __call__ dunder 方法中,该方法将提供的数据类型与存储在重载字典中的类型元组进行匹配。返回匹配的签名,并使用提供的 args/kwargs 进行调用。如果没有找到匹配的签名,则调用后备函数。

使用 overloadnamespace 类

此类不适合手动使用,它由装饰器使用,修饰器修改函数并使用与函数提供的名称相同的名称返回 overloadnamespace 类的实例。

def overload(*args) -> typing.callable[…, overloadnamespace] | overloadnamespace: """decorator used to create overloads of functions with same name. returns a [overloadnamespace]""" if len(args) == 1 and inspect.isfunction(args[0]): return overload_using_types(args[0]) def inner(func: typing.callable[…, typing.any]) -> overloadnamespace: sig = inspect.signature(func) assert len(args) == len(sig.parameters ), "number of types and args in function is not same." namespace = (bin[func.__name__]if bin.get(func.__name__)else overloadnamespace(func.__name__) ) namespace.overloads[tuple(args)] = func return namespace return innerdef overload_using_types(func: typing.callable[…, typing.any]) -> overloadnamespace: args = inspect.signature(func).parameters types = tuple(arg.annotation for arg in args.values()) namespace = ( bin[func.__name__] if bin.get(func.__name__) else overloadnamespace(func.__name__) ) namespace.overloads[types] = func return namespace

重载装饰器使用装饰器值或类型提示检查参数类型并返回命名空间类。

使用示例

# types in decorator@overload(int, int)def sum(a, b): return a+b# or as typehints@overload def sum(a: float, b: float): return int(a+b)+1sum(1,2) # 3sum(1.23, 2.0) # 4

这只是一个基本想法,适用于非联合固定类型。

倒退

后备函数用作当没有参数模式与调用模式匹配时要调用的函数。

def fallback( func: typing.Callable[…, typing.Any],) -> OverloadNamespace: """Fallback function to be called if no overloads match to the provided arguments.""" namespace = ( bin[func.__name__] if bin.get(func.__name__) else OverloadNamespace(func.__name__) ) namespace.fallback = func return namespace@fallback def sum(*args): return sum(args)sum(1,2,3,4) # 10

以上就是Python 中的重载函数的详细内容,更多请关注范的资源库其它相关文章!

转载请注明:范的资源库 » Python中的重载函数

喜欢 (0)