1

我有一些 Python 3.7 代码，我正在尝试向其中添加类型。我要添加的一种类型实际上是Union几个可能的字符串之一：

from typing import Union, Optional, Dict

PossibleKey = Union["fruits", "cars", "vegetables"]
PossibleType = Dict[PossibleKey, str]

def some_function(target: Optional[PossibleType] = None):
  if target:
    all_fruits = target["fruits"]
    print(f"I have {all_fruits}")

这里的问题是 Pyright 抱怨PossibleKey. 它说：

“水果没有定义”

我想让 Pyright/Pylance 工作。

我已经from enum import Enum从另一个 SO 答案中检查了模块，但是如果我尝试这样做，我最终会遇到更多问题，因为我实际上是在处理 aDict[str, Any]而不是Enum.

表示我的类型的正确 Pythonic 方式是什么？

4

2 回答 2

5

"fruits"不是类型（提示），而是Literal["fruits"]。

from typing import Union, Literal

PossibleKey = Union[Literal["fruits"], Literal["cars"], Literal["vegetables"]]

或更短的版本，

PossibleKey = Literal["fruits", "cars", "vegetables"]

或者，正如您所提到的，定义一个Enum由三个值填充的值。

from enum import Enum


class Key(Enum):
    Fruits = "fruits"
    Cars = "cars"
    Vegetables = "vegetables"


def some_function(target: Optional[PossibleType] = None):
    if target:
        all_fruits = target[Key.Fruits]
        print(f"I have {all_fruits}")

（但是，仅仅因为targetis notNone并不一定意味着它实际上具有 "fruits"作为键，只是它没有除Key.Fruits、Key.Cars或之外的键Key.Vegetables。）

于 2021-10-29T10:54:42.430 回答
2

Pyright如果您定义PossibleKey如下，错误就会消失Enum。这只需要对原始代码进行一行更改。如果使用 Enum 有问题，请详细说明。

from typing import Union, Optional, Dict
from enum import Enum

PossibleKey = Enum("PossibleKey", ["fruits", "cars", "vegetables"])
PossibleType = Dict[PossibleKey, str]

def some_function(target: Optional[PossibleType] = None):
  if target:
    all_fruits = target["fruits"]
    print(f"I have {all_fruits}")
于 2021-10-29T11:05:20.800 回答