43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- encoding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
Topic: 让对象支持上下文管理器
|
||
|
|
Desc :
|
||
|
|
"""
|
||
|
|
|
||
|
|
from socket import socket, AF_INET, SOCK_STREAM
|
||
|
|
|
||
|
|
|
||
|
|
class LazyConnection:
|
||
|
|
def __init__(self, address, family=AF_INET, type=SOCK_STREAM):
|
||
|
|
self.address = address
|
||
|
|
self.family = family
|
||
|
|
self.type = type
|
||
|
|
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
|
||
|
|
|
||
|
|
|
||
|
|
from functools import partial
|
||
|
|
|
||
|
|
conn = LazyConnection(('www.python.org', 80))
|
||
|
|
# Connection closed
|
||
|
|
with conn as s:
|
||
|
|
# conn.__enter__() executes: connection open
|
||
|
|
s.send(b'GET /index.html HTTP/1.0\r\n')
|
||
|
|
s.send(b'Host: www.python.org\r\n')
|
||
|
|
s.send(b'\r\n')
|
||
|
|
resp = b''.join(iter(partial(s.recv, 8192), b''))
|
||
|
|
print(resp)
|
||
|
|
# conn.__exit__() executes: connection closed
|
||
|
|
|