如何在 Python 中使用枚举类型提示

使用枚举类在 Python 中键入提示枚举,例如 def get_value_from_enum(size: Sizes):。 然后,我们可以访问函数主体中枚举的任何成员而不会收到警告。

from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


def get_value_from_enum(size: Sizes):
    print(size.name)  # 👉️ MEDIUM
    print(size.value)  # 👉️ 2

    return size.value


result = get_value_from_enum(Sizes.MEDIUM)

print(result)  # 👉️ 2

如何在 Python 中使用枚举类型提示

我们使用 Sizes 枚举类来键入提示枚举。

另一种方法是使用 Literal 类型来指定函数参数可以是枚举中的成员之一。

from typing import Literal
from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


def get_value_from_enum(size: Literal[Sizes.SMALL, Sizes.MEDIUM, Sizes.LARGE]):
    print(size.name)  # 👉️ MEDIUM
    print(size.value)  # 👉️ 2

    return size.value


result = get_value_from_enum(Sizes.MEDIUM)

print(result)  # 👉️ 2

文字类型提示可用于向类型检查器指示函数参数的值等于提供的文字(或示例中的几个文字之一)。

免责声明:
1.本站所有内容由本站原创、网络转载、消息撰写、网友投稿等几部分组成。
2.本站原创文字内容若未经特别声明,则遵循协议CC3.0共享协议,转载请务必注明原文链接。
3.本站部分来源于网络转载的文章信息是出于传递更多信息之目的,不意味着赞同其观点。
4.本站所有源码与软件均为原作者提供,仅供学习和研究使用。
5.如您对本网站的相关版权有任何异议,或者认为侵犯了您的合法权益,请及时通知我们处理。
火焰兔 » 如何在 Python 中使用枚举类型提示