How to use socket in Python as a context manager?

by

The socket module is fairly low-level, giving you almost direct access to the C library functionality.

You can always use the contextlib.contextmanager decorator to build your own:

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:

or use contextlib.closing() to achieve the same effect:

from contextlib import closing

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

but the contextmanager() decorator gives you the opportunity to do other things with the socket first.

Python 3.x does make socket() a context manager, but the documentation wasn’t updated to reflect this until well into the Python 3.5 cycle, in 2016. See the socket class in the source code, which adds __enter__ and __exit__ methods.

More Related Contents:

  1. Python socket not receiving without sending
  2. TypeError: a bytes-like object is required, not ‘str’
  3. Python Socket Receive Large Amount of Data
  4. Create a “with” block on several context managers? [duplicate]
  5. Problem with multi threaded Python app and socket connections
  6. How to set timeout on python’s socket recv method?
  7. On localhost, how do I pick a free port number?
  8. What does Python’s socket.recv() return for non-blocking sockets if no data is received until a timeout occurs?
  9. Errno 10061 : No connection could be made because the target machine actively refused it ( client – server )
  10. Python simple socket client/server using asyncio
  11. Python Socket Flush
  12. Opening a SSL socket connection in Python
  13. Python server “Only one usage of each socket address is normally permitted”
  14. Python Error: io.UnsupportedOperation: fileno
  15. How to connect to a remote Windows machine to execute commands using python?
  16. Why am I getting the error “connection refused” in Python? (Sockets)
  17. How does the python socket.recv() method know that the end of the message has been reached?
  18. Can Python select what network adapter when opening a socket?
  19. 104, ‘Connection reset by peer’ socket error, or When does closing a socket result in a RST rather than FIN?
  20. Send a file through sockets in Python
  21. Python socket.accept nonblocking?
  22. Imported module not found in PyInstaller
  23. Python [Errno 98] Address already in use
  24. How do I determine all of my IP addresses when I have multiple NICs?
  25. Making moves w/ websockets and python / django ( / twisted? )
  26. Alternative to contextlib.nested with variable number of context managers
  27. Catching exception in context manager __enter__()
  28. Context manager for Python’s MySQLdb
  29. WinError 10049: The requested address is not valid in its context
  30. Sending multiple files through a TCP socket

Leave a Comment