python注解

标签：python   注解   装饰器   

http://blog.csdn.net/pipisorry/article/details/46009031

注解是 Python 3 中的一个新元素。下面类型检查机制将会以一个装饰器的形式实现。

[装饰器]

类型检查器的一个例子

def gcd(a, b):
    '''Return the greatest common divisor of a and b.'''
    a = abs(a)
    b = abs(b)
    if a < b:
        a, b = b, a
    while b != 0:
        a, b = b, a % b
    return a

>>> gcd.__annotations__
{'return': <class 'int'>, 'b': <class 'int'>, 'a': <class 'int'>}

检查返回值类型

返回值注解存储在字典“__annotations__”中的“return”键下。我们将使用这个值来检查返回值（假设注解存在）。我们将参数传递给原始函数，如果存在注解，我们将通过注解中的值来验证其类型：

检查参数类型

函数的参数存在于关联代码对象的“co_varnames”属性中，在我们的例子中是“gcd.__code__.co_varnames”。元组包含了所有局部变量的名称，并且该元组以参数开始.

参数数量存储在“co_nlocals”中。

我们需要遍历包括索引在内的所有变量，并从参数“args”中获取参数值，最后对其进行类型检查。

为了对关键字参数进行类型检查，我们需要遍历参数kwargs。此时的类型检查几乎与第一个循环中相同：

在上面的循环中，i是数组args中参数的以0起始的索引，arg是包含其值的字符串。可以利用“f.__code__.co_varnames[i]”读取到参数的名称。类型检查代码与返回值类型检查完全一样（包括错误消息的异常）。

def typecheck(f):
    def do_typecheck(name, arg):
        expected_type = f.__annotations__.get(name, None)
        if expected_type and not isinstance(arg, expected_type):
            raise RuntimeError("{} should be of type {} instead of {}".format(name, expected_type.__name__, type(arg).__name__))

    @functools.wraps(f)
    def wrapper(*args, **kwargs):
        for i, arg in enumerate(args[:f.__code__.co_nlocals]):
            do_typecheck(f.__code__.co_varnames[i], arg)
        for name, arg in kwargs.items():
            do_typecheck(name, arg)

        result = f(*args, **kwargs)

        do_typecheck(‘return‘, result)
        return result
    return wrapper

ref:How to Implement a Type Checker in Python 3

在Python 3中实现类型检查器

python注解

标签：python   注解   装饰器   

原文地址：http://blog.csdn.net/pipisorry/article/details/46009031