python - 字段类型取决于其他字段的类型

Question

是否可以创建一个类似的类

from typing import Union, Literal

class Foo:
    bar: Union[str, int]
    qux: Literal["str", "int"]

这样，如果 qux 是 Literal["str"]，则 bar 是 str 类型，如果 qux 是 Literal["int"]，那么 bar 是 int 类型？是否可以对其进行注释？

我知道 typing.overload，但我认为它与此示例无关

Answer 1

Python 的typing 系统通常不支持依赖类型。但是，可以模拟某些特定情况。

对于较少数量的依赖类型，可以枚举案例。这需要使各个类型通用:

from typing import Union, Literal, Generic, TypeVar

Bar = TypeVar("Bar", str, int)
Qux = TypeVar("Qux", Literal["str"], Literal["int"])


class GenericFoo(Generic[Bar, Qux]):
    bar: Bar
    qux: Qux

    # not always needed – used to infer types from instantiation
    def __init__(self, bar: Bar, qux: Qux): pass

然后可以定义依赖关系

• 作为案例的 Union:

  Foo = Union[GenericFoo[str, Literal["str"]], GenericFoo[int, Literal["int"]]]
  
  f: Foo
  f = GenericFoo("one", "str")
  f = GenericFoo(2, "int")
  f = GenericFoo("three", "int")
  

• 通过重载实例化:

  class GenericFoo(Generic[Bar, Qux]):
      bar: Bar
      qux: Qux
  
      @overload
      def __init__(self, bar: str, qux: Literal["str"]):
          pass
  
      @overload
      def __init__(self, bar: int, qux: Literal["int"]):
          pass
  
      def __init__(self, bar: Bar, qux: Qux):  # type: ignore
          pass