Как да използвам сокет в 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 клас в изходния код, който добавя __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 в имената на методите.