Python中的_check_arg_types()函数：保证函数参数的正确类型

在Python中，_check_arg_types()可以作为一个装饰器函数，用于确保函数的参数的正确类型。

def _check_arg_types(*arg_types):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for i, arg in enumerate(args):
                if not isinstance(arg, arg_types[i]):
                    raise TypeError(f"Argument {i+1} should be of type {arg_types[i]}")
            for key, value in kwargs.items():
                if not isinstance(value, arg_types[len(args) + 1]):
                    raise TypeError(f"Argument {key} should be of type {arg_types[len(args) + 1]}")
            return func(*args, **kwargs)
        return wrapper
    return decorator

以上的函数实现了一个装饰器函数_check_arg_types()，它接受参数*arg_types，用于指定每个参数的正确类型。

使用例子：

@_check_arg_types(str, int)
def concatenate_and_repeat(s, n):
    return s * n

print(concatenate_and_repeat("Hello", 3)) # 输出："HelloHelloHello"
print(concatenate_and_repeat("Hello", "3")) # 抛出TypeError

@_check_arg_types(int, int)
def add(a, b):
    return a + b

print(add(2, 3)) # 输出：5
print(add(2, "3")) # 抛出TypeError

在上述例子中， 个装饰的函数concatenate_and_repeat()有两个参数， 个参数应该是一个字符串，第二个参数应该是一个整数。当传入参数的类型不符合期望时，会抛出一个TypeError异常。

第二个装饰的函数add()有两个参数，两个参数都应该是整数类型。同样，当传入的参数类型不符合期望时，会抛出一个TypeError异常。

通过使用_check_arg_types()装饰器，可以帮助我们确保函数的参数的正确类型，从而避免由于参数类型错误而引发的潜在错误。