但是，这种方法比较繁琐，需要为每种可能的输入类型组合都定义一个重载。
我们可以使用 TypeVar 和 Generic 来实现更简洁的类型提示。首先，我们定义两个 TypeVar，分别表示 a 和 b 的类型：
from typing import TypeVar, Generic
T1 = TypeVar('T1', str, float)
T2 = TypeVar('T2', str, float)然后，我们定义一个 Generic 类 Promoted，它接受两个类型参数 T1 和 T2，并根据这两个类型参数返回相应的类型：
class Promoted(Generic[T1, T2]):
def __class_getitem__(cls, types):
t1, t2 = types
if t1 == str or t2 == str:
return str
return float最后，我们可以使用 Promoted 类来注释 mul 函数的返回类型：
def method(a: T1, b: T2) -> Promoted[T1, T2]:
if isinstance(a, str) or isinstance(b, str):
return f"{a} {b}"
else:
return a*b这样，mul 函数的返回类型将根据输入参数的类型自动推断。
立即学习“Python免费学习笔记（深入）”；
示例代码
以下是一个完整的示例代码：
from typing import TypeVar, Generic
T1 = TypeVar('T1', str, float)
T2 = TypeVar('T2', str, float)
class Promoted(Generic[T1, T2]):
def __class_getitem__(cls, types):
t1, t2 = types
if t1 == str or t2 == str:
return str
return float
def method(a: T1, b: T2) -> Promoted[T1, T2]:
if isinstance(a, str) or isinstance(b, str):
return f"{a} {b}"
else:
return a*b
result1 = method("Hello", "World") # should be inferred as str
result2 = method(3.0, 4.0) # should be inferred as float
result3 = method("Price", 5.0) # should be inferred as str
print(f"{result1=}, {type(result1)=}")
print(f"{result2=}, {type(result2)=}")
print(f"{result3=}, {type(result3)=}")注意： 目前，一些 IDE 可能无法正确推断 Promoted 类的类型。在这种情况下，类型提示可能会显示为 "Promoted" 而不是 str 或 float。但是，这并不影响代码的实际行为。
总结
本文介绍了一种使用 TypeVar 和 Generic 的方法，以在 Python 中实现类型提升的类型提示。这种方法比使用 @overload 更加简洁，并且可以自动推断函数的返回类型。虽然目前一些 IDE 可能无法完全支持这种方法，但它仍然是一种非常有用的技术，可以提高代码的可读性和可维护性。
