with: __enter__ and __exit__
要使我们的对象支持with语句,必须为它实现 __enter__
和__exit__
方法。
先看一个例子:
from socket import socket, AF_INET, type=SOCK_STREAM class lazyConnection: def __init__(self, address, family=AF_INET, type=SOCK_STREAM): self.address = address self.family = AF_INET self.type = SOCK_STREAM self.sock = None def __enter__(self): if self.sock is not None: raise RuntimeError('already connected') self.sock = socket(self.family, self.type) self.sock.connect(self.address) return self.sock def __exit__(self, exc_ty, exc_val, tb): self.sock.close() self.sock = None
例子中的类是一个网络连接类,但是他在初始化的时候 并没有真正建立连接,而是在使用with语句的时候才 建立连接,所以是惰性lazy的,eg
from functools import partial conn = LazyConnection( ('www.python.org', 80) ) with conn as s: # conn.__enter__() exec, connection open s.send(b'GET /index.html HTTP/1.0\r\n') s.send('Host: www.python.org\r\n') s.send('\r\n') resp = b''.join(iter(partial(s.recv, 8192), b'')) # conn.__exit__() exec, connection close
使用with语句的时候,会触发__enter__
函数, 函数的返回值会赋值给with as后面的标识符。 然后就会执行with的body,完成后触发__exit__
函数
实际中使用with更多的情况是操作一些资源如文件, 网络连接,锁等。因为这些资源要显式的打开和关闭 ,如果忘记关闭容易引起问题,所以这种情况使用with 比较好。