Python上下文管理器

Posted 2020-11-30

#with上下文管理器
#语法：with ... as ... 如：with open(‘test.txt‘, ‘r‘) as fp,打开一个文件作为文件句柄对象赋值给fp
#with是一个语句块，上下文管理器中里面实现了两个方法:enter, exit,enter是进入代码块前自动调用的方法，exit是
#退出with语句块时调用的，例如，文件对象在退出上下文管理器时调用了exit中的self.close()语句，所以就不用手动去关闭文件
#只要重写了enter方法和exit两个方法，就可以自定义一个with上下文管理器

class SayHello(object):
    def __init__(self, text):
        self.text = text
    def __enter__(self):
        return self #必须返回对象
    def __exit__(self, exc_type, exc_value, traceback):
        self.text = "Bye!"

with SayHello("Hello") as greet:
    print(greet.text)
print(greet.text)

输出：
Hello
Bye!