如何在Python中使用socket作为上下文管理器？

Question

看起来像做以下事情是很自然的:

with socket(socket.AF_INET, socket.SOCK_DGRAM) as s:

但Python没有为socket实现上下文管理器.我可以轻松地将其用作上下文管理器,如果是,如何使用？

Answer 1

该socket模块相当低级,几乎可以直接访问C库功能.

您始终可以使用contextlib.contextmanager装饰器来构建自己的:

import socket
from contextlib import contextmanager

@contextmanager
def socketcontext(*args, **kw):
    s = socket.socket(*args, **kw)
    try:
        yield s
    finally:
        s.close()

with socketcontext(socket.AF_INET, socket.SOCK_DGRAM) as s:

或contextlib.closing()用来达到同样的效果:

from contextlib import closing

with closing(socket.socket(socket.AF_INET, socket.SOCK_DGRAM)) as s:

但contextmanager()装饰器让你有机会先用套接字做其他事情.

Python 3.x确实构建socket()了一个上下文管理器,即使文档没有提到它.请参阅源代码中的socket类,其中包含添加__enter__和__exit__方法.

Answer 2

套接字模块只是BSD套接字接口的包装器.它是低级的,并没有真正尝试为您提供方便或易于使用的Pythonic API.你可能想要使用更高级别的东西.

也就是说,它确实实现了一个上下文管理器:

>>> with socket.socket() as s:
...   print(s)
... 
<socket.socket object, fd=3, family=2, type=1, proto=0>

但是你需要使用Python 3.

对于Python 2兼容性,您可以使用contextlib.

from contextlib import closing
import socket

with closing(socket.socket()) as s:
    print s