Question

我观察到我可以将某些整数值转换为它们的标志分解类型，并且我可以将标志字符串名称转换为标志，但我还没有想出是否有转换字符串联合标志的方法名称为enum.Flag。

import enum
flag = enum.Flag('flg' , ['a', 'b', 'c', 'd'] )

# Valid 1:1 conversions...
>>> flag["a"]
<flg.a: 1>
>>> flag["b"]
<flg.b: 2>
>>> flag(1)
<flg.a: 1>
>>> flag(2)
<flg.b: 2>

# Valid union int conversion
>>> flag(7)
<flg.c|b|a: 7>
>>> flag(15) # largest integer flag can represent (1+2+4+8)
<flg.d|c|b|a: 15>

# Out of bounds
>>> flag(16)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Anaconda3\lib\enum.py", line 291, in __call__
    return cls.__new__(cls, value)
  File "C:\Anaconda3\lib\enum.py", line 533, in __new__
    return cls._missing_(value)
  File "C:\Anaconda3\lib\enum.py", line 673, in _missing_
    possible_member = cls._create_pseudo_member_(value)
  File "C:\Anaconda3\lib\enum.py", line 688, in _create_pseudo_member_
    raise ValueError("%r is not a valid %s" % (value, cls.__name__))
ValueError: 16 is not a valid flg

我希望我能够以精神的方式拉开一些东西 flag["d|c|b|a"]，但作为字典，此转换仅适用于不同的标志成员名称，而不适用于它们的任何聚合。同时我正在手动进行拆分，转换，联合合并，但我很好奇是否有像int转换案例那样的更直接的路径。

编辑：查看enum.py的来源，看来正确的术语不是联合，而是复合

# enum.py
def _create_pseudo_member_(cls, value):
    """
    Create a composite member iff value contains only members.
    """

Answer 1

您可以继承enum.Flag并拦截其_missing_()方法：

from functools import reduce

import enum


class UnionFlag(enum.Flag):

    @classmethod
    def _missing_(cls, value):
        if isinstance(value, str):
            return reduce(cls.__or__, (cls[s] for s in value.split('|')))
        else:
            return super()._missing_(value)

>>> flag = UnionFlag('uflg', ['a', 'b', 'c', 'd'])
>>> flag.a
<uflg.a: 1>
>>> flag(3)
<uflg.b|a: 3>
>>> flag['c']
<uflg.c: 4>
>>> flag('a|b|d')
<uflg.d|b|a: 11>

有没有办法将表示多个标志成员的字符串转换为复合枚举.Flag？

1 个答案: