如何在Python中手动实现整数加法场景的函数重载

原有代码的核心问题 #

1. Python没有原生函数重载支持，同作用域下的同名函数会被最后定义的版本完全覆盖。你依次定义了接收2、3、4个参数的add函数，最终生效的只有要求4个入参的版本，调用时传入2个参数必然会报参数数量不匹配错误。
2. 内置sum()方法使用错误：该方法要求第一个入参为可迭代对象（如列表、元组），不支持直接传入多个独立数值。
3. 输入处理逻辑存在bug：int(input("Enter two numbers: ").split())中split()返回的是字符串列表，无法直接用int()转换，只有4个参数的输入处理写法是正确的。

方案1：简化实现（可变参数适配任意数量整数相加） #

无需复杂的重载逻辑，用Python可变参数即可天然支持任意数量的整数加法：

def add(*args):
    # 校验所有参数为整数
    for arg in args:
        if not isinstance(arg, int):
            raise TypeError("所有入参必须为整数类型")
    return sum(args)

# 修正所有输入处理逻辑
a,b = [int(x) for x in input("Enter two numbers: ").split()]
print("The sum is: ", add(a,b))

c,d,e = [int(x) for x in input("Enter three numbers: ").split()]
print("The sum is: ", add(c,d,e))

f,g,h,i  = [int(x) for x in input("Enter four numbers: ").split()]
print("The sum of is: ", add(f,g,h,i))

方案2：手动实现严格的按参数数量重载 #

如果需要严格实现不同参数数量对应独立函数的重载特性，可以通过装饰器实现分发逻辑：

# 重载分发装饰器
def overload(func):
    registry = {}
    def register(arg_count, target_func):
        registry[arg_count] = target_func
    def dispatcher(*args):
        arg_count = len(args)
        if arg_count not in registry:
            raise TypeError(f"不存在接收{arg_count}个参数的add实现")
        return registry[arg_count](*args)
    register(func.__code__.co_argcount, func)
    dispatcher.register = register
    return dispatcher

@overload
def add(a, b):
    return a + b

@add.register(3)
def add(c, d, e):
    return c + d + e

@add.register(4)
def add(f, g, h, i):
    return f + g + h + i

# 修正所有输入处理逻辑
a,b = [int(x) for x in input("Enter two numbers: ").split()]
print("The sum is: ", add(a,b))

c,d,e = [int(x) for x in input("Enter three numbers: ").split()]
print("The sum is: ", add(c,d,e))

f,g,h,i  = [int(x) for x in input("Enter four numbers: ").split()]
print("The sum of is: ", add(f,g,h,i))

该实现会将不同参数数量的add函数存入注册表，调用时根据传入的参数个数自动匹配对应的函数执行，完全符合函数重载的特性要求。

内容的提问来源于stack exchange，提问作者Zhen Zhen