Python 重载|极客笔记

Python 重载

在面向对象编程中，重载是指在同一个类中定义多个同名方法，但是这些方法的参数个数或参数类型不同。Python 是一种支持函数和方法重载的语言，也可以通过使用装饰器或者默认参数值实现函数和方法的重载。

函数重载

在 Python 中，函数重载的实现需要借助装饰器或者默认参数值。下面分别介绍这两种方法。

使用装饰器实现函数重载

我们可以使用装饰器来实现函数重载。下面是一个示例代码：

def overload(*args):
    def decorator(func):
        func._overloads = args
        return func
    return decorator

@overload(int, int)
def add(x, y):
    return x + y

@overload(str, str)
def add(s1, s2):
    return s1 + s2

print(add(1, 2))
print(add("Hello, ", "world!"))

在这段代码中，我们定义了一个名为 overload 的装饰器函数，用于标记函数的重载签名。然后通过 @overload 装饰器来定义具体的重载函数。在调用 add 函数时，Python 会根据参数类型来选择合适的重载版本。

运行结果为：

3
Hello, world!

使用默认参数值实现函数重载

另一种实现函数重载的方法是使用默认参数值。下面是一个示例代码：

def add(x, y=None):
    if y is not None:
        return x + y
    else:
        return x

print(add(1, 2))
print(add("Hello, ", "world!"))

在这个示例中，我们定义了一个名为 add 的函数，其中第二个参数 y 的默认值为 None。根据 y 参数是否为 None 来决定返回值，并实现函数的重载功能。

运行结果为：

3
Hello, world!

方法重载

Python 语言并不直接支持方法重载，因为 Python 中方法的重载是基于方法名而不是参数。但是可以通过一些技巧来模拟方法重载的效果。

使用装饰器实现方法重载

和函数重载类似，可以使用装饰器来实现方法重载。下面是一个示例代码：

class Calculator:
    def add(self, x, y):
        return x + y

    def add(self, s1, s2):
        return s1 + s2

calculator = Calculator()
print(calculator.add(1, 2))
print(calculator.add("Hello, ", "world!"))

在这段代码中，我们定义了一个名为 Calculator 的类，其中包含两个具有相同方法名 add 的方法。Python 会选择最后定义的方法作为实际调用的方法。

运行结果为：

Hello, world!

使用可变参数实现方法重载

另一种模拟方法重载的方式是使用可变参数。下面是一个示例代码：

class Calculator:
    def add(self, *args):
        if len(args) == 2:
            return args[0] + args[1]
        elif len(args) == 1:
            return args[0]

calculator = Calculator()
print(calculator.add(1, 2))
print(calculator.add("Hello, ", "world!"))

在这个示例中，我们定义了一个名为 Calculator 的类，其中的 add 方法接受不定数量的参数，并根据参数个数来决定返回值，从而实现方法重载的效果。

运行结果为：