Python3.13 Enum实例循环引用的手动序列化解决方案咨询

在Python 3.13项目中，需要手动序列化Enum创建的对象（比如MyEnum.VALUE），且不能使用marshal和pickle。但Enum对象存在多处循环引用，导致序列化受阻。已知可以通过字符串特殊处理实现序列化，想知道有没有其他方案，比如修改MyEnum.VALUE对象数据来移除循环引用？

代码示例：

from enum import Enum, auto


class MyEnum(Enum):
    VALUE = auto()


def build_object_tree(obj, seen=None):
    if seen is None:
        seen = set()

    obj_id = id(obj)
    if obj_id in seen:
        return f"<CircularRef to id={obj_id}>"

    seen.add(obj_id)

    if isinstance(obj, (int, float, str, bool, type(None))):
        return obj

    if isinstance(obj, (list, tuple, set)):
        return type(obj)(build_object_tree(item, seen) for item in obj)

    if isinstance(obj, dict):
        return {build_object_tree(k, seen): build_object_tree(v, seen) for k, v in obj.items()}

    if hasattr(obj, "__dict__"):
        return {
            "__class__": obj.__class__.__name__,
            "id": obj_id,
            "attributes": {k: build_object_tree(v, seen) for k, v in obj.__dict__.items()}
        }

    return repr(obj)


print(build_object_tree(MyEnum.VALUE))

输出结果（含循环引用标记）：

{'__class__': 'MyEnum', 'id': 140330736074240,
 'attributes':
 {'_value_':
    1, '_name_': 'VALUE',
    '__objclass__':
    {'__class__': 'EnumType', 'id': 784520928,
     'attributes':
      {'_generate_next_value_':
          {'__class__': 'staticmethod', 'id': 140330735054144,
           'attributes':
           {'__module__': 'enum',
            '__name__':
            '_generate_next_value_',
            '__qualname__': 'Enum._generate_next_value_',
            '__doc__': '\nGenerate the next value when not given.\n\nname: the name of the member\nstart: the initial start value or None\ncount: the number of existing members\nlast_values: the list of values assigned\n', '__annotations__': {}}
          },
       '__module__': '__main__',
       '__firstlineno__': 4,
       '__static_attributes__': (),
       '_new_member_': '<built-in method __new__ of type object at 0xa4ff40>',
       '_use_args_': False,
       '_member_names_': ['<CircularRef to id=140330734841328>'],
       '_member_map_':
          {'<CircularRef to id=140330734841328>': '<CircularRef to id=140330736074240>'},
       '_value2member_map_':
          {'<CircularRef to id=11053296>': '<CircularRef to id=140330736074240>'},
       '_unhashable_values_': [],
       '_unhashable_values_map_': {},
       '_member_type_':
          {'__class__': 'type', 'id': 10813248,
           'attributes':
              {'__new__': '<CircularRef to id=140330735861824>',
               '__repr__': "<slot wrapper '__repr__' of 'object' objects>",
               '__hash__': "<slot wrapper '__hash__' of 'object' objects>",
               '__str__': "<slot wrapper '__str__' of 'object' objects>",
               '__getattribute__': "<slot wrapper '__getattribute__' of 'object' objects>",
               '__setattr__': "<slot wrapper '__setattr__' of 'object' objects>",
               '__delattr__': "<slot wrapper '__delattr__' of 'object' objects>",
               '__lt__': "<slot wrapper '__lt__' of 'object' objects>",
               '__le__': "<slot wrapper '__le__' of 'object' objects>",
               '__eq__': "<slot wrapper '__eq__' of 'object' objects>",
               '__ne__': "<slot wrapper '__ne__' of 'object' objects>",
               '__gt__': "<slot wrapper '__gt__' of 'object' objects>",
               '__ge__': "<slot wrapper '__ge__' of 'object' objects>",
               '__init__': "<slot wrapper '__init__' of 'object' objects>",
               '__reduce_ex__': "<method '__reduce_ex__' of 'object' objects>",
               '__reduce__': "<method '__reduce__' of 'object' objects>",
               '__getstate__': "<method '__getstate__' of 'object' objects>",
               '__subclasshook__': "<method '__subclasshook__' of 'object' objects>",
               '__init_subclass__': "<method '__init_subclass__' of 'object' objects>",
               '__format__': "<method '__format__' of 'object' objects>",
               '__sizeof__': "<method '__sizeof__' of 'object' objects>",
               '__dir__': "<method '__dir__' of 'object' objects>",
               '__class__': "<attribute '__class__' of 'object' objects>",
               '__doc__': 'The base class of the class hierarchy.\n\nWhen called, it accepts no arguments and returns a new featureless\ninstance that has no instance attributes and cannot be given any.\n'}
          },
       '_value_repr_': None,
       '__doc__': '<CircularRef to id=10814656>',
       'VALUE': '<CircularRef to id=140330736074240>',
       '__new__':
          {'__class__': 'function', 'id': 140330735209344,
           'attributes': {}
          }
      }
    },
    '_sort_order_': 0
 }
}

可行解决方案 #

方案1：针对Enum类型做专属序列化逻辑（最推荐） #

不用遍历整个对象树，直接提取Enum成员的核心信息——枚举类名、成员名称、成员值，这些信息足够反序列化时重建对象。修改build_object_tree函数，增加对Enum实例的判断：

from enum import Enum, auto, EnumMeta

class MyEnum(Enum):
    VALUE = auto()

def build_object_tree(obj, seen=None):
    if seen is None:
        seen = set()

    obj_id = id(obj)
    if obj_id in seen:
        return f"<CircularRef to id={obj_id}>"

    seen.add(obj_id)

    # 处理Enum实例
    if isinstance(obj, Enum):
        return {
            "__type__": "EnumMember",
            "enum_class": obj.__class__.__name__,
            "member_name": obj.name,
            "member_value": obj.value
        }
    
    # 处理Enum类本身
    if isinstance(obj, EnumMeta):
        return {
            "__type__": "EnumClass",
            "class_name": obj.__name__,
            "module": obj.__module__,
            "members": {name: build_object_tree(member, seen) for name, member in obj.__members__.items()}
        }

    if isinstance(obj, (int, float, str, bool, type(None))):
        return obj

    if isinstance(obj, (list, tuple, set)):
        return type(obj)(build_object_tree(item, seen) for item in obj)

    if isinstance(obj, dict):
        return {build_object_tree(k, seen): build_object_tree(v, seen) for k, v in obj.items()}

    if hasattr(obj, "__dict__"):
        return {
            "__class__": obj.__class__.__name__,
            "id": obj_id,
            "attributes": {k: build_object_tree(v, seen) for k, v in obj.__dict__.items()}
        }

    return repr(obj)

print(build_object_tree(MyEnum.VALUE))

输出会变成简洁无循环的结构：

{
    "__type__": "EnumMember",
    "enum_class": "MyEnum",
    "member_name": "VALUE",
    "member_value": 1
}

反序列化时，只需通过getattr(importlib.import_module(module), enum_class)[member_name]就能重建Enum对象。

方案2：过滤循环引用相关属性（不修改原对象） #

如果一定要遍历对象树，就在处理Enum实例和Enum类时，主动过滤掉会引发循环引用的属性，比如__objclass__、_member_map_、_value2member_map_这些：

from enum import Enum, auto

class MyEnum(Enum):
    VALUE = auto()

def build_object_tree(obj, seen=None):
    if seen is None:
        seen = set()

    obj_id = id(obj)
    if obj_id in seen:
        return f"<CircularRef to id={obj_id}>"

    seen.add(obj_id)

    if isinstance(obj, Enum):
        # 只保留核心属性，跳过指向枚举类的关联属性
        return {
            "__class__": obj.__class__.__name__,
            "id": obj_id,
            "attributes": {
                "_name_": obj._name_,
                "_value_": obj._value_,
                "_sort_order_": obj._sort_order_
            }
        }
    
    # 处理枚举类时过滤循环引用属性
    if hasattr(obj, "__class__") and obj.__class__.__name__ == "EnumType":
        filtered_attrs = {}
        for k, v in obj.__dict__.items():
            # 跳过会引用回成员的属性
            if k in ("_member_map_", "_value2member_map_", "__objclass__", "VALUE"):
                continue
            filtered_attrs[k] = build_object_tree(v, seen)
        return {
            "__class__": "EnumType",
            "id": obj_id,
            "attributes": filtered_attrs
        }

    if isinstance(obj, (int, float, str, bool, type(None))):
        return obj

    if isinstance(obj, (list, tuple, set)):
        return type(obj)(build_object_tree(item, seen) for item in obj)

    if isinstance(obj, dict):
        return {build_object_tree(k, seen): build_object_tree(v, seen) for k, v in obj.items()}

    if hasattr(obj, "__dict__"):
        return {
            "__class__": obj.__class__.__name__,
            "id": obj_id,
            "attributes": {k: build_object_tree(v, seen) for k, v in obj.__dict__.items()}
        }

    return repr(obj)

print(build_object_tree(MyEnum.VALUE))

这种方式不会修改原Enum对象，只是在序列化时选择性忽略引发循环的属性，输出结果会干净很多，没有循环引用标记。

方案3：不要修改Enum对象（不推荐） #

修改MyEnum.VALUE的内部数据来移除循环引用是不推荐的：

• Enum对象的内部结构由Python标准库维护，修改它可能导致枚举类功能异常，比如成员查找、比较等操作失效。
• 枚举类和成员之间的引用是设计上的必要关联，强行移除会破坏Enum的语义。

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