python - 检查参数和返回类型

Question

是否可以使用 Python 3 语法来声明输入参数和返回值类型来确定这些类型？类似于确定函数的参数个数？

def foo(name: str) -> int:
    ....

我想分别得到str和int。

Answer 1

typing 模块有一个方便的功能:

>>> import typing
>>> typing.get_type_hints(foo)
{'name': <class 'str'>, 'return': <class 'int'>}

( the documentation )

这与 foo.__annotations__ 的不同之处在于 get_type_hints 可以解析存储在字符串中的前向引用和其他注释，例如

>>> def foo(name: 'foo') -> 'int':
...     ...
... 
>>> foo.__annotations__
{'name': 'foo', 'return': 'int'}
>>> typing.get_type_hints(foo)
{'name': <function foo at 0x7f4c9cacb268>, 'return': <class 'int'>}

它将在 Python 4.0 中特别有用，因为那时 all annotations will be stored in string form .