Python enum - getting value of enum on string conv

2019-02-03 21:40发布

I have following enum defined

from enum import Enum


class D(Enum):
    x = 1
    y = 2


print(D.x)

now the printed value is

D.x

instead I wanted the enum's value to be print

1

what can be done to achieve this functionality.

2条回答
Luminary・发光体
2楼-- · 2019-02-03 22:03

You are printing the enum object. Use the .value attribute if you wanted just to print that:

print(D.x.value)

See the Programmatic access to enumeration members and their attributes section:

If you have an enum member and need its name or value:

>>>
>>> member = Color.red
>>> member.name
'red'
>>> member.value
1

You could add a __str__ method to your enum, if all you wanted was to provide a custom string representation:

class D(Enum):
    def __str__(self):
        return str(self.value)

    x = 1
    y = 2

Demo:

>>> from enum import Enum
>>> class D(Enum):
...     def __str__(self):
...         return str(self.value)
...     x = 1
...     y = 2
... 
>>> D.x
<D.x: 1>
>>> print(D.x)
1
查看更多
3楼-- · 2019-02-03 22:17

I implemented access using the following

class D(Enum):
    x = 1
    y = 2

    def __str__(self):
        return '%s' % self._value_

now I can just do

print D.x to get the print.

查看更多
登录 后发表回答