python 3.5类型提示:我可以检查函数参数是否匹配类型提示?

时间:2015-09-29 12:53:37

标签: python python-internals python-3.5

python 3.5是否提供允许测试是否给定的函数 参数是否适合函数声明中给出的类型提示?

如果我有例如这个功能:

def f(name: List[str]):
    pass

是否有一个python方法可以检查是否

name = ['a', 'b']
name = [0, 1]
name = []
name = None
...

适合类型提示吗?

我知道'在运行时没有进行类型检查'但是我仍然可以检查 这些参数在python中的有效性?

或者如果python本身不提供该功能:我的工具是什么 需要用吗?

2 个答案:

答案 0 :(得分:10)

Python本身不提供此类功能,您可以阅读更多相关信息here

我为此写了一个装饰师。这是我装饰的代码:

from typing import get_type_hints

def strict_types(function):
    def type_checker(*args, **kwargs):
        hints = get_type_hints(function)

        all_args = kwargs.copy()
        all_args.update(dict(zip(function.__code__.co_varnames, args)))

        for argument, argument_type in ((i, type(j)) for i, j in all_args.items()):
            if argument in hints:
                if not issubclass(argument_type, hints[argument]):
                    raise TypeError('Type of {} is {} and not {}'.format(argument, argument_type, hints[argument]))

        result = function(*args, **kwargs)

        if 'return' in hints:
            if type(result) != hints['return']:
                raise TypeError('Type of result is {} and not {}'.format(type(result), hints['return']))

        return result

    return type_checker

你可以这样使用它:

@strict_types
def repeat_str(mystr: str, times: int):
    return mystr * times

虽然限制你的函数只接受一种类型并不是非常pythonic。虽然您可以使用abc(抽象基类),如number(或自定义abc)作为类型提示,并限制您的函数不仅接受一种类型,而且接受您想要的任何类型组合。

如果有人想要使用它的话,为它添加了一个github repo

答案 1 :(得分:5)

这是一个老问题,但我已经编写了一个基于类型提示进行运行时类型检查的工具:https://pypi.org/project/typeguard/