Enum: add extended AutoNumber example (GH-22349)
This commit is contained in:
parent
40a0625792
commit
62e40d8450
|
@ -925,6 +925,32 @@ Using an auto-numbering :meth:`__new__` would look like::
|
||||||
>>> Color.GREEN.value
|
>>> Color.GREEN.value
|
||||||
2
|
2
|
||||||
|
|
||||||
|
To make a more general purpose ``AutoNumber``, add ``*args`` to the signature::
|
||||||
|
|
||||||
|
>>> class AutoNumber(NoValue):
|
||||||
|
... def __new__(cls, *args): # this is the only change from above
|
||||||
|
... value = len(cls.__members__) + 1
|
||||||
|
... obj = object.__new__(cls)
|
||||||
|
... obj._value_ = value
|
||||||
|
... return obj
|
||||||
|
...
|
||||||
|
|
||||||
|
Then when you inherit from ``AutoNumber`` you can write your own ``__init__``
|
||||||
|
to handle any extra arguments::
|
||||||
|
|
||||||
|
>>> class Swatch(AutoNumber):
|
||||||
|
... def __init__(self, pantone='unknown'):
|
||||||
|
... self.pantone = pantone
|
||||||
|
... AUBURN = '3497'
|
||||||
|
... SEA_GREEN = '1246'
|
||||||
|
... BLEACHED_CORAL = () # New color, no Pantone code yet!
|
||||||
|
...
|
||||||
|
>>> Swatch.SEA_GREEN
|
||||||
|
<Swatch.SEA_GREEN: 2>
|
||||||
|
>>> Swatch.SEA_GREEN.pantone
|
||||||
|
'1246'
|
||||||
|
>>> Swatch.BLEACHED_CORAL.pantone
|
||||||
|
'unknown'
|
||||||
|
|
||||||
.. note::
|
.. note::
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue