python每日小程序2简易聊天室服务端qq41600708的博客-

原理

利用pytho标准库的socket模块，可以快速建立TCP连接，可以利用TCP连接实现简单的聊天室

编译环境是在linux中，win应该也可以

代码

重要的地方都有注释，主要为两个类，
Server类主要用来存储当前连接到服务器上的用户名和连接到服务器上的socket对象(用于服务器主动发送消息回去)
ChatRoom_s用来绑定address来监听客户端的连接，然后开启多线程接受数据，并且把收到的一个客户信息，发送给其他所有的连接上来的用户

from socket import * import threading import atexit  class Server(object): def __init__(self):         self.addr_user_map = {} #addr对应的用户名         self.addr_sock_map = {} #addr对应的客户端通信socket def add_user(self,addr,username):         self.addr_user_map[addr] = username      def del_user(self,addr): del self.addr_user_map[addr] def get_user(self,addr): return self.addr_user_map.get(addr,'None') def check_user(self,username): return username in self.addr_user_map.values() def add_client(self,addr,s):         self.addr_sock_map[addr] = s      def del_client(self,addr): del self.addr_sock_map[addr] def iter_client(self,addr): '''         用来获得，除了参数addr外的所有的客户端socket,用以发送消息给所有聊天室的人         ''' for a in self.addr_sock_map: if a == addr: continue else: yield self.addr_sock_map[a] class ChatRoom_s(object): '''     聊天室类     ''' def __init__(self): '''         变量的初始化         '''         self.bind_addr = ('127.0.0.1',8899)         self.server = Server()         self.s = None def start(self): '''         开始服务器socket的监听和创建子线程来处理客户端请求         '''         self.s = socket()         self.s.bind(self.bind_addr)         self.s.listen(10) print('服务器开启监听--->') while True:             c_s,addr = self.s.accept()             self.server.add_client(addr,c_s)              t = threading.Thread(target=self.handle_client,args=(c_s,addr))             t.start() def handle_client(self,s,addr): '''         一直循环等待，客户的发送消息         ''' while True:             a = s.recv(1024)             a = a.decode('utf-8') if '[name]' in a:                 name = a.split()[1]                 self.server.add_user(addr,name) print('当前的用户',self.server.addr_user_map) continue #改用户名的消息，不用广播 #如果用户退出聊天器就把服务器里关于addr的记录删了，然后退出这个函数(线程) if '[stop]' in a: print(f'[info]: {self.server.get_user(addr)}-->logout')                 self.server.del_user(addr)                 self.server.del_client(addr) return 0 print(f'接收到来自:{addr}{self.server.get_user(addr)}的消息:{a}')             b = f"[{self.server.get_user(addr)}]"              c = b + a + 'n'             self.send_all(addr,c) def send_all(self,addr,message): '''         给除了发送过来的每一个人发送消息         ''' for s in self.server.iter_client(addr):             s.send(message.encode('utf-8')) def clean(self): '''         当程序结束时候，用来关闭socket（发送一个,[stop] 消息）         ''' print('程序退出了...')         self.s.close() if __name__ == "__main__":     a = ChatRoom_s()     a.start()     atexit.register(a.clean)