在 python 中使用键盘快捷方式终止客户端与服务器的连接

Terminating client connection to server using keyboard shortcut in python

我正在尝试修改 python 中实现的服务器和客户端聊天脚本。其中一项要求是当用户键入 CTRL-D 时客户端退出。我的问题是如何读取用户键入的 (ctrl-D) 并在此代码中实现它。我是否应该像 c_socket.close() 这样关闭套接字而不向服务器返回任何消息我正在退出?

谢谢!

# telnet program example
import socket, select, string, sys
import readline 
def prompt() :
    sys.stdout.write('<You> ')
    sys.stdout.flush()

#main function
if __name__ == "__main__":

    if(len(sys.argv) < 3) :
        print 'Usage : python MClient hostname port'
        sys.exit()

    host = sys.argv[1]
    port = int(sys.argv[2])

    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.settimeout(2)

    # connect to remote host
    try :
        s.connect((host, port))
    except :
        print 'Unable to connect'
        sys.exit()

    print 'Connected to remote host. Start sending messages'
    prompt()

    while 1:
        socket_list = [sys.stdin, s]

        # Get the list sockets which are readable
        read_sockets, write_sockets, error_sockets = select.select(socket_list , [], [])

        for sock in read_sockets:
            #incoming message from remote server
            if sock == s:
                data = sock.recv(4096)
                if not data :
                    print '\nDisconnected from chat server'
                    sys.exit()
                else :
                    #print data
                    sys.stdout.write(data)
                    prompt()

            #user entered a message
            else :
                msg = sys.stdin.readline()
                s.send(msg)
                prompt()

最后一件事 - 在关闭套接字之前是否需要向服务器发送一些内容完全取决于您的协议。鉴于所提供的客户端代码的简单性,我猜想关闭套接字就足够了(并且服务器应该无论如何都应处理未经通知的断开连接,因为不应该将网络 I/O 视为持久的)。

其次,CTRL+D 会导致从 sys.stdin.readline() 到 return 的 returned message 为空,因此您可以对其进行测试,例如:

msg = sys.stdin.readline()
if msg:
    s.send(msg)
    prompt()
else:
    s.close()
    sys.exit()

如果你真的需要向服务器发送一些东西(同样,取决于协议)你可以在调用 close().

之前使用套接字的 shutdown() 方法

但是,请记住,此处提供的代码不考虑在从套接字读取数据(即接收数据)时按下 CTRL+D,因此如果有特别长的数据流进入您的 CTRL+D不会被注册。您可以通过在检索过程中检查用户输入(并将其写入缓冲区以供稍后发送)来解决这个问题,或者您可以将接收代码放在一个单独的线程中,而让主线程只用于用户输入。

然后您可以随时使用 atexit 模块彻底退出您的客户端。