修改 enum.IntFlag __repr__

Question

我编写了一个小实用程序来解码由位标志组成的日志输出。我在解码这些位标志时使用枚举。

python --version
Python 3.8.6

为简洁起见，我只想打印枚举成员名称。我已将我的问题提炼为这个简短的 sn-p：

from enum import IntFlag
class Foo(IntFlag):
    A = 0x01
    B = 0x02
    C = 0x04
    def __str__(self):
        return self.name
    __repr__ = __str__

如果我将它们单独显示或将它们作为列表的一部分打印，则效果很好。但我无法打印 or 表达式。我想在A|B|C 表单上显示标志。

>>> Foo.A
A
>>> Foo.B
B
>>> [Foo.A, Foo.B]
[A, C]
>>> Foo.A | Foo.B
TypeError: __repr__ returned non-string (type NoneType)

更新：我查看了错误的源代码版本。 3.8.6 标签的IntFlag::__repr__ 确实包含打印每个成员https://github.com/python/cpython/blob/v3.8.6/Lib/enum.py#L761 的代码

我尝试阅读 CPythons Lib/enum.py，但无法弄清楚默认 IntFlag（没有我覆盖的 __repr__）如何在 Foo.A|Foo.B 表单上打印输出。~~

我可以看到IntFlag.__or__ 通过调用__class__ 返回了一个新对象（我真的不知道__class__ 做了什么）。而Enum.__repr__ 只打印一个类名和枚举名。

如何编写一个 __repr__ 函数和一个返回当前设置的枚举成员名称的字符串表示形式的 __str__ 函数？

Answer 1

from enum import Flag, _power_of_two

class BetterFlag(Flag):
    def __repr__(self):
        if self.value == 0:
            return "%s(0)" % self.__class__.__name__
        return '|'.join(
                m.name
                for m in self.__class__
                if m.value & self.value and _power_of_two(m.value)
                )
    __str__ = __repr__

Python 3.10 及更早版本通过测试name 属性是否为None 来执行此操作，如果是则循环遍历内容。对于更奇特的枚举，循环存在缺陷，已在 3.11 中重做。

注意：以上是向前和向后兼容的。让您的 Flags 继承自 BetterFlag。

Answer 2

我想出了这个解决方案

class Foo(IntFlag):
    A = 0x01
    B = 0x02
    C = 0x04
    def __repr__(self):
        return '|'.join(val.name for val in Foo if self.value & val)

以及repl中的一个演示：

>>> Foo.A | Foo.C | Foo.B
A|B|C
>>> Foo.A
A