Annotations de type pour l'attribut Enum

Question

J'ai ce morceau de code:

 import enum


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


def get_color_return_something(some_color):
    pass

Comment ajouter correctement des annotations de type à la some_color dans cette fonction, si je suppose que je vais recevoir un attribut enum de l'énumération Color (par exemple: Color.RED )?

Answer 1

Tapez indiquant que la classe Color devrait fonctionner:

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

Answer 2

def get_color_return_something(some_color: Color):
    pass

Answer 3

Ce qui suit fonctionnera avec Pyton 3.9 / PyCharm

 from enum import Enum
from typing import Optional, Union


class Color(Enum):
    RED: int = 1
    GREEN: int = 2


def guess_color(x: Union[Color.RED, Color.GREEN]) -> Optional[ValueError]:
    if x == Color.RED:
        print("Gotcha!")
    else:
        return ValueError(f"It's not {Color.RED}")


guess_color(Color.RED)