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

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

该

socket

模块相当底层，使您几乎可以直接访问C库功能。

您始终可以使用

contextlib.contextmanager

装饰器来构建自己的装饰器：

import socketfrom contextlib import contextmanager@contextmanagerdef 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 closingwith closing(socket.socket(socket.AF_INET, socket.SOCK_DGRAM)) as s:

但是

contextmanager()

装饰器使您有机会首先使用套接字执行其他 *** 作。

Python 3.x确实可以

socket()

创建上下文管理器，但是直到2016年Python
3.5周期才进行
更新，文档并未得到反映。请参阅源代码中的

socket

类，其中添加了

__enter__

和

__exit__

方法。