使用对象作为枚举的值？

Question

我目前正在尝试创建一个基于枚举/常量的系统，用于在 Python 中定义颜色。我想避免在尝试使用/访问颜色时使用字符串。

在理想的世界里，我希望能够做到：

>>> print(Colours.BLACK)
Black

>>> print(Colours.WHITE.hex)
ffffff

到目前为止，我已经提出了一个基于枚举和对象的系统，如下所示：

from enum import Enum

class Colour:

  def __init__(self, name, hex = "000000"):
    self.name = name
    self.hex = hex

  def __str__(self):
    return self.name


class Colours(Enum):
  WHITE = Colour("White", "ffffff")
  BLACK = Colour("Black", "000000")

print(Colours.BLACK.value)

print(Colours.WHITE.value.hex)

哪个打印：

Black
ffffff

但是，将对象分配为枚举的值感觉像是一种糟糕的做法，我以前在其他地方没有见过这种做法。

这被认为是坏事吗？如果有，为什么？

还有没有更好的方法来实现我想要的结果？

编辑：

我最终采用了这种方法，按照@MichealButscher 的建议使用__hash__，并使用__str__ 和__repr__ 来避免使用.value。

from enum import Enum

class Colour:

  def __init__(self, label, hex_value = "000000"):
    self.label = label
    self.hex_value = hex

  def __str__(self):
    return self.label

  def __hash__(self):
    hash(self.label)


class Colours(Enum):
  WHITE = Colour("White", "ffffff")
  BLACK = Colour("Black", "000000")

  def __repr__(self):
    return self.value

  def __str__(self):
    return str(self.value)

Answer 1

一切都是对象，所以 some 对象将被分配。但是，您可以仅使用元组中的颜色名称和十六进制值作为值，并让Colour.__init__ 处理使用元组初始化每个实例。

class Colour(Enum):
    WHITE = ("White", "ffffff")
    BLACK = ("Black", "000000")

    def __init__(self, label, hexvalue):
        self.label = label
        self.hexvalue = hexvalue

(警告：某些属性名称已在使用中，不能被覆盖甚至分配给，例如，如果您尝试使用name 而不是label 和value 而不是hex value.)

Answer 2

除非您需要单独的颜色类，否则只需将其全部组合到枚举中即可：

from enum import Enum

class Colour(Enum):
    WHITE = "ffffff"
    BLACK = "000000"
    #
    def __str__(self):
        return self.name

并在使用中：

>>> Colour.WHITE
<Colour.WHITE: 'ffffff'>

>>> str(Colour.WHITE)
'WHITE'

>>> Colour.WHITE.value
'ffffff'

---

@martineau，@chepner 说以下将失败：

from enum import Enum

class Colour(Enum):
    WHITE = 'White', "ffffff"
    BLACK = 'black', "000000"
    #
    def __init__(self, name, hex):
        self.name = name              # uh-oh
        self.hex = hex
    #
    def __str__(self):
        return self.name

确实如此：

Traceback (most recent call last):
  ...
AttributeError: <enum 'Enum'> cannot set attribute 'name'

我认为您所说的是以下内容将起作用：

class Example(Enum):
    name = 'a name member'
    value = 'a value member'

使用中：

>>> list(Example)
[<Example.name: 'a name member'>, <Example.value: 'a value member'>]

>>> Example.name.name
'name'

如果你不这么说，那你就错了。 :-/

就 PEP 20 而言，逗号是明确的。