mirror of https://github.com/python/cpython
1118 lines
33 KiB
ReStructuredText
1118 lines
33 KiB
ReStructuredText
:mod:`enum` --- Support for enumerations
|
|
========================================
|
|
|
|
.. module:: enum
|
|
:synopsis: Implementation of an enumeration class.
|
|
|
|
.. moduleauthor:: Ethan Furman <ethan@stoneleaf.us>
|
|
.. sectionauthor:: Barry Warsaw <barry@python.org>
|
|
.. sectionauthor:: Eli Bendersky <eliben@gmail.com>
|
|
.. sectionauthor:: Ethan Furman <ethan@stoneleaf.us>
|
|
|
|
.. versionadded:: 3.4
|
|
|
|
**Source code:** :source:`Lib/enum.py`
|
|
|
|
----------------
|
|
|
|
An enumeration is a set of symbolic names (members) bound to unique,
|
|
constant values. Within an enumeration, the members can be compared
|
|
by identity, and the enumeration itself can be iterated over.
|
|
|
|
|
|
Module Contents
|
|
---------------
|
|
|
|
This module defines four enumeration classes that can be used to define unique
|
|
sets of names and values: :class:`Enum`, :class:`IntEnum`, :class:`Flag`, and
|
|
:class:`IntFlag`. It also defines one decorator, :func:`unique`, and one
|
|
helper, :class:`auto`.
|
|
|
|
.. class:: Enum
|
|
|
|
Base class for creating enumerated constants. See section
|
|
`Functional API`_ for an alternate construction syntax.
|
|
|
|
.. class:: IntEnum
|
|
|
|
Base class for creating enumerated constants that are also
|
|
subclasses of :class:`int`.
|
|
|
|
.. class:: IntFlag
|
|
|
|
Base class for creating enumerated constants that can be combined using
|
|
the bitwise operators without losing their :class:`IntFlag` membership.
|
|
:class:`IntFlag` members are also subclasses of :class:`int`.
|
|
|
|
.. class:: Flag
|
|
|
|
Base class for creating enumerated constants that can be combined using
|
|
the bitwise operations without losing their :class:`Flag` membership.
|
|
|
|
.. function:: unique
|
|
|
|
Enum class decorator that ensures only one name is bound to any one value.
|
|
|
|
.. class:: auto
|
|
|
|
Instances are replaced with an appropriate value for Enum members.
|
|
|
|
.. versionadded:: 3.6 ``Flag``, ``IntFlag``, ``auto``
|
|
|
|
|
|
Creating an Enum
|
|
----------------
|
|
|
|
Enumerations are created using the :keyword:`class` syntax, which makes them
|
|
easy to read and write. An alternative creation method is described in
|
|
`Functional API`_. To define an enumeration, subclass :class:`Enum` as
|
|
follows::
|
|
|
|
>>> from enum import Enum
|
|
>>> class Color(Enum):
|
|
... RED = 1
|
|
... GREEN = 2
|
|
... BLUE = 3
|
|
...
|
|
|
|
.. note:: Enum member values
|
|
|
|
Member values can be anything: :class:`int`, :class:`str`, etc.. If
|
|
the exact value is unimportant you may use :class:`auto` instances and an
|
|
appropriate value will be chosen for you. Care must be taken if you mix
|
|
:class:`auto` with other values.
|
|
|
|
.. note:: Nomenclature
|
|
|
|
- The class :class:`Color` is an *enumeration* (or *enum*)
|
|
- The attributes :attr:`Color.RED`, :attr:`Color.GREEN`, etc., are
|
|
*enumeration members* (or *enum members*) and are functionally constants.
|
|
- The enum members have *names* and *values* (the name of
|
|
:attr:`Color.RED` is ``RED``, the value of :attr:`Color.BLUE` is
|
|
``3``, etc.)
|
|
|
|
.. note::
|
|
|
|
Even though we use the :keyword:`class` syntax to create Enums, Enums
|
|
are not normal Python classes. See `How are Enums different?`_ for
|
|
more details.
|
|
|
|
Enumeration members have human readable string representations::
|
|
|
|
>>> print(Color.RED)
|
|
Color.RED
|
|
|
|
...while their ``repr`` has more information::
|
|
|
|
>>> print(repr(Color.RED))
|
|
<Color.RED: 1>
|
|
|
|
The *type* of an enumeration member is the enumeration it belongs to::
|
|
|
|
>>> type(Color.RED)
|
|
<enum 'Color'>
|
|
>>> isinstance(Color.GREEN, Color)
|
|
True
|
|
>>>
|
|
|
|
Enum members also have a property that contains just their item name::
|
|
|
|
>>> print(Color.RED.name)
|
|
RED
|
|
|
|
Enumerations support iteration, in definition order::
|
|
|
|
>>> class Shake(Enum):
|
|
... VANILLA = 7
|
|
... CHOCOLATE = 4
|
|
... COOKIES = 9
|
|
... MINT = 3
|
|
...
|
|
>>> for shake in Shake:
|
|
... print(shake)
|
|
...
|
|
Shake.VANILLA
|
|
Shake.CHOCOLATE
|
|
Shake.COOKIES
|
|
Shake.MINT
|
|
|
|
Enumeration members are hashable, so they can be used in dictionaries and sets::
|
|
|
|
>>> apples = {}
|
|
>>> apples[Color.RED] = 'red delicious'
|
|
>>> apples[Color.GREEN] = 'granny smith'
|
|
>>> apples == {Color.RED: 'red delicious', Color.GREEN: 'granny smith'}
|
|
True
|
|
|
|
|
|
Programmatic access to enumeration members and their attributes
|
|
---------------------------------------------------------------
|
|
|
|
Sometimes it's useful to access members in enumerations programmatically (i.e.
|
|
situations where ``Color.RED`` won't do because the exact color is not known
|
|
at program-writing time). ``Enum`` allows such access::
|
|
|
|
>>> Color(1)
|
|
<Color.RED: 1>
|
|
>>> Color(3)
|
|
<Color.BLUE: 3>
|
|
|
|
If you want to access enum members by *name*, use item access::
|
|
|
|
>>> Color['RED']
|
|
<Color.RED: 1>
|
|
>>> Color['GREEN']
|
|
<Color.GREEN: 2>
|
|
|
|
If you have an enum member and need its :attr:`name` or :attr:`value`::
|
|
|
|
>>> member = Color.RED
|
|
>>> member.name
|
|
'RED'
|
|
>>> member.value
|
|
1
|
|
|
|
|
|
Duplicating enum members and values
|
|
-----------------------------------
|
|
|
|
Having two enum members with the same name is invalid::
|
|
|
|
>>> class Shape(Enum):
|
|
... SQUARE = 2
|
|
... SQUARE = 3
|
|
...
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: Attempted to reuse key: 'SQUARE'
|
|
|
|
However, two enum members are allowed to have the same value. Given two members
|
|
A and B with the same value (and A defined first), B is an alias to A. By-value
|
|
lookup of the value of A and B will return A. By-name lookup of B will also
|
|
return A::
|
|
|
|
>>> class Shape(Enum):
|
|
... SQUARE = 2
|
|
... DIAMOND = 1
|
|
... CIRCLE = 3
|
|
... ALIAS_FOR_SQUARE = 2
|
|
...
|
|
>>> Shape.SQUARE
|
|
<Shape.SQUARE: 2>
|
|
>>> Shape.ALIAS_FOR_SQUARE
|
|
<Shape.SQUARE: 2>
|
|
>>> Shape(2)
|
|
<Shape.SQUARE: 2>
|
|
|
|
.. note::
|
|
|
|
Attempting to create a member with the same name as an already
|
|
defined attribute (another member, a method, etc.) or attempting to create
|
|
an attribute with the same name as a member is not allowed.
|
|
|
|
|
|
Ensuring unique enumeration values
|
|
----------------------------------
|
|
|
|
By default, enumerations allow multiple names as aliases for the same value.
|
|
When this behavior isn't desired, the following decorator can be used to
|
|
ensure each value is used only once in the enumeration:
|
|
|
|
.. decorator:: unique
|
|
|
|
A :keyword:`class` decorator specifically for enumerations. It searches an
|
|
enumeration's :attr:`__members__` gathering any aliases it finds; if any are
|
|
found :exc:`ValueError` is raised with the details::
|
|
|
|
>>> from enum import Enum, unique
|
|
>>> @unique
|
|
... class Mistake(Enum):
|
|
... ONE = 1
|
|
... TWO = 2
|
|
... THREE = 3
|
|
... FOUR = 3
|
|
...
|
|
Traceback (most recent call last):
|
|
...
|
|
ValueError: duplicate values found in <enum 'Mistake'>: FOUR -> THREE
|
|
|
|
|
|
Using automatic values
|
|
----------------------
|
|
|
|
If the exact value is unimportant you can use :class:`auto`::
|
|
|
|
>>> from enum import Enum, auto
|
|
>>> class Color(Enum):
|
|
... RED = auto()
|
|
... BLUE = auto()
|
|
... GREEN = auto()
|
|
...
|
|
>>> list(Color)
|
|
[<Color.RED: 1>, <Color.BLUE: 2>, <Color.GREEN: 3>]
|
|
|
|
The values are chosen by :func:`_generate_next_value_`, which can be
|
|
overridden::
|
|
|
|
>>> class AutoName(Enum):
|
|
... def _generate_next_value_(name, start, count, last_values):
|
|
... return name
|
|
...
|
|
>>> class Ordinal(AutoName):
|
|
... NORTH = auto()
|
|
... SOUTH = auto()
|
|
... EAST = auto()
|
|
... WEST = auto()
|
|
...
|
|
>>> list(Ordinal)
|
|
[<Ordinal.NORTH: 'NORTH'>, <Ordinal.SOUTH: 'SOUTH'>, <Ordinal.EAST: 'EAST'>, <Ordinal.WEST: 'WEST'>]
|
|
|
|
.. note::
|
|
|
|
The goal of the default :meth:`_generate_next_value_` methods is to provide
|
|
the next :class:`int` in sequence with the last :class:`int` provided, but
|
|
the way it does this is an implementation detail and may change.
|
|
|
|
Iteration
|
|
---------
|
|
|
|
Iterating over the members of an enum does not provide the aliases::
|
|
|
|
>>> list(Shape)
|
|
[<Shape.SQUARE: 2>, <Shape.DIAMOND: 1>, <Shape.CIRCLE: 3>]
|
|
|
|
The special attribute ``__members__`` is a read-only ordered mapping of names
|
|
to members. It includes all names defined in the enumeration, including the
|
|
aliases::
|
|
|
|
>>> for name, member in Shape.__members__.items():
|
|
... name, member
|
|
...
|
|
('SQUARE', <Shape.SQUARE: 2>)
|
|
('DIAMOND', <Shape.DIAMOND: 1>)
|
|
('CIRCLE', <Shape.CIRCLE: 3>)
|
|
('ALIAS_FOR_SQUARE', <Shape.SQUARE: 2>)
|
|
|
|
The ``__members__`` attribute can be used for detailed programmatic access to
|
|
the enumeration members. For example, finding all the aliases::
|
|
|
|
>>> [name for name, member in Shape.__members__.items() if member.name != name]
|
|
['ALIAS_FOR_SQUARE']
|
|
|
|
|
|
Comparisons
|
|
-----------
|
|
|
|
Enumeration members are compared by identity::
|
|
|
|
>>> Color.RED is Color.RED
|
|
True
|
|
>>> Color.RED is Color.BLUE
|
|
False
|
|
>>> Color.RED is not Color.BLUE
|
|
True
|
|
|
|
Ordered comparisons between enumeration values are *not* supported. Enum
|
|
members are not integers (but see `IntEnum`_ below)::
|
|
|
|
>>> Color.RED < Color.BLUE
|
|
Traceback (most recent call last):
|
|
File "<stdin>", line 1, in <module>
|
|
TypeError: '<' not supported between instances of 'Color' and 'Color'
|
|
|
|
Equality comparisons are defined though::
|
|
|
|
>>> Color.BLUE == Color.RED
|
|
False
|
|
>>> Color.BLUE != Color.RED
|
|
True
|
|
>>> Color.BLUE == Color.BLUE
|
|
True
|
|
|
|
Comparisons against non-enumeration values will always compare not equal
|
|
(again, :class:`IntEnum` was explicitly designed to behave differently, see
|
|
below)::
|
|
|
|
>>> Color.BLUE == 2
|
|
False
|
|
|
|
|
|
Allowed members and attributes of enumerations
|
|
----------------------------------------------
|
|
|
|
The examples above use integers for enumeration values. Using integers is
|
|
short and handy (and provided by default by the `Functional API`_), but not
|
|
strictly enforced. In the vast majority of use-cases, one doesn't care what
|
|
the actual value of an enumeration is. But if the value *is* important,
|
|
enumerations can have arbitrary values.
|
|
|
|
Enumerations are Python classes, and can have methods and special methods as
|
|
usual. If we have this enumeration::
|
|
|
|
>>> class Mood(Enum):
|
|
... FUNKY = 1
|
|
... HAPPY = 3
|
|
...
|
|
... def describe(self):
|
|
... # self is the member here
|
|
... return self.name, self.value
|
|
...
|
|
... def __str__(self):
|
|
... return 'my custom str! {0}'.format(self.value)
|
|
...
|
|
... @classmethod
|
|
... def favorite_mood(cls):
|
|
... # cls here is the enumeration
|
|
... return cls.HAPPY
|
|
...
|
|
|
|
Then::
|
|
|
|
>>> Mood.favorite_mood()
|
|
<Mood.HAPPY: 3>
|
|
>>> Mood.HAPPY.describe()
|
|
('HAPPY', 3)
|
|
>>> str(Mood.FUNKY)
|
|
'my custom str! 1'
|
|
|
|
The rules for what is allowed are as follows: names that start and end with
|
|
a single underscore are reserved by enum and cannot be used; all other
|
|
attributes defined within an enumeration will become members of this
|
|
enumeration, with the exception of special methods (:meth:`__str__`,
|
|
:meth:`__add__`, etc.), descriptors (methods are also descriptors), and
|
|
variable names listed in :attr:`_ignore_`.
|
|
|
|
Note: if your enumeration defines :meth:`__new__` and/or :meth:`__init__` then
|
|
whatever value(s) were given to the enum member will be passed into those
|
|
methods. See `Planet`_ for an example.
|
|
|
|
|
|
Restricted subclassing of enumerations
|
|
--------------------------------------
|
|
|
|
Subclassing an enumeration is allowed only if the enumeration does not define
|
|
any members. So this is forbidden::
|
|
|
|
>>> class MoreColor(Color):
|
|
... PINK = 17
|
|
...
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: Cannot extend enumerations
|
|
|
|
But this is allowed::
|
|
|
|
>>> class Foo(Enum):
|
|
... def some_behavior(self):
|
|
... pass
|
|
...
|
|
>>> class Bar(Foo):
|
|
... HAPPY = 1
|
|
... SAD = 2
|
|
...
|
|
|
|
Allowing subclassing of enums that define members would lead to a violation of
|
|
some important invariants of types and instances. On the other hand, it makes
|
|
sense to allow sharing some common behavior between a group of enumerations.
|
|
(See `OrderedEnum`_ for an example.)
|
|
|
|
|
|
Pickling
|
|
--------
|
|
|
|
Enumerations can be pickled and unpickled::
|
|
|
|
>>> from test.test_enum import Fruit
|
|
>>> from pickle import dumps, loads
|
|
>>> Fruit.TOMATO is loads(dumps(Fruit.TOMATO))
|
|
True
|
|
|
|
The usual restrictions for pickling apply: picklable enums must be defined in
|
|
the top level of a module, since unpickling requires them to be importable
|
|
from that module.
|
|
|
|
.. note::
|
|
|
|
With pickle protocol version 4 it is possible to easily pickle enums
|
|
nested in other classes.
|
|
|
|
It is possible to modify how Enum members are pickled/unpickled by defining
|
|
:meth:`__reduce_ex__` in the enumeration class.
|
|
|
|
|
|
Functional API
|
|
--------------
|
|
|
|
The :class:`Enum` class is callable, providing the following functional API::
|
|
|
|
>>> Animal = Enum('Animal', 'ANT BEE CAT DOG')
|
|
>>> Animal
|
|
<enum 'Animal'>
|
|
>>> Animal.ANT
|
|
<Animal.ANT: 1>
|
|
>>> Animal.ANT.value
|
|
1
|
|
>>> list(Animal)
|
|
[<Animal.ANT: 1>, <Animal.BEE: 2>, <Animal.CAT: 3>, <Animal.DOG: 4>]
|
|
|
|
The semantics of this API resemble :class:`~collections.namedtuple`. The first
|
|
argument of the call to :class:`Enum` is the name of the enumeration.
|
|
|
|
The second argument is the *source* of enumeration member names. It can be a
|
|
whitespace-separated string of names, a sequence of names, a sequence of
|
|
2-tuples with key/value pairs, or a mapping (e.g. dictionary) of names to
|
|
values. The last two options enable assigning arbitrary values to
|
|
enumerations; the others auto-assign increasing integers starting with 1 (use
|
|
the ``start`` parameter to specify a different starting value). A
|
|
new class derived from :class:`Enum` is returned. In other words, the above
|
|
assignment to :class:`Animal` is equivalent to::
|
|
|
|
>>> class Animal(Enum):
|
|
... ANT = 1
|
|
... BEE = 2
|
|
... CAT = 3
|
|
... DOG = 4
|
|
...
|
|
|
|
The reason for defaulting to ``1`` as the starting number and not ``0`` is
|
|
that ``0`` is ``False`` in a boolean sense, but enum members all evaluate
|
|
to ``True``.
|
|
|
|
Pickling enums created with the functional API can be tricky as frame stack
|
|
implementation details are used to try and figure out which module the
|
|
enumeration is being created in (e.g. it will fail if you use a utility
|
|
function in separate module, and also may not work on IronPython or Jython).
|
|
The solution is to specify the module name explicitly as follows::
|
|
|
|
>>> Animal = Enum('Animal', 'ANT BEE CAT DOG', module=__name__)
|
|
|
|
.. warning::
|
|
|
|
If ``module`` is not supplied, and Enum cannot determine what it is,
|
|
the new Enum members will not be unpicklable; to keep errors closer to
|
|
the source, pickling will be disabled.
|
|
|
|
The new pickle protocol 4 also, in some circumstances, relies on
|
|
:attr:`~definition.__qualname__` being set to the location where pickle will be able
|
|
to find the class. For example, if the class was made available in class
|
|
SomeData in the global scope::
|
|
|
|
>>> Animal = Enum('Animal', 'ANT BEE CAT DOG', qualname='SomeData.Animal')
|
|
|
|
The complete signature is::
|
|
|
|
Enum(value='NewEnumName', names=<...>, *, module='...', qualname='...', type=<mixed-in class>, start=1)
|
|
|
|
:value: What the new Enum class will record as its name.
|
|
|
|
:names: The Enum members. This can be a whitespace or comma separated string
|
|
(values will start at 1 unless otherwise specified)::
|
|
|
|
'RED GREEN BLUE' | 'RED,GREEN,BLUE' | 'RED, GREEN, BLUE'
|
|
|
|
or an iterator of names::
|
|
|
|
['RED', 'GREEN', 'BLUE']
|
|
|
|
or an iterator of (name, value) pairs::
|
|
|
|
[('CYAN', 4), ('MAGENTA', 5), ('YELLOW', 6)]
|
|
|
|
or a mapping::
|
|
|
|
{'CHARTREUSE': 7, 'SEA_GREEN': 11, 'ROSEMARY': 42}
|
|
|
|
:module: name of module where new Enum class can be found.
|
|
|
|
:qualname: where in module new Enum class can be found.
|
|
|
|
:type: type to mix in to new Enum class.
|
|
|
|
:start: number to start counting at if only names are passed in.
|
|
|
|
.. versionchanged:: 3.5
|
|
The *start* parameter was added.
|
|
|
|
|
|
Derived Enumerations
|
|
--------------------
|
|
|
|
IntEnum
|
|
^^^^^^^
|
|
|
|
The first variation of :class:`Enum` that is provided is also a subclass of
|
|
:class:`int`. Members of an :class:`IntEnum` can be compared to integers;
|
|
by extension, integer enumerations of different types can also be compared
|
|
to each other::
|
|
|
|
>>> from enum import IntEnum
|
|
>>> class Shape(IntEnum):
|
|
... CIRCLE = 1
|
|
... SQUARE = 2
|
|
...
|
|
>>> class Request(IntEnum):
|
|
... POST = 1
|
|
... GET = 2
|
|
...
|
|
>>> Shape == 1
|
|
False
|
|
>>> Shape.CIRCLE == 1
|
|
True
|
|
>>> Shape.CIRCLE == Request.POST
|
|
True
|
|
|
|
However, they still can't be compared to standard :class:`Enum` enumerations::
|
|
|
|
>>> class Shape(IntEnum):
|
|
... CIRCLE = 1
|
|
... SQUARE = 2
|
|
...
|
|
>>> class Color(Enum):
|
|
... RED = 1
|
|
... GREEN = 2
|
|
...
|
|
>>> Shape.CIRCLE == Color.RED
|
|
False
|
|
|
|
:class:`IntEnum` values behave like integers in other ways you'd expect::
|
|
|
|
>>> int(Shape.CIRCLE)
|
|
1
|
|
>>> ['a', 'b', 'c'][Shape.CIRCLE]
|
|
'b'
|
|
>>> [i for i in range(Shape.SQUARE)]
|
|
[0, 1]
|
|
|
|
|
|
IntFlag
|
|
^^^^^^^
|
|
|
|
The next variation of :class:`Enum` provided, :class:`IntFlag`, is also based
|
|
on :class:`int`. The difference being :class:`IntFlag` members can be combined
|
|
using the bitwise operators (&, \|, ^, ~) and the result is still an
|
|
:class:`IntFlag` member. However, as the name implies, :class:`IntFlag`
|
|
members also subclass :class:`int` and can be used wherever an :class:`int` is
|
|
used. Any operation on an :class:`IntFlag` member besides the bit-wise
|
|
operations will lose the :class:`IntFlag` membership.
|
|
|
|
.. versionadded:: 3.6
|
|
|
|
Sample :class:`IntFlag` class::
|
|
|
|
>>> from enum import IntFlag
|
|
>>> class Perm(IntFlag):
|
|
... R = 4
|
|
... W = 2
|
|
... X = 1
|
|
...
|
|
>>> Perm.R | Perm.W
|
|
<Perm.R|W: 6>
|
|
>>> Perm.R + Perm.W
|
|
6
|
|
>>> RW = Perm.R | Perm.W
|
|
>>> Perm.R in RW
|
|
True
|
|
|
|
It is also possible to name the combinations::
|
|
|
|
>>> class Perm(IntFlag):
|
|
... R = 4
|
|
... W = 2
|
|
... X = 1
|
|
... RWX = 7
|
|
>>> Perm.RWX
|
|
<Perm.RWX: 7>
|
|
>>> ~Perm.RWX
|
|
<Perm.-8: -8>
|
|
|
|
Another important difference between :class:`IntFlag` and :class:`Enum` is that
|
|
if no flags are set (the value is 0), its boolean evaluation is :data:`False`::
|
|
|
|
>>> Perm.R & Perm.X
|
|
<Perm.0: 0>
|
|
>>> bool(Perm.R & Perm.X)
|
|
False
|
|
|
|
Because :class:`IntFlag` members are also subclasses of :class:`int` they can
|
|
be combined with them::
|
|
|
|
>>> Perm.X | 8
|
|
<Perm.8|X: 9>
|
|
|
|
|
|
Flag
|
|
^^^^
|
|
|
|
The last variation is :class:`Flag`. Like :class:`IntFlag`, :class:`Flag`
|
|
members can be combined using the bitwise operators (&, \|, ^, ~). Unlike
|
|
:class:`IntFlag`, they cannot be combined with, nor compared against, any
|
|
other :class:`Flag` enumeration, nor :class:`int`. While it is possible to
|
|
specify the values directly it is recommended to use :class:`auto` as the
|
|
value and let :class:`Flag` select an appropriate value.
|
|
|
|
.. versionadded:: 3.6
|
|
|
|
Like :class:`IntFlag`, if a combination of :class:`Flag` members results in no
|
|
flags being set, the boolean evaluation is :data:`False`::
|
|
|
|
>>> from enum import Flag, auto
|
|
>>> class Color(Flag):
|
|
... RED = auto()
|
|
... BLUE = auto()
|
|
... GREEN = auto()
|
|
...
|
|
>>> Color.RED & Color.GREEN
|
|
<Color.0: 0>
|
|
>>> bool(Color.RED & Color.GREEN)
|
|
False
|
|
|
|
Individual flags should have values that are powers of two (1, 2, 4, 8, ...),
|
|
while combinations of flags won't::
|
|
|
|
>>> class Color(Flag):
|
|
... RED = auto()
|
|
... BLUE = auto()
|
|
... GREEN = auto()
|
|
... WHITE = RED | BLUE | GREEN
|
|
...
|
|
>>> Color.WHITE
|
|
<Color.WHITE: 7>
|
|
|
|
Giving a name to the "no flags set" condition does not change its boolean
|
|
value::
|
|
|
|
>>> class Color(Flag):
|
|
... BLACK = 0
|
|
... RED = auto()
|
|
... BLUE = auto()
|
|
... GREEN = auto()
|
|
...
|
|
>>> Color.BLACK
|
|
<Color.BLACK: 0>
|
|
>>> bool(Color.BLACK)
|
|
False
|
|
|
|
.. note::
|
|
|
|
For the majority of new code, :class:`Enum` and :class:`Flag` are strongly
|
|
recommended, since :class:`IntEnum` and :class:`IntFlag` break some
|
|
semantic promises of an enumeration (by being comparable to integers, and
|
|
thus by transitivity to other unrelated enumerations). :class:`IntEnum`
|
|
and :class:`IntFlag` should be used only in cases where :class:`Enum` and
|
|
:class:`Flag` will not do; for example, when integer constants are replaced
|
|
with enumerations, or for interoperability with other systems.
|
|
|
|
|
|
Others
|
|
^^^^^^
|
|
|
|
While :class:`IntEnum` is part of the :mod:`enum` module, it would be very
|
|
simple to implement independently::
|
|
|
|
class IntEnum(int, Enum):
|
|
pass
|
|
|
|
This demonstrates how similar derived enumerations can be defined; for example
|
|
a :class:`StrEnum` that mixes in :class:`str` instead of :class:`int`.
|
|
|
|
Some rules:
|
|
|
|
1. When subclassing :class:`Enum`, mix-in types must appear before
|
|
:class:`Enum` itself in the sequence of bases, as in the :class:`IntEnum`
|
|
example above.
|
|
2. While :class:`Enum` can have members of any type, once you mix in an
|
|
additional type, all the members must have values of that type, e.g.
|
|
:class:`int` above. This restriction does not apply to mix-ins which only
|
|
add methods and don't specify another data type such as :class:`int` or
|
|
:class:`str`.
|
|
3. When another data type is mixed in, the :attr:`value` attribute is *not the
|
|
same* as the enum member itself, although it is equivalent and will compare
|
|
equal.
|
|
4. %-style formatting: `%s` and `%r` call the :class:`Enum` class's
|
|
:meth:`__str__` and :meth:`__repr__` respectively; other codes (such as
|
|
`%i` or `%h` for IntEnum) treat the enum member as its mixed-in type.
|
|
5. :ref:`Formatted string literals <f-strings>`, :meth:`str.format`,
|
|
and :func:`format` will use the mixed-in
|
|
type's :meth:`__format__`. If the :class:`Enum` class's :func:`str` or
|
|
:func:`repr` is desired, use the `!s` or `!r` format codes.
|
|
|
|
|
|
Interesting examples
|
|
--------------------
|
|
|
|
While :class:`Enum`, :class:`IntEnum`, :class:`IntFlag`, and :class:`Flag` are
|
|
expected to cover the majority of use-cases, they cannot cover them all. Here
|
|
are recipes for some different types of enumerations that can be used directly,
|
|
or as examples for creating one's own.
|
|
|
|
|
|
Omitting values
|
|
^^^^^^^^^^^^^^^
|
|
|
|
In many use-cases one doesn't care what the actual value of an enumeration
|
|
is. There are several ways to define this type of simple enumeration:
|
|
|
|
- use instances of :class:`auto` for the value
|
|
- use instances of :class:`object` as the value
|
|
- use a descriptive string as the value
|
|
- use a tuple as the value and a custom :meth:`__new__` to replace the
|
|
tuple with an :class:`int` value
|
|
|
|
Using any of these methods signifies to the user that these values are not
|
|
important, and also enables one to add, remove, or reorder members without
|
|
having to renumber the remaining members.
|
|
|
|
Whichever method you choose, you should provide a :meth:`repr` that also hides
|
|
the (unimportant) value::
|
|
|
|
>>> class NoValue(Enum):
|
|
... def __repr__(self):
|
|
... return '<%s.%s>' % (self.__class__.__name__, self.name)
|
|
...
|
|
|
|
|
|
Using :class:`auto`
|
|
"""""""""""""""""""
|
|
|
|
Using :class:`auto` would look like::
|
|
|
|
>>> class Color(NoValue):
|
|
... RED = auto()
|
|
... BLUE = auto()
|
|
... GREEN = auto()
|
|
...
|
|
>>> Color.GREEN
|
|
<Color.GREEN>
|
|
|
|
|
|
Using :class:`object`
|
|
"""""""""""""""""""""
|
|
|
|
Using :class:`object` would look like::
|
|
|
|
>>> class Color(NoValue):
|
|
... RED = object()
|
|
... GREEN = object()
|
|
... BLUE = object()
|
|
...
|
|
>>> Color.GREEN
|
|
<Color.GREEN>
|
|
|
|
|
|
Using a descriptive string
|
|
""""""""""""""""""""""""""
|
|
|
|
Using a string as the value would look like::
|
|
|
|
>>> class Color(NoValue):
|
|
... RED = 'stop'
|
|
... GREEN = 'go'
|
|
... BLUE = 'too fast!'
|
|
...
|
|
>>> Color.GREEN
|
|
<Color.GREEN>
|
|
>>> Color.GREEN.value
|
|
'go'
|
|
|
|
|
|
Using a custom :meth:`__new__`
|
|
""""""""""""""""""""""""""""""
|
|
|
|
Using an auto-numbering :meth:`__new__` would look like::
|
|
|
|
>>> class AutoNumber(NoValue):
|
|
... def __new__(cls):
|
|
... value = len(cls.__members__) + 1
|
|
... obj = object.__new__(cls)
|
|
... obj._value_ = value
|
|
... return obj
|
|
...
|
|
>>> class Color(AutoNumber):
|
|
... RED = ()
|
|
... GREEN = ()
|
|
... BLUE = ()
|
|
...
|
|
>>> Color.GREEN
|
|
<Color.GREEN>
|
|
>>> Color.GREEN.value
|
|
2
|
|
|
|
|
|
.. note::
|
|
|
|
The :meth:`__new__` method, if defined, is used during creation of the Enum
|
|
members; it is then replaced by Enum's :meth:`__new__` which is used after
|
|
class creation for lookup of existing members.
|
|
|
|
|
|
OrderedEnum
|
|
^^^^^^^^^^^
|
|
|
|
An ordered enumeration that is not based on :class:`IntEnum` and so maintains
|
|
the normal :class:`Enum` invariants (such as not being comparable to other
|
|
enumerations)::
|
|
|
|
>>> class OrderedEnum(Enum):
|
|
... def __ge__(self, other):
|
|
... if self.__class__ is other.__class__:
|
|
... return self.value >= other.value
|
|
... return NotImplemented
|
|
... def __gt__(self, other):
|
|
... if self.__class__ is other.__class__:
|
|
... return self.value > other.value
|
|
... return NotImplemented
|
|
... def __le__(self, other):
|
|
... if self.__class__ is other.__class__:
|
|
... return self.value <= other.value
|
|
... return NotImplemented
|
|
... def __lt__(self, other):
|
|
... if self.__class__ is other.__class__:
|
|
... return self.value < other.value
|
|
... return NotImplemented
|
|
...
|
|
>>> class Grade(OrderedEnum):
|
|
... A = 5
|
|
... B = 4
|
|
... C = 3
|
|
... D = 2
|
|
... F = 1
|
|
...
|
|
>>> Grade.C < Grade.A
|
|
True
|
|
|
|
|
|
DuplicateFreeEnum
|
|
^^^^^^^^^^^^^^^^^
|
|
|
|
Raises an error if a duplicate member name is found instead of creating an
|
|
alias::
|
|
|
|
>>> class DuplicateFreeEnum(Enum):
|
|
... def __init__(self, *args):
|
|
... cls = self.__class__
|
|
... if any(self.value == e.value for e in cls):
|
|
... a = self.name
|
|
... e = cls(self.value).name
|
|
... raise ValueError(
|
|
... "aliases not allowed in DuplicateFreeEnum: %r --> %r"
|
|
... % (a, e))
|
|
...
|
|
>>> class Color(DuplicateFreeEnum):
|
|
... RED = 1
|
|
... GREEN = 2
|
|
... BLUE = 3
|
|
... GRENE = 2
|
|
...
|
|
Traceback (most recent call last):
|
|
...
|
|
ValueError: aliases not allowed in DuplicateFreeEnum: 'GRENE' --> 'GREEN'
|
|
|
|
.. note::
|
|
|
|
This is a useful example for subclassing Enum to add or change other
|
|
behaviors as well as disallowing aliases. If the only desired change is
|
|
disallowing aliases, the :func:`unique` decorator can be used instead.
|
|
|
|
|
|
Planet
|
|
^^^^^^
|
|
|
|
If :meth:`__new__` or :meth:`__init__` is defined the value of the enum member
|
|
will be passed to those methods::
|
|
|
|
>>> class Planet(Enum):
|
|
... MERCURY = (3.303e+23, 2.4397e6)
|
|
... VENUS = (4.869e+24, 6.0518e6)
|
|
... EARTH = (5.976e+24, 6.37814e6)
|
|
... MARS = (6.421e+23, 3.3972e6)
|
|
... JUPITER = (1.9e+27, 7.1492e7)
|
|
... SATURN = (5.688e+26, 6.0268e7)
|
|
... URANUS = (8.686e+25, 2.5559e7)
|
|
... NEPTUNE = (1.024e+26, 2.4746e7)
|
|
... def __init__(self, mass, radius):
|
|
... self.mass = mass # in kilograms
|
|
... self.radius = radius # in meters
|
|
... @property
|
|
... def surface_gravity(self):
|
|
... # universal gravitational constant (m3 kg-1 s-2)
|
|
... G = 6.67300E-11
|
|
... return G * self.mass / (self.radius * self.radius)
|
|
...
|
|
>>> Planet.EARTH.value
|
|
(5.976e+24, 6378140.0)
|
|
>>> Planet.EARTH.surface_gravity
|
|
9.802652743337129
|
|
|
|
|
|
TimePeriod
|
|
^^^^^^^^^^
|
|
|
|
An example to show the :attr:`_ignore_` attribute in use::
|
|
|
|
>>> from datetime import timedelta
|
|
>>> class Period(timedelta, Enum):
|
|
... "different lengths of time"
|
|
... _ignore_ = 'Period i'
|
|
... Period = vars()
|
|
... for i in range(367):
|
|
... Period['day_%d' % i] = i
|
|
...
|
|
>>> list(Period)[:2]
|
|
[<Period.day_0: datetime.timedelta(0)>, <Period.day_1: datetime.timedelta(days=1)>]
|
|
>>> list(Period)[-2:]
|
|
[<Period.day_365: datetime.timedelta(days=365)>, <Period.day_366: datetime.timedelta(days=366)>]
|
|
|
|
|
|
How are Enums different?
|
|
------------------------
|
|
|
|
Enums have a custom metaclass that affects many aspects of both derived Enum
|
|
classes and their instances (members).
|
|
|
|
|
|
Enum Classes
|
|
^^^^^^^^^^^^
|
|
|
|
The :class:`EnumMeta` metaclass is responsible for providing the
|
|
:meth:`__contains__`, :meth:`__dir__`, :meth:`__iter__` and other methods that
|
|
allow one to do things with an :class:`Enum` class that fail on a typical
|
|
class, such as `list(Color)` or `some_var in Color`. :class:`EnumMeta` is
|
|
responsible for ensuring that various other methods on the final :class:`Enum`
|
|
class are correct (such as :meth:`__new__`, :meth:`__getnewargs__`,
|
|
:meth:`__str__` and :meth:`__repr__`).
|
|
|
|
|
|
Enum Members (aka instances)
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
The most interesting thing about Enum members is that they are singletons.
|
|
:class:`EnumMeta` creates them all while it is creating the :class:`Enum`
|
|
class itself, and then puts a custom :meth:`__new__` in place to ensure
|
|
that no new ones are ever instantiated by returning only the existing
|
|
member instances.
|
|
|
|
|
|
Finer Points
|
|
^^^^^^^^^^^^
|
|
|
|
Supported ``__dunder__`` names
|
|
""""""""""""""""""""""""""""""
|
|
|
|
:attr:`__members__` is a read-only ordered mapping of ``member_name``:``member``
|
|
items. It is only available on the class.
|
|
|
|
:meth:`__new__`, if specified, must create and return the enum members; it is
|
|
also a very good idea to set the member's :attr:`_value_` appropriately. Once
|
|
all the members are created it is no longer used.
|
|
|
|
|
|
Supported ``_sunder_`` names
|
|
""""""""""""""""""""""""""""
|
|
|
|
- ``_name_`` -- name of the member
|
|
- ``_value_`` -- value of the member; can be set / modified in ``__new__``
|
|
|
|
- ``_missing_`` -- a lookup function used when a value is not found; may be
|
|
overridden
|
|
- ``_ignore_`` -- a list of names, either as a :func:`list` or a :func:`str`,
|
|
that will not be transformed into members, and will be removed from the final
|
|
class
|
|
- ``_order_`` -- used in Python 2/3 code to ensure member order is consistent
|
|
(class attribute, removed during class creation)
|
|
- ``_generate_next_value_`` -- used by the `Functional API`_ and by
|
|
:class:`auto` to get an appropriate value for an enum member; may be
|
|
overridden
|
|
|
|
.. versionadded:: 3.6 ``_missing_``, ``_order_``, ``_generate_next_value_``
|
|
.. versionadded:: 3.7 ``_ignore_``
|
|
|
|
To help keep Python 2 / Python 3 code in sync an :attr:`_order_` attribute can
|
|
be provided. It will be checked against the actual order of the enumeration
|
|
and raise an error if the two do not match::
|
|
|
|
>>> class Color(Enum):
|
|
... _order_ = 'RED GREEN BLUE'
|
|
... RED = 1
|
|
... BLUE = 3
|
|
... GREEN = 2
|
|
...
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: member order does not match _order_
|
|
|
|
.. note::
|
|
|
|
In Python 2 code the :attr:`_order_` attribute is necessary as definition
|
|
order is lost before it can be recorded.
|
|
|
|
``Enum`` member type
|
|
""""""""""""""""""""
|
|
|
|
:class:`Enum` members are instances of their :class:`Enum` class, and are
|
|
normally accessed as ``EnumClass.member``. Under certain circumstances they
|
|
can also be accessed as ``EnumClass.member.member``, but you should never do
|
|
this as that lookup may fail or, worse, return something besides the
|
|
:class:`Enum` member you are looking for (this is another good reason to use
|
|
all-uppercase names for members)::
|
|
|
|
>>> class FieldTypes(Enum):
|
|
... name = 0
|
|
... value = 1
|
|
... size = 2
|
|
...
|
|
>>> FieldTypes.value.size
|
|
<FieldTypes.size: 2>
|
|
>>> FieldTypes.size.value
|
|
2
|
|
|
|
.. versionchanged:: 3.5
|
|
|
|
|
|
Boolean value of ``Enum`` classes and members
|
|
"""""""""""""""""""""""""""""""""""""""""""""
|
|
|
|
:class:`Enum` members that are mixed with non-:class:`Enum` types (such as
|
|
:class:`int`, :class:`str`, etc.) are evaluated according to the mixed-in
|
|
type's rules; otherwise, all members evaluate as :data:`True`. To make your
|
|
own Enum's boolean evaluation depend on the member's value add the following to
|
|
your class::
|
|
|
|
def __bool__(self):
|
|
return bool(self.value)
|
|
|
|
:class:`Enum` classes always evaluate as :data:`True`.
|
|
|
|
|
|
``Enum`` classes with methods
|
|
"""""""""""""""""""""""""""""
|
|
|
|
If you give your :class:`Enum` subclass extra methods, like the `Planet`_
|
|
class above, those methods will show up in a :func:`dir` of the member,
|
|
but not of the class::
|
|
|
|
>>> dir(Planet)
|
|
['EARTH', 'JUPITER', 'MARS', 'MERCURY', 'NEPTUNE', 'SATURN', 'URANUS', 'VENUS', '__class__', '__doc__', '__members__', '__module__']
|
|
>>> dir(Planet.EARTH)
|
|
['__class__', '__doc__', '__module__', 'name', 'surface_gravity', 'value']
|
|
|
|
|
|
Combining members of ``Flag``
|
|
"""""""""""""""""""""""""""""
|
|
|
|
If a combination of Flag members is not named, the :func:`repr` will include
|
|
all named flags and all named combinations of flags that are in the value::
|
|
|
|
>>> class Color(Flag):
|
|
... RED = auto()
|
|
... GREEN = auto()
|
|
... BLUE = auto()
|
|
... MAGENTA = RED | BLUE
|
|
... YELLOW = RED | GREEN
|
|
... CYAN = GREEN | BLUE
|
|
...
|
|
>>> Color(3) # named combination
|
|
<Color.YELLOW: 3>
|
|
>>> Color(7) # not named combination
|
|
<Color.CYAN|MAGENTA|BLUE|YELLOW|GREEN|RED: 7>
|
|
|