Python上下文管理器

坚持是一种品质 2019-11-09

#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!

相关推荐