带有`.. as ..`的python`语句和多个返回值

Question

我正在尝试使用带声明的python（又称上下文管理器）来确保由server_socket.accept()创建的TCP连接套接字始终处于关闭状态。 但是，明显的形式不起作用，因为accept()返回多个值。

是否可以对具有多个返回值的函数使用with语句？

下面是一个最小的示例，我想使用诸如注释代码之类的东西替换try / finally块。

#!/usr/bin/env python3

import socket
from socket import socket as Socket

with Socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket:

        server_socket.bind(('', 8011))
        server_socket.listen(1)
        print("server ready")

        while True:

            # with server_socket.accept() as (connection_socket, _):
            #     request = connection_socket.recv(1024).decode('ascii')
            #     reply = request.upper()
            #     connection_socket.send(reply.encode('ascii'))

            try:
                connection_socket, _ = server_socket.accept()
                request = connection_socket.recv(1024).decode('ascii')
                reply = request.upper()
                connection_socket.send(reply.encode('ascii'))

            finally:
                connection_socket.close()

使用带注释的with语句时的错误消息是：

Traceback (most recent call last):
  File "./test.py", line 26, in <module>
    with server_socket.accept() as (connection_socket, _):
AttributeError: __exit__

据推测，这是因为一个元组不具备__exit__所需的属性with 。

Answer 1

socket.socket的返回值具有内置的上下文管理器，该上下文管理器实现__exit__和__enter__ 。 accept返回的元组没有，但是有一个库可以添加它：

import socket
import contextlib

@contextlib.contextmanager
def accept(s):
    c,a = s.accept()
    print('client connected on',a)
    yield c,a
    print('client disconnected on',a)
    c.close()

with socket.socket() as s:
    s.bind(('',8000))
    s.listen(1)
    with accept(s) as (c,a):
        while True:
            data = c.recv(1024)
            if not data: break
            c.sendall(data)

看到：

https://docs.python.org/3.3/library/contextlib.html

PS没有contextlib ：

import socket

with socket.socket() as s:
    s.bind(('',8000))
    s.listen(1)
    with s.accept()[0] as c:
        a = c.getpeername()
        print('client connected on',a)
        while True:
            data = c.recv(1024)
            if not data: break
            c.sendall(data)
        print('client disconnected on',a)