用 Python 开发一个「聊天室」 _Python

接下来我们就使用 Python 来操作 socket，实现一个聊天室的一些主要功能。
首先我们来回想下，一般的聊天室都是怎样的，有多个用户可以同时在线，他们可以实时获取到消息，实时发送消息。

服务端的实现

那么服务端要实现的就有这么几点：

监听客户端的连接
同时操作多个用户
广播消息通知

代码撸起来：

因为我们要做到 “同时” 去操作用户，就需要用到多线程：

import socketfrom threading import Thread
接着创建一下 socket，绑定地址和端口号：

host = '127.0.0.1'port = 8080s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)s.bind((host, port))
我们可以定义字典来存放用户的数据，比如连接用户的昵称以及地址：

client = {}addresses = {}
再来定义下服务器可接收的 client 连接数：

accept_num = 10
接下来可以在 main 方法在监听用户的连接：

if __name__ == '__main__': s.listen(accept_num)print('服务器已经开启，正在监听用户的请求..')
接着可以写一个 whie 循环来接收用户的连接：

while True:conn, address = s.accept()print(address, '已经建立连接')conn.send('欢迎你来到帅帅的聊天室，请输入你的昵称进行聊天'.encode('utf8'))

接收到用户的连接之后，我们就可以获取到用户的连接和地址信息，可以把地址保存到我们刚刚定义的字典里面来：

addresses[conn] = address
要支持多个用户的信息收发，我们可以开启线程：

Thread(target=handle_client_in, args=(conn, address)).start()
至此，我们的 main 方法是这样的：

接着我们来实现一下用户的消息处理方法，我们可以接收用户发来的昵称消息，这时候就可以在聊天室里面进行广播，告诉大家 “xxx 加进来了”，另外我们可以把用户的昵称加到字典中来：

def handle_client_in(conn, addr):nikename = conn.recv(1024).decode('utf8')welcome = f'欢迎 {nikename} 加入聊天室'client[conn] = nikenamebrodcast(bytes(welcome, 'utf8'))

接下来可以定义一个 While 循环，来监听用户发送的消息，当服务端获取到用户发来的消息之后，我们可以在聊天室进行广播，告诉大家 “xxx 发来了 xxx 消息”，而当用户由于异常而退出聊天室的时候，我们可以将连接关掉，并且把字典存着的用户数据给删掉：

while True:¦try:¦¦msg = conn.recv(1024)¦¦brodcast(msg, nikename+':')¦except:¦¦conn.close()¦¦del client[conn]¦¦brodcast(bytes(f'{nikename} 离开聊天室', 'utf8'))

那么如何对聊天室的用户进行广播呢，因为我们刚刚在字典中都存储了连接进来的用户连接，那么就可以通过循环的方式向每个用户发送消息：

def brodcast(msg, nikename=''):for conn in client:conn.send(bytes(nikename, 'utf8') + msg)
这样，一个聊天室的服务端主要功能就完成了：