Аннотации типа Python для значения Enum - PullRequest
0 голосов
/ 03 октября 2018

У меня есть этот фрагмент кода:

import enum


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


def get_color_return_something(some_color):
    pass

Как правильно добавить аннотации типов в переменную some_color в этой функции, если я предполагаю получить значение из перечисления Color (например,: Color.RED)?

Ответы [ 2 ]

0 голосов
/ 03 октября 2018

Тип, намекающий на класс Color, должен работать:

def get_color_return_something(some_color: Color):
    print(some_color.value)
0 голосов
/ 03 октября 2018
def get_color_return_something(some_color: Color):
pass
...