Python socket C/S结构的聊天室应用实现
PythonsocketC/S结构的聊天室应用
服务端:
#!/usr/bin/envpython #coding:utf8 importsocket,select defbroadcast_data(sock,message): forsocketinconn_list: ifsocket!=server_socketandsocket!=sock: try: socket.send(message) except: socket.close() conn_list.remove(socket) if__name__=="__main__": conn_list=[] recv_buffer=4096 PORT=9999 server_socket=socket.socket(socket.AF_INET,socket.SOCK_STREAM) server_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)#端口重用,可选 server_socket.bind(('',PORT)) server_socket.listen(99) conn_list.append(server_socket) print"Chatserverstartedonport"+str(PORT) while1: read_sockets,write_sockets,error_sockets=select.select(conn_list,[],[]) forsockinread_sockets: #新建连接 ifsock==server_socket: sockfd,addr=server_socket.accept() conn_list.append(sockfd) print"Client(%s,%s)connected"%addr broadcast_data(sockfd,"[%s:%s]enteredroom\n"%addr) #进入聊天室 else: try: data=sock.recv(recv_buffer) ifdata: broadcast_data(sock,"\r"+'<'+str(sock.getpeername())+'>'+data) except: broadcast_data(sock,"Client(%s,%s)isoffline"%addr) print"Client(%s,%s)isoffline"%addr sock.close() conn_list.remove(sock) continue server_socket.close()
客户端:
#!/usr/bin/envpython #coding:utf8 importsocket,select,string,sys defprompt(): sys.stdout.write('<You>') sys.stdout.flush() if__name__=="__main__": if(len(sys.argv)<3): print'Usage:pythonchat_client.pyhostnameport' sys.exit() HOST=sys.argv[1] PORT=int(sys.argv[2]) s=socket.socket(socket.AF_INET,socket.SOCK_STREAM) s.settimeout(2) try: s.connect((HOST,PORT)) except: print'Unabletoconnect' sys.exit() print'Connectedtoremotehost.Startsendingmessages' prompt() while1: socket_list=[sys.stdin,s] read_sockets,write_sockets,error_sockets=select.select(socket_list,[],[]) forsockinread_sockets: ifsock==s: data=sock.recv(4096) ifnotdata: print'\nDisconnectedfromchatserver' sys.exit() else: sys.stdout.write(data) prompt() else: msg=sys.stdin.readline() s.send(msg) prompt()