python 对传参进行参数检查的装饰器

from inspect import signature
 from functools import wraps
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)
 ...
 spam(1, 2, 3)
 1 2 3
 spam(1, 'hello', 3)
 1 hello 3
 spam(1, 'hello', 'world')
 Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
 File "contract.py", line 33, in wrapper
 TypeError: Argument z must be <class 'int'>
 
 
 

def decorate(func):
 # If in optimized mode, disable type checking
 if not debug:
 return func
 其次，这里还对被包装函数的参数签名进行了检查，我们使用了 inspect.signature() 函数。 简单来讲，它运行你提取一个可调用对象的参数签名信息。例如：

   from inspect import signature
 def spam(x, y, z=42):
 ...     pass
 ...
 sig = signature(spam)
 print(sig)
 (x, y, z=42)
 sig.parameters
 mappingproxy(OrderedDict([('x', <Parameter at 0x10077a050 'x'>),
 ('y', <Parameter at 0x10077a158 'y'>), ('z', <Parameter at 0x10077a1b0 'z'>)]))
 sig.parameters['z'].name
 'z'
 sig.parameters['z'].default
 42
 sig.parameters['z'].kind
 <_ParameterKind: 'POSITIONAL_OR_KEYWORD'>
 
 
 

   bound_types = sig.bind_partial(int,z=int)
 bound_types
 <inspect.BoundArguments object at 0x10069bb50>
 bound_types.arguments
 OrderedDict([('x', <class 'int'>), ('z', <class 'int'>)])
 
 
 

   bound_values = sig.bind(1, 2, 3)
 bound_values.arguments
 OrderedDict([('x', 1), ('y', 2), ('z', 3)])
 
 
 

   for name, value in bound_values.arguments.items():
 ...     if name in bound_types.arguments:
 ...         if not isinstance(value, bound_types.arguments[name]):
 ...             raise TypeError()
 ...
 
 
 

   @typeassert(int, list)
 ... def bar(x, items=None):
 ...     if items is None:
  sig = signature(spam)
 >>> print(sig)
 (x, y, z=42)
 >>> sig.parameters
 mappingproxy(OrderedDict([('x', <Parameter at 0x10077a050 'x'>),
 ('y', <Parameter at 0x10077a158 'y'>), ('z', <Parameter at 0x10077a1b0 'z'>)]))
 >>> sig.parameters['z'].name
 'z'
 >>> sig.parameters['z'].default
 42
 >>> sig.parameters['z'].kind
 <_ParameterKind: 'POSITIONAL_OR_KEYWORD'>
 >>>
 装饰器的开始部分，我们使用了 bind_partial() 方法来执行从指定类型到名称的部分绑定。 下面是例子演示：
 
 >>> bound_types = sig.bind_partial(int,z=int)
 >>> bound_types
 <inspect.BoundArguments object at 0x10069bb50>
 >>> bound_types.arguments
 OrderedDict([('x', <class 'int'>), ('z', <class 'int'>)])
 >>>
 在这个部分绑定中，你可以注意到缺失的参数被忽略了(比如并没有对y进行绑定)。 不过最重要的是创建了一个有序字典 bound_types.arguments 。 这个字典会将参数名以函数签名中相同顺序映射到指定的类型值上面去。 在我们的装饰器例子中，这个映射包含了我们要强制指定的类型断言。
 
 在装饰器创建的实际包装函数中使用到了 sig.bind() 方法。 bind() 跟 bind_partial() 类似，但是它不允许忽略任何参数。因此有了下面的结果：
 
 >>> bound_values = sig.bind(1, 2, 3)
 >>> bound_values.arguments
 OrderedDict([('x', 1), ('y', 2), ('z', 3)])
 >>>
 使用这个映射我们可以很轻松的实现我们的强制类型检查：
 
 >>> for name, value in bound_values.arguments.items():
 ...     if name in bound_types.arguments:
 ...         if not isinstance(value, bound_types.arguments[name]):
 ...             raise TypeError()
 ...
 >>>
 不过这个方案还有点小瑕疵，它对于有默认值的参数并不适用。 比如下面的代码可以正常工作，尽管items的类型是错误的：
 
 >>> @typeassert(int, list)
 ... def bar(x, items=None):
 ...     if items is None:
 ...         items = []
 ...     items.append(x)
 ...     return items
 bar(2)
 [2]
 bar(2,3)
 Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
 File "contract.py", line 33, in wrapper
 TypeError: Argument items must be <class 'list'>
 bar(4, [1, 2, 3])
 sig.parameters['z'].kind
 <_ParameterKind: 'POSITIONAL_OR_KEYWORD'>
 >>>
 装饰器的开始部分，我们使用了 bind_partial() 方法来执行从指定类型到名称的部分绑定。 下面是例子演示：
 
 >>> bound_types = sig.bind_partial(int,z=int)
 >>> bound_types
 <inspect.BoundArguments object at 0x10069bb50>
 >>> bound_types.arguments
 OrderedDict([('x', <class 'int'>), ('z', <class 'int'>)])
 >>>
 在这个部分绑定中，你可以注意到缺失的参数被忽略了(比如并没有对y进行绑定)。 不过最重要的是创建了一个有序字典 bound_types.arguments 。 这个字典会将参数名以函数签名中相同顺序映射到指定的类型值上面去。 在我们的装饰器例子中，这个映射包含了我们要强制指定的类型断言。
 
 在装饰器创建的实际包装函数中使用到了 sig.bind() 方法。 bind() 跟 bind_partial() 类似，但是它不允许忽略任何参数。因此有了下面的结果：
 
 >>> bound_values = sig.bind(1, 2, 3)
 >>> bound_values.arguments
 OrderedDict([('x', 1), ('y', 2), ('z', 3)])
 >>>
 使用这个映射我们可以很轻松的实现我们的强制类型检查：
 
 >>> for name, value in bound_values.arguments.items():
 ...     if name in bound_types.arguments:
 ...         if not isinstance(value, bound_types.arguments[name]):
 ...             raise TypeError()
 ...
 >>>
 不过这个方案还有点小瑕疵，它对于有默认值的参数并不适用。 比如下面的代码可以正常工作，尽管items的类型是错误的：
 
 >>> @typeassert(int, list)
 ... def bar(x, items=None):
 ...     if items is None:
 ...         items = []
 ...     items.append(x)
 ...     return items
 >>> bar(2)
 [2]
 >>> bar(2,3)
 Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
 File "contract.py", line 33, in wrapper
 TypeError: Argument items must be <class 'list'>
 >>> bar(4, [1, 2, 3])
 [1, 2, 3, 4]
 
 
 
 

@typeassert
 def spam(x:int, y, z:int = 42):
 print(x,y,z)
 一个可能的原因是如果使用了函数参数注解，那么就被限制了。 如果注解被用来做类型检查就不能做其他事情了。而且 @typeassert 不能再用于使用注解做其他事情的函数了。 而使用上面的装饰器参数灵活性大多了，也更加通用。