Python 类型注解中泛型应用

介绍

 Python解释器不强制执行函数和变量类型注解。但它们可以被第三方工具使用，例如类型检查器，IDE等。使用类型注解，IDE可以准确地推断出当前对象的具体类型，能更好地提示该对象的属性、方法，以及对于方法的传入参数进行检查，在能够保证代码逻辑正确的情况下，如果检查通过，甚至可以不调试，直接运行程序即可正常工作，对于稍复杂的项目，可以有效地提高开发效率。

就本人而言，在自己所负责的项目中大量使用类型注解功能，深刻地体会到了类型注解的好处，但之前的使用过程中，对于某些函数，如下面的示例代码，传入两个参数a、b对a、b求和后返回，但Python3上不同类型的变量是不能做+运算的，而且如果参数传入类型是str，那么返回类型也必须是str，而示例代码中的类型注解明显没有对上述两种情况加以约束。

def add(a: Union[str, int], b: Union[str, int]) -> Union[str, int]:
    return a + b

文档

官方文档：https://docs.python.org/zh-cn/3.7/library/typing.html#generics

应用

使用泛型可以很好地解决上述问题，示例代码如下所示，定义一个泛型类型T，并约束它的类型为int，str（有点疑惑约束为和必须填两个参数？），并将a、b参数和函数返回值类型均设置为T。

from typing import TypeVar

T = TypeVar('T', int, str)


def add(a: T, b: T) -> T:
    return a + b

IDE提示效果如下图：

1. 返回类型为str与value1的类型不匹配；

2. 正常

3. 正常

4. 返回类型为int与value4的类型不匹配；

5. 第一个参数为int类型，第二个参数也必须为int，提示类型不一致。

综上所述，泛型可以解决传入参数的类型不确定，而函数的返回值类型与参数类型一致的情况，由于在此前工作中并不知道泛型，故一直认为Python不支持此类型的类型注解，所以在此记记录一下。