VSCode编写Python时如何让字典取值显示类型补全提示

可以实现该效果，核心是给你封装的字典操作方法补充泛型类型注解，让VSCode依赖的Pyright/Pylance类型检查器可以推导返回值的具体类型。

你现有代码没有提示的原因是：Get方法的返回值注解为object，作为所有Python类型的基类，类型检查器无法识别它实际存储的MyClass类型的专属方法，因此不会弹出对应提示。

改造后可实现类型提示的代码 #

from typing import TypeVar, Optional, cast

# 定义泛型类型变量
T = TypeVar('T')

class TestClassNew(object):
    def __init__(self) -> None:
        super().__init__()
        self.obj = dict()
    
    def FormatKey(self, key : str, *args) -> str:
        if args:
            for k in args:
                key += "_{}"
            key = key.format(*args)
        return key
    
    # 给Set方法添加泛型，标记传入value的类型
    def Set(self, key : str, value : T, *args) -> None:
        key = self.FormatKey(key, *args)
        self.obj[key] = value

    # 给Get方法添加泛型，标记返回值和类型参数对应
    def Get(self, key : str, *args) -> Optional[T]:
        key = self.FormatKey(key, *args)
        if key not in self.obj:
            return None
        return self.obj[key]

class MyClass():
    def Method1(self):
        pass
    def Method2(self):
        pass
    def Method3(self):
        pass

class Wnd(TestClassNew):
    def __init__(self):
        super().__init__()

        a = MyClass()
        a.Method1() # 原有正常提示逻辑不变

        self.Set("somekey", MyClass())
        # 方式1：变量显式标注类型+非空判断，即可正常弹出方法提示
        res: Optional[MyClass] = self.Get("somekey")
        if res:
            res.Method1()
        
        # 方式2：用类型断言直接指定类型，适合100%确定返回值类型的场景
        res2 = cast(MyClass, self.Get("somekey"))
        res2.Method1()

其他适配场景 #

如果你的字典只会存储固定类型的对象，不需要兼容多种类型存储，可以直接给字典本身加类型注解，不需要改造Get/Set方法也能获得提示：

# 仅存储MyClass类型的场景
self.obj: dict[str, MyClass] = dict()

需确保你已经在VSCode中安装了Pylance插件，且类型检查功能正常开启（在设置中搜索python.analysis.typeCheckingMode，设置为basic即可）。

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