Taper hint Enums en Python #
Utilisez la classe d’énumération pour indiquer une énumération en Python, par ex.
def get_value_from_enum(size: Sizes):
. Vous pouvez alors accéder à n’importe quel membre de l’enum
dans le corps de la fonction sans recevoir d’avertissement.
from enum import Enum
class Sizes(Enum):
SMALL = 1
MEDIUM = 2
LARGE = 3
def get_value_from_enum(size: Sizes):
print(size.name)
print(size.value)
return size.value
result = get_value_from_enum(Sizes.MEDIUM)
print(result)
Nous avons utilisé la fonction Sizes
classe d’énumération pour indiquer un type d’énumération.
Une autre approche consiste à utiliser la classe Literal
pour spécifier que l’argument de la fonction
peut être l’un des membres de l’énumération.
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)
print(size.value)
return size.value
result = get_value_from_enum(Sizes.MEDIUM)
print(result)
Le type littéral
peut être utilisé pour indiquer aux vérificateurs de type que le paramètre de la fonction possède une
valeur égale au littéral fourni (ou à l’un de plusieurs littéraux comme dans l’exemple).
exemple).