Python自定义类支持with关键字

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

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

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())

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