Python pour la valeur 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 au some_color disponible dans cette fonction, si je suppose recevoir une valeur de l'énumération Color (par exemple: Color.RED)?

Answer 1

Type indiquant que la classe Couleur 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

Vous pouvez imprimer le nom et la valeur comme ceci:

import enum

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

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

get_color_return_something(Color.RED)
# RED
# 1