python进阶之装饰器之3利用装饰器强制函数上的类型检查

最新推荐文章于 2025-08-12 18:06:10 发布

weixin_30826095

最新推荐文章于 2025-08-12 18:06:10 发布

阅读量128

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：http://www.cnblogs.com/max520liuhu/p/9350516.html

本文介绍了一种使用Python装饰器实现的类型断言方法。该装饰器能够在运行时检查传入函数的参数类型是否符合预期，并在不符合时抛出TypeError异常。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 让装饰器对函数的参数进行强制执行检查
from inspect import signature
from functools import wraps

#定义装饰器@typeassert
def typeassert(*ty_args, **ty_kwargs):
    def decorate(func):
        # If in optimized mode, disable type checking
        if not __debug__:
            return func

        # Map function argument names to supplied types
        sig = signature(func)
        bound_types = sig.bind_partial(*ty_args, **ty_kwargs).arguments

        @wraps(func)
        def wrapper(*args, **kwargs):
            bound_values = sig.bind(*args, **kwargs)
            # Enforce type assertions across supplied arguments
            for name, value in bound_values.arguments.items():
                if name in bound_types:
                    if not isinstance(value, bound_types[name]):
                        raise TypeError(
                            'Argument {} must be {}'.format(name, bound_types[name])
                            )
            return func(*args, **kwargs)
        return wrapper
    return decorate

@typeassert(int, z=int)
def spam(x, y, z=42):
    print(x, y, z)

print(spam(1, 2, 3))
# 1 2 3
print(spam(1, 'hello', 3))
# 1 hello 3
print(spam(1, 'hello', 'world'))
# ypeError: Argument z must be <class 'int'>