python套接字和套接字服务器

Question

我一直在尝试制作一个多客户端服务器，我终于做到了，而且它工作得很好，但我现在想做的是让客户端输入他们的名字，而不是获取客户端地址，然后程序会说“Bob:大家好”而不是“127.0.0.1:大家好”。

我使用了来自 python 文档的预制服务器和客户端。这是服务器:

import socketserver

class MyUDPHandler(socketserver.BaseRequestHandler):


    def handle(self):
        data = self.request[0].strip()
        name = self.request[0].strip()
        socket = self.request[1]
        print(name,"wrote:".format(self.client_address[0]))
        print(data)
        socket.sendto(data.upper(), self.client_address)

if __name__ == "__main__":
    HOST, PORT = "localhost", 9999
    server = socketserver.UDPServer((HOST, PORT), MyUDPHandler)
    server.serve_forever()

这是客户端:

import socket
import sys

HOST, PORT = "localhost", 9999
data = "".join(sys.argv[1:])
name = "".join(sys.argv[1:])


sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

sock.sendto(bytes(name + "Bob", 'utf-8'), (HOST, PORT))
sock.sendto(bytes(data + "hello my name is Bob", "utf-8"), (HOST, PORT))
received = str(sock.recv(1024), "utf-8")

print("Sent:     {}".format(data))
print("Received: {}".format(received))

一切正常，但由于某种原因，一旦客户端连接，我就会在服务器中得到它。

b'Bob'  wrote:
b'Bob'
b'hello my name is bob'  wrote:
b'hello my name is bob'

我希望它是这样的:

Bob wrote:
b'Hello my name is bob'

希望有人能帮帮我，谢谢。

Answer 1

这里有多个问题。

---

首先是您直接打印 bytes 对象:

print(name,"wrote:".format(self.client_address[0]))

这就是为什么您得到 b'Bob' wrote: 而不是 Bob wrote: 的原因。当您在 Python 3 中打印一个 bytes 对象时，就会发生这种情况。如果你想将它解码为字符串，你必须明确地这样做。

您的代码可以在所有地方执行此操作。使用 decode 和 encode 方法通常比 str 和 bytes 构造函数更干净，如果你已经在使用 format 有更好的方法来处理这个问题，但要坚持你现有的风格:

print(str(name, "utf-8"), "wrote:".format(self.client_address[0]))

---

接下来，我不确定为什么要在没有格式参数的字符串上调用 format，或者为什么要混合使用多参数 print 函数和format 一起调用。看起来您正在尝试获取 self.client_address[0]，但您并没有这样做。再一次，你想要的输出没有显示它，所以......如果你不想要它，只需删除那个 format 调用，在格式字符串的某处添加一个 {}如果你这样做。 (您也可能想要解码 client_address[0]。)

---

接下来，您将相同的值存储在 name 和 data 中:

data = self.request[0].strip()
name = self.request[0].strip()

因此，当您稍后执行此操作时:

print(data)

…这只是再次打印 name — 而且，同样，没有对其进行解码。所以，即使你解决了解码问题，你仍然会得到这个:

Bob wrote:
Bob

... 而不仅仅是:

Bob wrote:

要解决这个问题，只需去掉 data 变量和 print(data) 调用即可。

---

接下来，您将发送两个单独的数据包，一个带有 name，另一个带有 data，但尝试从每个数据包中恢复这两个数据包。所以，即使你解决了以上所有问题，你也会在一个数据包中得到 Bob 作为名字，而 hello my name is bob 在下一个数据包中，结果:

Bob wrote:
hello my name is bob wrote:

如果您希望它是有状态的，您需要将状态实际存储在某个地方。在您的情况下，状态非常简单——只是一个标志，说明这是否是来自给定客户端的第一条消息——但它仍然必须去某个地方。一种解决方案是使用字典将新状态与每个地址相关联——尽管在这种情况下，由于状态要么是“之前看到的”，要么什么都没有，我们可以只使用一个集合。

综合起来:

class MyUDPHandler(socketserver.BaseRequestHandler):
    def __init__(self, *args, **kw):
        self.seen = set()
        super().__init__(*args, **kw)

    def handle(self):
        data = self.request[0].strip()
        addr = self.client_address[0]
        if not addr in self.seen:
            print(str(data, "utf-8"), "wrote:")
            self.seen.add(addr)
        else:
            print(str(data, "utf-8"))
        socket.sendto(data.upper(), self.client_address)

---

与此同时，看起来您实际上想要的是将第一个请求中的名称存储为您的每个客户端状态，以便您可以在以后的每个请求中重用它。这几乎一样容易。例如:

class MyUDPHandler(socketserver.BaseRequestHandler):
    def __init__(self, *args, **kw):
        self.clients = {}
        super().__init__(*args, **kw)

    def handle(self):
        data = str(self.request[0].strip(), 'utf-8')
        addr = self.client_address[0]
        if not addr in self.clients:                
            print(data, "joined!")
            self.clients[addr] = data
        else:
            print(self.clients[addr], 'wrote:', data)
        socket.sendto(data.upper(), self.client_address)