117笔记问答在Python中，上下文管理器通过使用with语句来确保在执行代码块时资源的正确分配和释放。它们通过实现两个特殊的方法来确保一致性：__enter__() 和 __exit__()。
-
__enter__()方法：当进入with语句块时，这个方法会被调用。它的返回值会被赋给as子句中的变量。如果这个方法抛出异常，with语句会立即退出，并执行__exit__()方法。 -
__exit__()方法：当with语句块中的代码执行完毕（或抛出异常）时，这个方法会被调用。它负责清理资源，例如关闭文件、释放锁等。如果__exit__()方法返回True，则异常会被忽略；如果返回False或抛出异常，异常会继续向上抛出。
下面是一个简单的上下文管理器的例子，用于确保文件在使用后被正确关闭：
class FileHandler:
def __init__(self, filename, mode):
self.filename = filename
self.mode = mode
self.file = None
def __enter__(self):
try:
self.file = open(self.filename, self.mode)
return self.file
except IOError as e:
print(f"Error opening file: {e}")
raise
def __exit__(self, exc_type, exc_value, traceback):
if self.file:
self.file.close()
if exc_type:
print(f"Exception type: {exc_type}")
print(f"Exception value: {exc_value}")
return True # 返回True以忽略异常，或返回False以重新抛出异常
# 使用上下文管理器
with FileHandler("example.txt", "w") as file:
file.write("Hello, World!")
在这个例子中，FileHandler类实现了上下文管理器，确保在使用文件后将其关闭。如果在打开文件时发生错误，__enter__()方法会抛出异常，with语句会立即退出并执行__exit__()方法。