В Python член перечисления содержит базовое значение, которое можно получить, прочитав атрибут value. И наоборот, член перечисления можно получить из его значения, вызвав тип перечисления.
Код: Выделить всё
class Food(enum.Enum):
MEAT = 'meat'
DAIRY = 'dairy'
print(Food.MEAT.value) # 'meat'
print(Food('meat') is Food.MEAT) # True
Мне бы хотелось, чтобы можно было получить один и тот же член перечисления, передав несколько разных значений; очевидно, что только один из них будет каноническим и его можно будет получить из атрибута value. Я могу сделать это следующим образом:
Код: Выделить всё
class Food(enum.Enum):
MEAT = 'meat'
DAIRY = 'dairy'
@classmethod
def from_str(cls, item):
match item:
case 'spam'|'ham':
return cls.MEAT
case 'eggs'|'milk':
return cls.DAIRY
return cls(item)
а затем не забудьте продолжать использовать Food.from_str(s) вместо Food(s). Это осуществимо, хотя я бы предпочел использовать псевдонимы в определении члена вместо повторения членов во вспомогательном методе (как в моем другом вопросе, связанном с перечислениями Python). Но есть и другие проблемы:
Код: Выделить всё
#!/usr/bin/env python3
import argparse
import enum
class Food(enum.Enum):
MEAT = 'meat'
DAIRY = 'dairy'
@classmethod
def from_str(cls, item):
match item:
case 'spam'|'ham':
return cls.MEAT
case 'eggs'|'milk':
return cls.DAIRY
return cls(item)
parser = argparse.ArgumentParser()
parser.add_argument('food', type=Food.from_str)
args = parser.parse_args()
print(args.food)
Когда я вызываю этот скрипт с «хлебом» в качестве аргумента командной строки, argparse жалуется на «недопустимое значение from_str». Поэтому я бы предпочел использовать type=Food в качестве аргумента add_argument, чтобы тип правильно определялся в сообщении об ошибке.
Как это можно решить?
Подробнее здесь:
https://stackoverflow.com/questions/798 ... num-member