Python 如何根据参数推断对象的返回类型

发布于04月18日

假设我们有一个功能

def get_attr_wrapper(obj: object, attr: str) -> ???:
    return getattr(obj, attr)

如何根据给定的参数推断返回类型get_attr_wrapper？

也许以某种方式使用仿制药？

例如，如果我通过

from dataclasses import dataclass

@dataclass
class Foo:
    bar: str

foo = Foo(bar="baz")

rv = get_attr_wrapper(foo, "bar")

在我们想要的场景中，Python的类型判断器将rv推断为类型string.

推荐答案

因此，没有真正的方法可以从__getattr__呼叫中找到类型.也就是说，可以让类型判断器相信您的对象 struct 略有不同，然后根据给定字符串推断类型.

TypedDict个对象允许通过字符串而不是属性获取，并且仍然推断类型.我们可以使用cast来"假装"Foo实际上是TypedDict，并以与get_attr_wrapper相同的方式实现__getitem__:

from dataclasses import dataclass
from typing import TypedDict, cast

class FooDict(TypedDict):
    bar: str

@dataclass
class Foo:
    bar: str

    def __getitem__(self, attr:str):
        # Behavior of get_attr_wrapper
        return getattr(self, attr)

foo = cast(FooDict, Foo(bar="baz"))

rv = foo["bar"]
print(rv)

希望这有用！