typing
为Python的一个标注库,此默认支持PEP 484
和PEP 526
指定的类型提示。最基本的支持由Any
、Union
、Tuple
、Callable
、TypeVar
和Generic
类型组成。有关完整的规范,请参阅PEP 484
,有关任何类型提示的简单介绍,请参阅PEP 483
。
举个栗子,函数接收并返回一个字符串,如下所示:
def func(name: str) -> str:
return "Hello" + name
在函数func
中,参数预期是str
类型,并且返回str
类型
typing
模块的作用:
类型别名通过将类型分配别名来进行定义,在这个例子中,Vector
和List[str]
可以视为可互换的同义词:
from typing import List
Vector = List[str]
def func(name: str) -> Vector:
return [name]
print(Vector, type(Vector)) # typing.List[str]
value = func("laozhang")
print(value, type(value)) # ['laozhang']
使用NewType()
辅助函数来创建不同的类型
from typing import NewType
UserId = NewType("UserId", int)
UserName = NewType("UserName", str)
静态类型检查器会将新类型视为它最原始类型的子类,这对于捕捉逻辑错误非常有用:
def from_int_to_str(user_id: UserId) -> str:
return str(user_id)
print(from_int_to_str(UserId(123))) # 123
print(from_int_to_str(123)) # 123
你仍然可以对UserId
类型的变量执行所有的int
支持的操作,但结果将始终为int
类型,如下:
value = UserId(123) + UserId(456)
print(value) # 579
print(type(value)) #
值得注意的是,UserId = NewType("UserId", int)
,UserId
是一个函数,该函数将会立即返回你传递给它的任何参数。这也意味着,无法创建UserId
的子类型,因为它是运行时的标识函数,而不是实际类型,下面这种写法是错误的:
class MyUser(UserId):
pass
但是,可以基于UserId
创建NewType
,如下:
ChildUserId = NewType("ChildUserId", UserId)
期望特定签名的回调函数可以将类型标注为Callable[[Arg1Type, Arg2Type], ReturnType]
。例如:
from typing import Callable
def finder(on_success: Callable[[str, int], None]) -> None:
pass
def on_success(s: str, i: int) -> None:
pass
finder(on_success=on_success)
如果传递过去的回调函数中的参数或返回类型不匹配,PyCharm
将会有警告提示
泛型可以使用typing
模块中名为TypeVar
的新工厂进行参数化,如下:
from typing import TypeVar
T = TypeVar("T")
def finder(s: T) -> T:
return s
泛型类型可以有任意数量的类型变量,这样的话类型变量可能会收到限制:
from typing import TypeVar
T = TypeVar("T", int, str)
def finder(s: T) -> T:
return s
这样的话,finder
函数将只能接收int
/str
类型的参数,否则将会有警告提示
更多关于typing模块的使用:https://docs.python.org/zh-cn/3.7/library/typing.html