Python自定义context manager嵌套使用报AttributeError如何解决

• Python 上下文管理器的__enter__方法执行完成后，返回值会自动赋值给with语句中as关键字后的变量。当方法无显式返回值时，默认返回None。你当前实现的__enter__没有返回实例自身，所以第一层with mycls() as cl执行后，cl被赋值为None，调用None.little()就会触发对应属性报错。
• 如果你已经给第一层的__enter__加了返回逻辑但嵌套使用时仍报错，是因为嵌套的with cl会再次触发实例的__enter__方法，若未处理嵌套场景，也可能出现返回None的问题。

基础修复只需要在__enter__中显式返回实例自身即可解决报错，如果需要适配你代码中的多层嵌套使用场景，建议补充进入计数器，避免多次调用__exit__时提前清理资源，参考代码如下：

class mycls:
    def __init__(self):
        # 嵌套场景用的进入计数器，非嵌套可省略
        self._enter_count = 0

    def __enter__(self):
        self._enter_count += 1
        # 核心修复：返回实例自身，赋值给as后的变量
        return self

    def little(self):
        # 你的业务逻辑
        ...

    def __exit__(self, exc_type, exc_val, exc_tb):
        self._enter_count -= 1
        # 只有计数器归零才执行实际清理逻辑，避免嵌套时提前释放资源
        if self._enter_count == 0:
            # 你的资源清理逻辑
            ...

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