有没有办法将表示多个标志成员的字符串转换为组合enum.Flag?

有没有办法将表示多个标志成员的字符串转换为组合enum.Flag?

问题描述:

我观察到我可以将某些整数值转换为它们的标志分解类型,并且我可以将标志字符串名称转换为标志,但是我还没有弄清楚是否有方法将标志名称的字符串联合到enum.Flag有没有办法将表示多个标志成员的字符串转换为组合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. 
    """ 

你可以继承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>