Python会不会支持函数重载？龟叔仅用30行代码搞定

def f(a: int):
    print(f'a={a}')

def f(a: int, b: float):
    print(f'a={a}, b')

f(1)

TypeError: f() missing 1 required positional argument: 'b'

# 这是 mm.py 中代码
# 这是函数重载装饰器multimethod
def multimethod(*types):
    def register(f):
        f_name = f.__name__
        mm = registry.get(f_name)
        if mm is None:
            mm = registry[f_name] = MultiMethod(f_name)
        mm.register(types, f)
        return mm
    return register

# 这是 mm.py 中代码
# 模块级变量
registry = {} # 函数注册字典

class MultiMethod(object):
    def __init__(self, f_name):
        self.f_name = f_name
        self.type_dict = {}
        
    def __call__(self, *args):
        types = tuple(type(arg) for arg in args) # 生成器表达式
        print(f"函数名={self.f_name}, 参数类型={types}")
        function = self.type_dict.get(types)
        if function is None:
            raise TypeError(f"{types}不支持")
        return function(*args)
    
    def register(self, types, function):
        if types not in self.type_dict:
            self.type_dict[types] = function

@multimethod(int, int)
def foo(a, b):
   # 对整型a和b处理
   print(f"a={a}, b={b}")

@multimethod(float, float)
def foo(a, b):
    # 对浮点型a和b处理
   print(f"a={a}, b={b}")

@multimethod(str, str)
def foo(a, b):
    # 对字符串a和b处理
   print(f"a={a}, b={b}")

foo(2,1)
foo(2.0, 1.0)
foo('2.0s', '1.0s')

函数名=foo, 参数类型=(<class 'int'>, <class 'int'>)
a=2, b=1

函数名=foo, 参数类型=(<class 'float'>, <class 'float'>)
a=2.0, b=1.0

函数名=foo, 参数类型=(<class 'str'>, <class 'str'>)
a=2.0s, b=1.0s