Python 3.4 Asynchio聊天服务器/客户端。 怎么样？

Question

我正在使用Python 3.4聊天asynchio服务器和客户端，但是将客户端移植到asynchio时遇到了一些麻烦。

这是供您参考的链接。

18.5.4.3.1。 TCP回显客户端协议

我的工作asynchio工作聊天服务器代码如下：

import asyncio

clients = []

class SimpleChatClientProtocol(asyncio.Protocol):
    def connection_made(self, transport):
        self.transport = transport
        peername = transport.get_extra_info("peername")
        self.username = "{:s}:{:d}".format(*peername)
        print("connection_made: {}".format(self.username))

        for client in clients:
            client.send("{:s} connected".format(self.username))

        clients.append(self)

    def send(self, text):
        self.transport.write("{:s}\n".format(text).encode())

    def data_received(self, data):
        print("data_received: {}".format(data.decode()))

        incoming = data.decode()

        if len(incoming) == 0:
            return

        if incoming.find("/") == 0:
            parts = incoming.split()

            command = parts[0]

            print("found command block: {:s}".format(command))

            if command == "/username":
                if len(parts) != 2:
                    self.send("< command: invalid payload! Usage: /username <username>")
                    return

                print("{:s} changing username to {:s}".format(self.username, parts[1]))

                for client in clients:
                    if client != self:
                        client.send("{:s} changed username to {:s}".format(self.username, parts[1]))

                self.username = parts[1]
                self.send("< username changed")

                return

            elif command == "/whisper":
                if len(parts) < 3:
                    self.send("< command: invalid payload! Usage: /w <username> <message>")
                    return

                for client in clients:
                    if client.username == parts[1]:
                        client.send("{:s} whispered: {:s}".format(self.username, " ".join(parts[2:])))
                        return

                self.send("< command: whisper target username not found!")
                return

            elif command == "/exit":
                self.send("< command: disconnecting, bye!")
                print("client_disconnect: {:s}".format(self.username))
                self.transport.close()
                return

        for client in clients:
            if client is not self:
                client.send("{:s} wrote: {:s}".format(self.username, incoming))

    def connection_lost(self, ex):
        print("connection_lost: {}".format(self.username))
        clients.remove(self)

        for client in clients:
            client.send("{:s} disconnected".format(self.username))

if __name__ == '__main__':
    print("starting up..")

    loop = asyncio.get_event_loop()
    coro = loop.create_server(SimpleChatClientProtocol, '127.0.0.1', 8888)
    server = loop.run_until_complete(coro)

    for socket in server.sockets:
        print("serving on {}".format(socket.getsockname()))

    loop.run_forever()

我的旧工作线程聊天客户端代码如下：

from socket import *
from threading import Thread

host = '127.0.0.1'
port = 8888
s = socket(AF_INET, SOCK_STREAM)
s.connect((host, port))

def Listener():
    try:
        while True:
            data = s.recv(1024).decode('utf-8')
            print('>', data)
    except ConnectionAbortedError:
        pass

t = Thread(target=Listener)
t.start()

try:
    while True:
        message = input('>')
        s.send(message.encode('utf-8'))
except EOFError:
    pass
finally:
    s.close()

使用下面的链接作为参考，有人可以帮助我修改它，以便它通过外壳输入（如我的旧客户端代码）发送消息吗？

非常感谢您的帮助。

18.5.4.3.1。 TCP回显客户端协议

Answer 1

如果我正确理解了您的问题，那么您想将STDIN消息广播到每个连接的客户端，对吗？

在这里尝试：

import asyncio

clients = []

class SimpleChatClientProtocol(asyncio.Protocol):
    def connection_made(self, transport):
        self.transport = transport
        peername = transport.get_extra_info("peername")
        self.username = "{:s}:{:d}".format(*peername)
        print("connection_made: {}".format(self.username))

        for client in clients:
            client.send("{:s} connected".format(self.username))

        clients.append(self)

    def send(self, text):
        self.transport.write("{:s}\n".format(text).encode())

    def data_received(self, data):
        print("data_received: {}".format(data.decode()))

        incoming = data.decode()

        if len(incoming) == 0:
            return

        if incoming.find("/") == 0:
            parts = incoming.split()

            command = parts[0]

            print("found command block: {:s}".format(command))

            if command == "/username":
                if len(parts) != 2:
                    self.send("< command: invalid payload! Usage: /username <username>")
                    return

                print("{:s} changing username to {:s}".format(self.username, parts[1]))

                for client in clients:
                    if client != self:
                        client.send("{:s} changed username to {:s}".format(self.username, parts[1]))

                self.username = parts[1]
                self.send("< username changed")

                return

            elif command == "/whisper":
                if len(parts) < 3:
                    self.send("< command: invalid payload! Usage: /w <username> <message>")
                    return

                for client in clients:
                    if client.username == parts[1]:
                        client.send("{:s} whispered: {:s}".format(self.username, " ".join(parts[2:])))
                        return

                self.send("< command: whisper target username not found!")
                return

            elif command == "/exit":
                self.send("< command: disconnecting, bye!")
                print("client_disconnect: {:s}".format(self.username))
                self.transport.close()
                return

        for client in clients:
            if client is not self:
                client.send("{:s} wrote: {:s}".format(self.username, incoming))

    def connection_lost(self, ex):
        print("connection_lost: {}".format(self.username))
        clients.remove(self)

        for client in clients:
            client.send("{:s} disconnected".format(self.username))

@asyncio.coroutine
def send_from_stdin(loop):
  while True:
    message = yield from loop.run_in_executor(None, input, ">")
    for client in clients:
      client.send(message)

if __name__ == '__main__':
    print("starting up..")

    loop = asyncio.get_event_loop()
    coro = loop.create_server(SimpleChatClientProtocol, '127.0.0.1', 8888)
    server = loop.run_until_complete(coro)

    # Start a task which reads from STDIN and pushes to clients.
    asyncio.async(send_from_stdin(loop))

    for socket in server.sockets:
        print("serving on {}".format(socket.getsockname()))

    loop.run_forever()

诀窍是通过在线程中运行input函数来使其异步化（这是run_in_executor作用）。 这样，当您等待来自STDIN数据时，不会阻塞整个循环。