命名参数可以与Python枚举一起使用吗？

Question

例子：

class Planet(Enum):

    MERCURY = (mass: 3.303e+23, radius: 2.4397e6)

    def __init__(self, mass, radius):
        self.mass = mass       # in kilograms
        self.radius = radius   # in meters

参考:https://docs.python.org/3/library/enum.html#planet

我为什么要这么做？如果构造函数列表中有一些基本类型(int，bool ),使用命名参数会更好。

Answer 1

对于Python 3.6.1，可以使用typing.NamedTuple，这也允许设置默认值，这会导致更漂亮的代码。@shao.lo的示例如下所示：

from enum import Enum
from typing import NamedTuple


class Body(NamedTuple):
    mass: float
    radius: float
    moons: int=0


class Planet(Body, Enum):
    MERCURY = Body(mass=3.303e+23, radius=2.4397e6)
    VENUS   = Body(mass=4.869e+24, radius=6.0518e6)
    EARTH   = Body(5.976e+24, 3.3972e6, moons=1)

这也支持酸洗。打字。如果不想指定类型，可以使用Any。

这要归功于@monk时间，谁在这里的答案启发了这个解决方案。

Answer 2

@zero-piraeus的公认答案可以稍微扩展一下，也允许默认参数。如果有一个大型枚举，其中大部分条目的元素值相同，这将非常方便。

class Body(namedtuple('Body', "mass radius moons")):
    def __new__(cls, mass, radius, moons=0):
        return super().__new__(cls, mass, radius, moons)
    def __getnewargs__(self):
        return (self.mass, self.radius, self.moons)

class Planet(Body, Enum):

    MERCURY = Body(mass=3.303e+23, radius=2.4397e6)
    VENUS   = Body(mass=4.869e+24, radius=6.0518e6)
    EARTH   = Body(5.976e+24, 3.3972e6, moons=1)

当心，如果没有__getnewargs__，酸洗将不起作用。

class Foo:
    def __init__(self):
        self.planet = Planet.EARTH  # pickle error in deepcopy

from copy import deepcopy

f1 = Foo()
f2 = deepcopy(f1)  # pickle error here

Answer 3

虽然您不能像使用枚举描述的那样使用命名参数，但您可以使用命名的 mixin 获得类似的效果：

from collections import namedtuple
from enum import Enum

Body = namedtuple("Body", ["mass", "radius"])

class Planet(Body, Enum):

    MERCURY = Body(mass=3.303e+23, radius=2.4397e6)
    VENUS   = Body(mass=4.869e+24, radius=6.0518e6)
    EARTH   = Body(mass=5.976e+24, radius=3.3972e6)
    # ... etc.

…在我看来，这更干净，因为您不必编写__init__方法。

使用示例：

>>> Planet.MERCURY
<Planet.MERCURY: Body(mass=3.303e+23, radius=2439700.0)>
>>> Planet.EARTH.mass
5.976e+24
>>> Planet.VENUS.radius
6051800.0

注意，根据文档，“在碱基序列中，混合类型必须出现在< code>Enum本身之前”。