Приятный отпечаток питона Enum - PullRequest
Новая
       23

Приятный отпечаток питона Enum

0 голосов
/ 20 ноября 2018

Предположим, у меня есть следующее: 

from enum import Enum

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

Как вывод print(Color), я хочу видеть: 

The colors are:
- RED
- GREEN
- BLUE

Я пробовал: 

from enum import Enum

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

    @classmethod
    def __str__(self):
        res = "The colors are:\n"
        for g in set(map(lambda c: c.name, Color)):
            res += '- ' + g + '\n'
        return res

Но это работает только как print(Color(1)).Как заставить его работать при использовании print(Color)?

1 Ответ

0 голосов
/ 20 ноября 2018

Чтобы переопределить печать класса , вы можете определить __str__ в метаклассе : 

from enum import Enum, EnumMeta

class MyEnumMeta(EnumMeta):
    def __str__(cls):
        lines = [f"The {cls.__name__.lower()}s are:"]
        for member in cls:
            lines.append(f"- {member.name}")
        return '\n'.join(lines)

class Color(Enum, metaclass=MyEnumMeta):
    RED = 1
    GREEN = 2
    BLUE = 3

Демо: 

>>> Color
<enum 'Color'>
>>> print(Color)
The colors are:
- RED
- GREEN
- BLUE
>>> Color.RED
<Color.RED: 1>
>>> print(Color.RED)
Color.RED

Обнаружено имя класса: 

>>> class Animal(Enum, metaclass=MyEnumMeta): 
...     cat = 'meow' 
...     dog = 'woof' 
...     badger = 'grrr' 
...
>>> print(Animal)
The animals are:
- cat
- dog
- badger
...