Python自定义类支持with关键字

2016-08-16 董付国 Python小屋 Python小屋

在任何语言中,文件操作都要遵循打开文件-操作文件-关闭文件,数据库操作也要遵循打开数据库连接-操作数据库-关闭数据库连接,网络编程中很多也具有同样的约定。下面的介绍以文件操作为例。

尽管都知道要关闭文件,但是经常会有程序员忘记关闭文件,或者写了关闭文件的代码却没有执行到,例如关闭文件文件之前的代码出现了异常

Python提供了关键字with用来防止这样的情况,不论代码是否出现异常,都能保证文件被正确关闭,例如:

with open('test.txt') as fp:

    print(fp.read())

那么这个with关键字是如何实现的呢?关键在于两个特殊方法__enter__()和__exit__(),实现了这两个特殊方法的类支持with关键字。例如:

class myOpen:

    def __init__(self, fileName, mode='r'):

        self.fp = open(fileName, mode)


    def __enter__(self):

        return self.fp


    def __exit__(self, exceptionType, exceptionVal, trace):

        self.fp.close()


with myOpen('test.txt') as fp:

    print(fp.read())

这段代码的效果与第一段代码相同。


温馨提示:单击文章顶部作者名字旁边浅蓝色的“Python小屋”进入公众号,关注后可以查看更多内容!


欢迎转发给您的朋友,或许这正是Ta需要的知识!