Python如何优雅将类实例重置为__init__执行后的初始状态

问题背景 #

现有基于ParentClass派生的约150个子类，子类分两种实现形态：

• 直接在__init__中接收参数完成实例初始化
• 额外实现静态build方法做参数预处理，再调用类构造方法返回实例

业务要求支持配置开关，在特定子类的run方法执行前，将实例重置为__init__刚执行完成的状态，同时满足两个硬约束：

• 调用run的上层逻辑无法获取实例初始化时的入参，不能通过重新传参创建新实例实现
• 部分子类通过build方法构造的耗时可达0.5s，重复创建实例性能无法接受

此前尝试的装饰器方案存在无法精准识别__init__设置属性的问题，需要更优的实现。
附问题复现示例：

class ParentClass:
    some = "parameters"
    here = None
    
    def run(self, **kwargs):
        pass 

class SubClass(ParentClass):
    def __init__(self, param1: int, param2: str):
        self.param1 = param1
        self.param2 = param2

    def run(self, params: dict) -> dict:
        self.param1 += 1
        return params  

instance = SubClass(5, "some_string")
# 后续逻辑不允许重新创建SubClass实例
instance.run({})
print(instance.param1) # 输出6，符合预期
instance.run({})
print(instance.param1) # 输出7，预期应该重置为初始值5，执行后为6

实现方案 #

最优雅的零侵入实现是在父类层面统一拦截子类初始化流程，自动保存初始化完成瞬间的实例状态快照，重置时直接从快照恢复，完全不需要修改任何子类业务代码，性能远高于重建实例。

核心设计逻辑 #

• 利用Python类初始化钩子，自动包装所有子类的__init__方法，在__init__原生逻辑执行完成的瞬间，对实例自身的所有属性做深拷贝，作为初始状态快照存为实例私有属性
• 在父类的run方法层面统一加拦截逻辑，通过类属性开关控制是否需要在run执行前重置实例状态
• 重置操作直接用快照的深拷贝覆盖实例当前属性字典，全程微秒级耗时，无重建实例的性能开销
• 自动兼容同步/异步run方法，兼容两种子类实现形式

完整实现代码 #

import copy
import functools
import inspect

class ParentClass:
    some = "parameters"
    here = None
    # 重置开关：子类需要开启自动重置时，直接覆盖这个类属性为True即可
    AUTO_RESET_BEFORE_RUN: bool = False

    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        # 包装子类的__init__方法，初始化完成后自动存快照
        original_init = cls.__init__
        @functools.wraps(original_init)
        def wrapped_init(self, *init_args, **init_kwargs):
            original_init(self, *init_args, **init_kwargs)
            # __init__执行完立刻存初始状态深拷贝快照
            self._initial_state_snapshot = copy.deepcopy(self.__dict__)
        cls.__init__ = wrapped_init

        # 包装子类的run方法，执行前按开关判断是否重置
        original_run = cls.run
        if inspect.iscoroutinefunction(original_run):
            @functools.wraps(original_run)
            async def wrapped_run(self, *run_args, **run_kwargs):
                if self.AUTO_RESET_BEFORE_RUN:
                    # 用快照的深拷贝覆盖当前实例属性，避免修改快照本身
                    self.__dict__.update(copy.deepcopy(self._initial_state_snapshot))
                return await original_run(self, *run_args, **run_kwargs)
        else:
            @functools.wraps(original_run)
            def wrapped_run(self, *run_args, **run_kwargs):
                if self.AUTO_RESET_BEFORE_RUN:
                    self.__dict__.update(copy.deepcopy(self._initial_state_snapshot))
                return original_run(self, *run_args, **run_kwargs)
        cls.run = wrapped_run

    async def run(self, **kwargs):
        pass

使用方式 #

1. 所有原有子类代码不需要做任何修改，逻辑完全兼容
2. 需要开启run前自动重置的子类，只要在类定义里把AUTO_RESET_BEFORE_RUN设为True即可

# 示例：开启自动重置的子类
class SubClass(ParentClass):
    AUTO_RESET_BEFORE_RUN = True
    def __init__(self, param1: int, param2: str):
        self.param1 = param1
        self.param2 = param2

    def run(self, params: dict) -> dict:
        self.param1 += 1
        return params  

instance = SubClass(5, "some_string")
instance.run({})
print(instance.param1) # 输出6
instance.run({})
print(instance.param1) # 输出6，符合重置预期

方案优势 #

• 零侵入：150个子类不需要修改任何业务逻辑，所有能力在父类统一实现
• 高性能：快照仅在实例首次初始化时生成一次，每次重置是微秒级的深拷贝操作，相比0.5s的实例重建性能提升上千倍
• 精准：快照完全对应__init__执行完成瞬间的实例状态，不会漏存、多存属性，不需要手动识别初始化阶段设置的参数
• 兼容性好：自动适配同步/异步run方法，兼容直接初始化、build方法构造实例两种子类形态
• 配置灵活：仅需要给指定子类加一个类属性开关即可开启重置能力，维护成本极低

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