Python-with

在 Python 中,使用 with 语句可以自动调用一个对象的上下文管理器(Context Manager),来简化异常处理和资源管理等工作。实现 with 语句的关键是定义上下文管理器对象,它应该包括 enter() 和 exit() 方法。

enter() 方法会在进入 with 代码块时被调用,而 exit() 方法则会在 with 代码块执行结束后被调用。例如,用于打开和关闭文件的上下文管理器可以实现如下:

class File:
    def __init__(self, filename):
        self.filename = filename
        
    def __enter__(self):
        self.file = open(self.filename, 'r')
        return self.file
        
    def __exit__(self, exc_type, exc_val, exc_tb):
        self.file.close()

在上述代码中,File 类定义了一个上下文管理器,将打开文件操作放在 __enter__() 方法中,并返回文件对象。当 with 语句块结束时,__exit__() 方法会自动关闭文件。

下面展示如何使用 with 语句使用上述上下文管理器:

with File('example.txt') as file:
    data = file.read()
    print(data)

上述代码会自动打开文件、读取数据、输出数据并关闭文件,即使出现异常也能保证文件正确地被关闭,从而减轻了代码的负担。

posted @ 2023-05-13 14:54  玄谷  阅读(40)  评论(0)    收藏  举报