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()