python - 根据名称和选项制作一个具有自动值的(类型化)python 枚举

Question

在 python 中，您可以定义具有自动值的类型化枚举:

import enum
from enum import auto

class Ordinals(enum.IntEnum):
  FIRST = auto()
  SECOND = auto()
  THIRD = auto()

Ordinals.FIRST == 1 #=> True

您还可以定义带参数的类型化枚举:(来自 docs 的示例):

class Coordinate(bytes, Enum):
     """
     Coordinate with binary codes that can be indexed by the int code.
     """
     def __new__(cls, value, label, unit):
         obj = bytes.__new__(cls, [value])
         obj._value_ = value
         obj.label = label
         obj.unit = unit
         return obj
     PX = (0, 'P.X', 'km')
     PY = (1, 'P.Y', 'km')
     VX = (2, 'V.X', 'km/s')
     VY = (3, 'V.Y', 'km/s')

Coordinate.PX._name_ #=> 'PX'
Coordinate.PX == bytes([0]) #=> True
Coordinate.PX.label #=> 'P.X'

此外，您可以定义一个通过名称查找值的枚举:

class LetterOrdinals(enum.IntEnum):
  def _generate_next_value_(name, *_):
    return ord(name)
  A = auto()
  B = auto()

LetterOrdinals.A == 65 #=> True

现在假设我想将所有这三个结合起来，并有一个类型化的枚举，它根据名称生成一个值并允许为 __new__/__init__ 提供参数。这可能吗？我遇到的第一个问题是 auto 不再对值起作用。似乎如果 auto 不是整个值，那么它会跳过生成值。相反，如果它不生成值，则该名称不可用于 __new__ 或 __init__。这是一个不起作用的例子(我已经尝试过这个变体):

class Coordinate(bytes, Enum):
     def _generate_next_value_(name, *_):
         return [ord(letter) for letter in value]
     def __new__(cls, value, label, unit):
         obj = bytes.__new__(cls, [*value])
         obj._value_ = value
         obj.label = label
         obj.unit = unit
         return obj
     PX = (auto(), 'P.X', 'km')
     PY = (auto(), 'P.Y', 'km')
     VX = (auto(), 'V.X', 'km/s')
     VY = ([2,2], 'V.Y', 'km/s')
# never generates the auto value, causes errors

那么有什么好的解决办法吗？

Answer 1

您可以在类中编写自己的项目定义方法，并让它返回由 auto() 创建的对象。 Enum 类对 enum.auto 对象进行了一些特殊处理，因此您需要暂时将其他属性存储在某个地方，直到调用 __init__() 并检索它们以进行实际实例初始化:

from enum import Enum,auto
class Coordinate: args = [] # placeholder for auto/init arguments
class Coordinate(Enum):

     def _generate_next_value_(name, *_):
        # Here, you can use label,unit = Coordinate.args[-1] 
        # if your value is based on other arguments
        return ord(name[0])*100+ord(name[1])

     def __init__(obj,value):
         obj.label,obj.unit = Coordinate.args.pop(0)

     def item(*args):
         Coordinate.args.append(args)
         return auto()
        
     PX = item('P.X', 'km')
     PY = item('P.Y', 'km')
     VX = item('V.X', 'km/s')
     VY = item('V.Y', 'km/s')

输出:

print(Coordinate['VY'])        # Coordinate.PY
print(Coordinate['VY'].value)  # 8689
print(Coordinate['VY'].label)  # P.Y
print(Coordinate['VY'].unit)   # km
print(Coordinate(8689))        # Coordinate.PY
print(Coordinate(8689).name)   # PY