python - 如何在 Python 中键入提示字符串的枚举

Question

这个问题在这里已经有了答案:





Type annotations for Enum attribute

(5 个回答)


去年关闭。




我很好奇我是如何对字符串枚举进行类型提示的，例如:
[“关键字1”，“关键字2”]
我想要一些变量 v 等于这些字符串中的任何一个。我可以用文字的联合来实现这一点 - Union[Literal["keyword1"], Literal["keyword2"]] 但是如果这些关键字之一在 future 发生变化，将会使可维护性变得困难。
理想情况下，我想定义这样的事情:

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

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