在 Python 中，with 语句用于简化资源管理，它保证在执行完代码块后正确地释放资源，比如文件、网络连接等。这是通过特定的魔术方法（特殊方法）实现的。自定义 with 语句涉及两个关键的魔术方法：
魔术方法
-
__enter__：当进入with语句块时会调用这个方法。它负责初始化需要的资源，并且可以返回一个值，这个值会赋给as后面的变量。 -
__exit__：当with语句块执行完成后（无论是否发生异常），会调用这个方法。它负责清理或释放资源，比如关闭文件、断开数据库连接等。它接受三个参数，分别是异常类型、异常值和异常追踪信息，如果异常被成功处理，它需要返回True来阻止异常向上传播。
原理
当使用 with 语句时，Python 会先调用上下文管理器对象的 __enter__ 方法进入上下文，然后执行 with 语句块中的代码，执行完成后，无论是否出现异常，都会调用 __exit__ 方法。
示例
我们来看一个自定义 with 语句的例子，创建一个简单的类来管理文件读写：
class FileManager:def __init__(self, filename, mode):self.filename = filenameself.mode = modedef __enter__(self):# 打开文件，并返回文件对象self.file = open(self.filename, self.mode)return self.filedef __exit__(self, exc_type, exc_val, exc_tb):# 关闭文件，确保资源释放if self.file:self.file.close()# 使用自定义的 with 语句
with FileManager('example.txt', 'w') as file:file.write('Hello, world!')
在这个例子中，FileManager 类实现了 __enter__ 和 __exit__ 方法，with 语句会自动调用这两个方法来管理文件的打开和关闭。
魔术方法解析
-
__enter__：在with FileManager(...)执行时，调用__enter__，该方法打开文件并返回文件对象。这个对象会被赋值给as后面的file变量。 -
__exit__：当with语句块结束时（无论正常结束还是出现异常），__exit__会被调用，它负责关闭文件，即使在写入过程中出现异常，也能确保资源被正确释放。
__exit__ 处理异常的示例
如果你想让 __exit__ 方法处理异常并阻止异常向外传播，你可以在 __exit__ 方法中返回 True：
class FileManager:def __enter__(self):self.file = open('example.txt', 'w')return self.filedef __exit__(self, exc_type, exc_val, exc_tb):if exc_type is not None:print(f"Exception: {exc_val}")self.file.close()# 返回 True 来阻止异常继续传播return Truewith FileManager() as file:file.write("Hello, world!")raise ValueError("An error occurred!") # 故意引发一个错误print("This message will still be printed!")
在这个例子中，虽然 with 语句块中引发了 ValueError 异常，但 __exit__ 方法捕获了该异常并返回 True，因此程序不会崩溃，错误信息也不会进一步传播。
总结
__enter__：进入上下文时调用，通常负责初始化资源。__exit__：退出上下文时调用，负责清理资源，可以处理异常。with语句提供了更优雅的资源管理方式，避免手动管理资源的麻烦，特别是在异常处理方面。
