键入可能值列表的提示

Question

我有一个可以采用固定值列表的函数：例如

def func(mode="a"):
   if mode not in ["a", "b"]:
      raise AttributeError("not ok")

有没有办法输入提示它只能是这两个值之一？

Answer 1

我认为你想要一个文字类型：

def func(mode = "a": Literal["a", "b"]):
    if mode not in ["a", "b"]:
        raise AttributeError("not ok")

这是通过PEP 586在 Python 3.8 中引入的。