# -*- coding: utf-8 -*-
import select
import socketinBufSize = 4096
outBufSize = 4096
CONNECTION_LIST = []

    def __init__(self,port=5247):# todo 使用socketserver来写self.serverSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)self.serverSocket.bind(('', port))self.serverSocket.listen(5)print "server wait for connect...."self.socketsMap = {}  # socket session字典 id : socketself.idMap = {} #socket session 字典 socket:idCONNECTION_LIST.append(self.serverSocket)

    def socet_handle(self):while 1:# Get the list sockets which are ready to be read through selectread_sockets, write_sockets, error_sockets = select.select(CONNECTION_LIST, [], [])for sock in read_sockets:# New connectionif sock == self.serverSocket:#用户通过主socket（即服务器开始创建的 socket，一直处于监听状态）来登录# Handle the case in which there is a new connection recieved through server_socketsockfd, addr = self.serverSocket.accept()id = sockfd.recv(100)self.login(id,sockfd)else:self.chat(sock)

    def login(self,id,sock):#新用户登录print "%s login"%idself.socketsMap[id] = sockself.idMap[sock] = idsock.send('hello %s,you login successed'%id)CONNECTION_LIST.append(sock)#要在这里把socket加进来才行

    def chat(self,sock):#点对点聊天，发送消息格式id||信息try:data = sock.recv(inBufSize)except Exception:sock.send("remote is offline")sock.close()else:remote_id = data.split('||')[0]message = data.split('||')[1]print "id = %s,message = %s"%(remote_id,message)local_id = self.idMap[sock]if remote_id == 'all':self.broadcast(local_id,message)else:self.p2psend(local_id,message,remote_id)def p2psend(self,local_id,message,remote_id):remote_socket = self.socketsMap[remote_id]message_send = "%s said : %s" % (local_id, message)try:remote_socket.sendall(message_send)except Exception,e:print eremote_socket.close()CONNECTION_LIST.remove(remote_socket)def broadcast(self,local_id,message):for sock in CONNECTION_LIST:if sock == self.serverSocket:continueelse:try:message_send = "%s said : %s" % (local_id, message)sock.send(message_send)except Exception,e:print esock.close()CONNECTION_LIST.remove(sock)continue

# -*- coding: utf-8 -*-
import select
import socketinBufSize = 4096
outBufSize = 4096
CONNECTION_LIST = []class ChatServer:def __init__(self,port=5247):# todo 使用socketserver来写self.serverSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)self.serverSocket.bind(('', port))self.serverSocket.listen(5)print "server wait for connect...."self.socketsMap = {}  # socket session字典 id : socketself.idMap = {} #socket session 字典 socket:id
        CONNECTION_LIST.append(self.serverSocket)def login(self,id,sock):#新用户登录print "%s login"%idself.socketsMap[id] = sockself.idMap[sock] = idsock.send('hello %s,you login successed'%id)CONNECTION_LIST.append(sock)#要在这里把socket加进来才行def chat(self,sock):#点对点聊天，发送消息格式id||信息try:data = sock.recv(inBufSize)except Exception:sock.send("remote is offline")sock.close()else:remote_id = data.split('||')[0]message = data.split('||')[1]print "id = %s,message = %s"%(remote_id,message)local_id = self.idMap[sock]if remote_id == 'all':self.broadcast(local_id,message)else:self.p2psend(local_id,message,remote_id)def p2psend(self,local_id,message,remote_id):remote_socket = self.socketsMap[remote_id]message_send = "%s said : %s" % (local_id, message)try:remote_socket.sendall(message_send)except Exception,e:print eremote_socket.close()CONNECTION_LIST.remove(remote_socket)def broadcast(self,local_id,message):for sock in CONNECTION_LIST:if sock == self.serverSocket:continueelse:try:message_send = "%s said : %s" % (local_id, message)sock.send(message_send)except Exception,e:print esock.close()CONNECTION_LIST.remove(sock)continuedef socet_handle(self):while 1:# Get the list sockets which are ready to be read through selectread_sockets, write_sockets, error_sockets = select.select(CONNECTION_LIST, [], [])for sock in read_sockets:# New connectionif sock == self.serverSocket:#用户通过主socket（即服务器开始创建的 socket，一直处于监听状态）来登录# Handle the case in which there is a new connection recieved through server_socketsockfd, addr = self.serverSocket.accept()id = sockfd.recv(100)self.login(id,sockfd)else:self.chat(sock)def main(self):self.socet_handle()self.serverSocket.close()if __name__ == '__main__':chat_server_obj = ChatServer()chat_server_obj.main()

    def socket_handler(self):while 1:rlist = [sys.stdin, self.client_socket]  # 接收列表read_list, write_list, error_list = select.select(rlist, [], [], 2)for sock in read_list:# incoming message from remote serverif sock == self.client_socket:data = sock.recv(4096)if not data:print '\nDisconnected from chat server'sys.exit()else:# print data
                        sys.stdout.write(data)self.prompt()# user entered a messageelse:msg = sys.stdin.readline()remote_id = raw_input("Please input remote id:")msg_send = "%s||%s"%(remote_id,msg)self.client_socket.send(msg_send)self.prompt()

# -*- coding:utf-8 -*-
import socket, select, string, sys
HOST = '127.0.0.1'
PORT = 5247
ID = '1'class ChatClient:def __init__(self):self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)self.client_socket.settimeout(2)self.connect()def connect(self):try:self.client_socket.connect((HOST, PORT))self.client_socket.send(ID)except Exception,e:print 'Unable to connect because of %s'%esys.exit()else:print 'Connected to remote host. Start sending messages'self.prompt()def prompt(self):sys.stdout.write('\n<You> ')sys.stdout.flush()def socket_handler(self):while 1:rlist = [sys.stdin, self.client_socket]  # 接收列表read_list, write_list, error_list = select.select(rlist, [], [], 2)for sock in read_list:# incoming message from remote serverif sock == self.client_socket:data = sock.recv(4096)if not data:print '\nDisconnected from chat server'sys.exit()else:# print data
                        sys.stdout.write(data)self.prompt()# user entered a messageelse:msg = sys.stdin.readline()remote_id = raw_input("Please input remote id:")msg_send = "%s||%s"%(remote_id,msg)self.client_socket.send(msg_send)self.prompt()if __name__ == '__main__':chat_client_obj = ChatClient()chat_client_obj.socket_handler()