如何在Python中键入提示字符串的枚举[重复]

Question

我很好奇如何键入提示字符串枚举，例如： ["keyword1", "keyword2"]

我希望某个变量 v 等于这些字符串文字中的任何一个。我可以通过文字联合来实现这一点 - Union[Literal["keyword1"], Literal["keyword2"]] 但如果这些关键字中的一个在未来发生变化，这将使得可维护性变得困难。 理想情况下，我想这样定义：

class Keywords(enum):
   keywordOne = "keyword1"
   keywordTwo = "keyword2"
v: valueOf[Keywords] = Keywords.keywordOne.value # v = "keyword1"

但我不确定如何在 MyPy 中完成这样的事情

Answer 1

你快到了。您正在寻找的似乎是一个自定义枚举对象，它本身是类型化的，然后键入指示该枚举使用的注释。像这样的：

from enum import Enum
from typing import Literal

class CustomKeyword(Enum):
   keywordOne: Literal["keyword1"] = "keyword1"
   keywordTwo: Literal["keyword2"] = "keyword2"

v: CustomKeyword = CustomKeyword.keywordOne

这不会给你预期的结果吗？