Как использовать сокет в Python в качестве менеджера контекста?

Модуль 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() менеджером контекста, но документация не была обновлена, чтобы отразить это до в цикле Python 3.5 в 2016 году. См. socket class в исходном коде, который добавляет методы __enter__ и __exit__.

Модуль сокета — это всего лишь оболочка интерфейса сокета 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

Пожалуйста, взгляните на следующие фрагменты для сокетов TCP и UDP.

import socket
from contextlib import contextmanager


@contextmanager
def tcp_connection_to(*args, **kwargs):
    s = socket.create_connection(*args, **kwargs)
    yield s
    s.close()


@contextmanager
def udp_connection():
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    yield s
    s.close()

Чтобы вы могли использовать их следующим образом:

MY_SERVER = ('localhost', 5000)   # Yes, we need tuple here
some_data = bytes("Hello.")

with tcp_connection_to(MY_SERVER) as conn:
    conn.send(some_data)

with udp_connection() as conn:
    conn.sendto(some_data, MY_SERVER)

Я также попытался подчеркнуть разницу в поведении и подходе к термину «соединение» между TCP и UDP в именах методов.