枚举值的类型注释

Question

我有这段代码：

import enum


class Color(enum.Enum):
    RED = '1'
    BLUE = '2'
    GREEN = '3'


def get_color_return_something(some_color):
    pass

如果我想从Color枚举中接收值（例如：some_color），如何在此函数的Color.RED变量中正确添加类型注释？

Answer 1

提示Color类的类型应该起作用：

def get_color_return_something(some_color: Color):
    print(some_color.value)

Answer 2

def get_color_return_something(some_color: Color):
    pass