如何让Python泛型参数获取函数兼容类型与类型定义？

要解决get_arg函数同时兼容基础类型、自定义类实例和Callable类型的问题，可以通过函数重载结合针对性的泛型约束来实现，让Mypy能正确识别所有场景的类型。

完整实现代码 #

from typing import TypeVar, Callable, Type, Any, overload

# 定义泛型变量：处理普通类型（类、基础类型）
T = TypeVar('T')
# 定义绑定Callable的泛型变量：处理可调用类型
CallableT = TypeVar('CallableT', bound=Callable[..., Any])

@overload
def get_arg(arg_type: Type[T]) -> T:
    """处理普通类型（如str、自定义类）的重载签名"""
    ...

@overload
def get_arg(arg_type: CallableT) -> CallableT:
    """处理Callable类型的重载签名"""
    ...

def get_arg(arg_type: Any) -> Any:
    """实际实现逻辑，根据传入类型返回对应值"""
    # 这里替换为你实际的参数获取逻辑
    if callable(arg_type):
        # 模拟返回符合Callable类型的对象，需根据实际需求调整
        return lambda *args, **kwargs: f"Callable result: {args} {kwargs}"
    else:
        # 处理普通类型，返回对应实例或值
        if arg_type is str:
            return "sample string"
        elif arg_type is int:
            return 123
        else:
            # 对于自定义类，返回其实例
            return arg_type()

关键设计点 #

• 函数重载：通过@overload为Mypy提供两个类型签名，分别匹配普通类型（Type[T]）和Callable类型（CallableT），解决单一泛型无法兼容两种场景的问题。
• Callable泛型约束：用bound=Callable[..., Any]的CallableT确保传入和返回的Callable类型能被Mypy正确识别，避免类型不兼容错误。
• 分支处理逻辑：函数体中通过callable()判断传入的是Callable类型还是普通类型，分别返回对应格式的结果，确保运行时和类型检查的一致性。

测试用例 #

# 测试基础类型
s: str = get_arg(str)
assert isinstance(s, str)

# 测试自定义类
class Class1:
    def __init__(self):
        self.value = "class1 instance"

obj: Class1 = get_arg(Class1)
assert isinstance(obj, Class1)
assert obj.value == "class1 instance"

# 测试Callable类型
concat_func: Callable[[str, str], str] = get_arg(Callable[[str, str], str])
result = concat_func("hello", "world")
assert isinstance(result, str)

运行Mypy检查上述代码不会抛出类型不兼容错误，同时返回值能被正确识别为对应类型。

内容的提问来源于stack exchange，提问作者brandonscript