如何实现支持位运算的枚举以检测文档中欧盟国家提及情况？

直接用Python标准库的IntEnum就能解决你的问题，它继承自int，让枚举成员可以直接和整数做位运算，完美规避类型错误。

步骤1：定义枚举 #

给每个欧盟国家分配一个2的幂次（位掩码），确保每个国家对应独立的二进制位：

from enum import IntEnum

class Country(IntEnum):
    Austria = 1 << 0   # 二进制0001，对应十进制1
    Belgium = 1 << 1   # 二进制0010，对应十进制2
    Bulgaria = 1 << 2  # 二进制0100，对应十进制4
    Croatia = 1 << 3   # 二进制1000，对应十进制8
    # 剩余23个欧盟国家依次按1<<4、1<<5...1<<26的规则添加

步骤2：位字段检测示例 #

比如你提到的文档同时提及Austria和Bulgaria，位字段值为1 | 4 = 5，现在判断是否包含Bulgaria：

doc_bitfield = 5
# 直接通过位运算判断，非零值在布尔逻辑中等价于True
if doc_bitfield & Country.Bulgaria:
    print("文档提到了保加利亚")
# 也可以显式转为布尔值
if bool(doc_bitfield & Country.Bulgaria):
    print("文档提到了保加利亚")

这里5 & Country.Bulgaria会得到4（非零），完全符合你期望的判断逻辑。

为什么原来的枚举报错？ #

普通Enum类创建的成员是Enum类型实例，并非整数，所以无法直接和int做位运算。而IntEnum的成员本质是整数，既保留了枚举的可读性，又支持所有整数运算。

备选方案：常量字典 #

如果不想用枚举，也可以直接定义常量字典，虽然可读性稍弱但更轻量：

COUNTRY_MASK = {
    "Austria": 1 << 0,
    "Belgium": 1 << 1,
    "Bulgaria": 1 << 2,
    "Croatia": 1 << 3,
}

doc_bitfield = 5
if doc_bitfield & COUNTRY_MASK["Bulgaria"]:
    print("文档提到了保加利亚")

但枚举方案更适合长期维护，新增国家时直接在枚举类里添加成员即可，不容易出错。

内容的提问来源于stack exchange，提问作者dude