8.3 通过with语句出发,让对象支持上下文管理协议
>>> from socket import socket,AF_INET,SOCK_STREAM
>>> class LC:
... def
... 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 = sock(self.family,self.type)
... self.sock.connect(self.address)
... return self.sock
... def __exit__(self,exc_ty,exc_val,tb):
... self.sock.close()
... self.sok = None
...
- 这个类的核心功能就是建立一条网络连接,但是在初始状态下,并不会建立连接
- 网络连接可以通过with语句来建立和close,示例如下
>>> from functools import partial
>>> con = LC("www.baidu.com",80)
>>> with con as s:
... s.send(b'GET /index.html HTTP/1.0\r\n')
... s.send(b'HOST: www.baidu.com\r\n')
... s.send(b'\r\n')
... resp = b''.join(iter(partial(s.recv,8192),b''))
...
网友评论