PythonsocketC/S结构的聊天室应用实现

#!/usr/bin/env python
#coding:utf8
 
import socket,select
def broadcast_data (sock,message):
 for socket in conn_list:
 if socket != server_socket and socket != 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 "Chat server started on port " + str(PORT)
 
 while 1:
 read_sockets,write_sockets,error_sockets = select.select(conn_list,[],[])
 for sock in read_sockets:
 #新建连接
 if sock == server_socket:
 sockfd, addr = server_socket.accept()
 conn_list.append(sockfd)
 print "Client (%s, %s) connected" % addr
 broadcast_data(sockfd, "[%s:%s] entered room
" % addr)
 #进入聊天室
 else:
 try:
 data = sock.recv(recv_buffer)
 if data:
 broadcast_data(sock, "
" + '<' + str(sock.getpeername()) + '> ' + data)
 except:
 broadcast_data(sock, "Client (%s, %s) is offline" % addr)
 print "Client (%s, %s) is offline" % addr
 sock.close()
 conn_list.remove(sock)
 continue
 
 server_socket.close()

#!/usr/bin/env python
#coding:utf8
import socket,select,string,sys
 
def prompt() :
 sys.stdout.write(' ')
 sys.stdout.flush()
 
if __name__ == "__main__":
 
 if(len(sys.argv) < 3) :
 print 'Usage : python chat_client.py hostname port'
 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 'Unable to connect'
 sys.exit()
 
 print 'Connected to remote host. Start sending messages'
 prompt()
 
 while 1:
 socket_list = [sys.stdin, s]
 
 read_sockets, write_sockets, error_sockets = select.select(socket_list , [], [])
 
 for sock in read_sockets:
 if sock == s:
 data = sock.recv(4096)
 if not data :
 print '
Disconnected from chat server'
 sys.exit()
 else :
 sys.stdout.write(data)
 prompt()
 
 else :
 msg = sys.stdin.readline()
 s.send(msg)
 prompt()