TypeVar 和 NewType 有什么区别?

What is the difference between TypeVar and NewType?

TypeVar and NewType 似乎相关,但我不确定何时应该使用它们,或者在运行时和静态时有什么区别。

这两个概念没有任何其他与类型相关的概念相关。

简而言之,TypeVar 是一个可以在类型签名中使用的变量,因此您可以多次引用相同的未指定类型,而 NewType 用于告诉类型检查器一些值应该被视为它们自己的类型。

Type Variables

为简化起见,类型变量让您可以多次引用同一类型,而无需指定确切的类型。

在一个定义中,单一类型的变量总是取相同的值。

# (This code will type check, but it won't run.)
from typing import TypeVar, Generic, List, Tuple

# Two type variables, named T and R
T = TypeVar('T')
R = TypeVar('R')

# Put in a list of Ts and get out one T
def get_one(x: List[T]) -> T: ...

# Put in a T and an R, get back an R and a T
def swap(x: T, y: R) -> Tuple[R, T]:
    return y, x

# A simple generic class that holds a value of type T
class ValueHolder(Generic[T]):
    def __init__(self, value: T):
        self.value = value
    def get(self) -> T:
        return self.value

x: ValueHolder[int] = ValueHolder(123)
y: ValueHolder[str] = ValueHolder('abc')

如果没有类型变量,就没有好的方法来声明 get_oneValueHolder.get.

的类型

TypeVar 上还有一些其他选项。您可以通过传入更多类型(例如 TypeVar(name, int, str))来限制可能的值,或者您可以给出一个上限,以便类型变量的每个值都必须是该类型的子类型(例如 TypeVar(name, bound=int))。

此外,您可以在声明类型变量时决定它是协变的、逆变的还是两者都不是。这实质上决定了何时可以使用 subclasses 或 superclasses 来代替泛型类型。 PEP 484 describes these concepts 更详细,并参考了其他资源。

NewType

A NewType 适用于当您想要声明一个不同的类型而不实际执行创建新类型的工作或担​​心创建新的 class 实例的开销时。

在类型检查器中,NewType('Name', int) 创建了一个 int 的子 class,名为“Name”。

在运行时,NewType('Name', int) 根本不是 class;它实际上是恒等函数,因此 x is NewType('Name', int)(x) 始终为真。

from typing import NewType

UserId = NewType('UserId', int)

def get_user(x: UserId): ...

get_user(UserId(123456)) # this is fine
get_user(123456) # that's an int, not a UserId

UserId(123456) + 123456 # fine, because UserId is a subclass of int

对于类型检查器,UserId 看起来像这样:

class UserId(int): pass

但是在运行时,UserId基本上就是这样:

def UserId(x): return x

在运行时几乎没有什么比 NewType 更重要的了。从 Python 3.8.1 开始,它的 implementation 几乎完全如下:

def NewType(name, type_):
    def identity(x):
        return x
    identity.__name__ = name
    return identity