python中open()方法既能直接返回也能通过with语句当作上下文管理器使用是怎么做到的?

如题。简单看了下io.py部分的源码,只看到了open的定义是直接返回对象,没有看到是如何实现上下文管理器的。google了半天也没有结果。求知乎大神解答!回复内容:
前段时间果壳 python 开发面试被问到了这个问题实现某个对象可以用 with 来管理,只需要改写 __enter__ 和 __exit__ 这两个 magic method 即可另外你说你在 io.py 源码里没找到,大哥读代码要仔细啊io.py 里的 io 函数都是从 _pyio.py 里 import 进来的,然后在 _pyio.py 的第 140 行 到 147 行有这么一段注释open() returns a file object whose type depends on the mode, andthrough which the standard file operations such as reading and writingare performed. when open() is used to open a file in a text mode (‘w’,’r’, ‘wt’, ‘rt’, etc.), it returns a textiowrapper. when used to opena file in a binary mode, the returned class varies: in read binarymode, it returns a bufferedreader; in write binary and append binarymodes, it returns a bufferedwriter, and in read/write mode, it returnsa bufferedrandom.所以 open() 这个函数在不同模式下会返回不同的对象,包括 textiowrapper/bufferedreader/bufferedwriter 这三种继续看上面这三个类的定义,发现这几个类还继承了其他的几个类,这里我就不领着你看了,直接用 visio 画了个类层次示意图,如下其中蓝色矩形表示 class,蓝色连线表示继承关系,蓝色连线上的数字表示这个继承关系的代码在源码中的哪一行可以看到 open() 函数所返回的几个 class,最终都继承于 iobase 这个 class,而在这个 class 中,就实现了 __enter__() 和 __exit() 两个 magic method,具体代码位于 _pyio.py 的第 428 行到 437 行

### context manager ###
def __enter__(self):
“””context management protocol. returns self.”””
self._checkclosed()
return self
def __exit__(self, *args):
“””context management protocol. calls close()”””
self.close()

只要有__enter__和__exit__方法,就能用在with语句里,open返回的对象有这两个方法,就可以了,不必在open方法内做什么手脚。
python 的上线文管理器使用with关键字,用来定义with后面这个缩进级别,进入和跳出的行为,是语法糖的一种。open是一个调用用于实例化file类的函数,而file类只不过是一个拥有__enter__和__exit__(这种前后都有两个下划线的在python中叫做“魔法方法”,除了这两个还有很多,有兴趣可以去查一下)的类,用在with中时,程序会自动在进出这个程序块时调用这两个函数,如果没用with就要手动管理,自己调用close()。如果想要自己实现类似的功能,用回调就,或者单独封装成装饰器(装饰器其实也是python的一种语法糖)都可以。
跟java的closeable一样 定义好释放资源的方式 通过语法糖隐藏部分重复代码
你google context manager就有结果了

Posted in 未分类

发表评论