Python动态类型下如何实现代码自动补全？（Service类场景）

要让PyCharm等IDE识别get方法的返回值类型，需要利用Python的泛型类型提示，明确关联传入的类类型与返回的实例类型。具体修改如下：

1. 导入必要的类型工具 #

首先从typing模块导入用于泛型定义的工具：

from typing import TypeVar, Type

2. 修改Service类的类型提示 #

定义一个泛型变量T来代表任意类类型，然后给get、register方法以及data字典添加精确的类型注解：

T = TypeVar('T')

class Service:
    def __init__(self):
        # 明确字典的键是类类型，值是对应类的实例
        self.data: dict[Type[T], T] = {}

    def get(self, clazz: Type[T]) -> T:
        # 告诉类型检查器：传入的clazz是T类的类型，返回T类的实例
        return self.data[clazz]

    def register(self, clazz: Type[T], *args, **kwargs) -> None:
        # 注册时同样关联类类型与实例类型
        self.data[clazz] = clazz(*args, **kwargs)

3. 验证效果 #

使用原有的业务类和调用代码：

class A:
    def method_of_a(self):
        pass

class B:
    pass

service = Service()
service.register(A)
service.register(B)

x = service.get(A)
x.method_of_a()  # 此时输入x.时IDE会自动提示method_of_a方法

原理说明 #

原来的写法中，IDE无法静态推断get方法返回值的具体类型，因为没有建立“传入的类”和“返回的实例”之间的类型关联。通过TypeVar和Type[T]的组合，我们给类型检查器提供了明确的逻辑：传入的clazz是某个类T的类型，返回的必然是T类的实例，这样IDE就能准确识别返回值的类型，实现自动补全。

内容的提问来源于stack exchange，提问作者Noisy-JohnSmith