Python实现多用户全双工聊天(一对一),python多用户,多用户全双工聊天简陋


多用户全双工聊天简陋版

简单实现了两个客户端之间的通信,客户端发送消息,先由服务器接收,然后服务器转发到另一客户端。

该版本功能非常简陋,仅仅实现了最简单的聊天,有很多地方需要注意。

工作步骤:

服务器端运行一个客户端运行,连接成功后输入用户名,服务器会保存该用户名在一个字典中,字典的对应关系是 username --> socket输入用户名之后,该客户端需要确定一个聊天用户,客户端输入To:user即可;如果客户端发送其他文本的话,会收到来自服务器的提示:“Nobody is chatting with you. Maybe the one talked with you is talking with someone else”当两个客户端成功连接之后就可以互相发送消息

服务器端代码如下:

#!/usr/bin/python#coding:utf-8#server.pyfrom socket import *from time import ctimeimport threadingimport reHOST = ‘‘PORT = 9999BUFSIZ = 1024ADDR = (HOST,PORT)tcpSerSock = socket(AF_INET,SOCK_STREAM)tcpSerSock.bind(ADDR)tcpSerSock.listen(5)clients = {} # username -> socketchatwith = {} # user1.socket -> user2.socket# clients字典中记录了连接的客户端的用户名和套接字的对应关系# chatwith字典中记录了通信双方的套接字的对应# messageTransform()处理客户端确定用户名之后发送的文本# 文本只有四种类型:#   None#   Quit#   To:someone#   其他文本def messageTransform(sock,user):    while True:        data = sock.recv(BUFSIZ)        if not data:            if chatwith.has_key(sock):                chatwith[sock].send(data)                del chatwith[chatwith[sock]]                del chatwith[sock]            del clients[user]            sock.close()            break        if data==‘Quit‘:            sock.send(data)            if chatwith.has_key(sock):                data = ‘%s.‘ % data                chatwith[sock].send(data)                del chatwith[chatwith[sock]]                del chatwith[sock]            del clients[user]            sock.close()            break        elif re.match(‘^To:.+‘, data) is not None:            data = data[3:]                if clients.has_key(data):                if data==user:                        sock.send(‘Please don\‘t try to talk with yourself.‘)                else:                        chatwith[sock] = clients[data]                        chatwith[clients[data]] = sock                else:                sock.send(‘the user %s is not exist‘ % data)        else:            if chatwith.has_key(sock):                chatwith[sock].send(‘[%s] %s: (%s)‘ % (ctime(),user,data))            else:                sock.send(‘Nobody is chating with you. Maybe the one talked with you is talking with someone else‘)             # 每个客户端连接之后,都会启动一个新线程# 连接成功后需要输入用户名# 输入的用户名可能会:#   已存在#   (客户端直接输入ctrl+c退出)#   合法用户名def connectThread(sock,test): # client‘s socket        user = None    while True: # receive the username        username = sock.recv(BUFSIZ)        if not username: # the client logout without input a name            print(‘The client logout without input a name‘)            break        if clients.has_key(username): # username existed            sock.send(‘Reuse‘)        else: # correct username            sock.send(‘OK‘)            clients[username] = sock # username -> socket            user = username            break    if not user:        sock.close()        return    print(‘The username is: %s‘ % user)    # get the correct username        messageTransform(sock,user)            if __name__==‘__main__‘:    while True:        print(‘...WAITING FOR CONNECTION‘)        tcpCliSock, addr = tcpSerSock.accept()        print(‘CONNECTED FROM: ‘, addr)        chat = threading.Thread(target = connectThread, args = (tcpCliSock,None))        chat.start()

客户端代码如下:

#!/usr/bin/python#coding:utf-8#client.pyfrom socket import *from time import ctime# from termios import tcflush,TCIFLUSHimport threadingimport sysHOST = ‘127.0.0.1‘PORT = 9999BUFSIZ = 1024ADDR = (HOST,PORT)tcpCliSock = socket(AF_INET,SOCK_STREAM)tcpCliSock.connect(ADDR)‘‘‘因为每个客户端接收消息和发送消息是相互独立的,所以这里将两者分开,开启两个线程处理‘‘‘def Send(sock,test):    while True:    try:        data = raw_input()        sock.send(data)        if data==‘Quit‘:            break    except KeyboardInterrupt:        sock.send(‘Quit‘)        break    def Recv(sock,test):    while True:    data = sock.recv(BUFSIZ)    if data==‘Quit.‘:        print(‘He/She logout‘)        continue    if data==‘Quit‘:        break    print(‘         %s‘ % data)    if __name__==‘__main__‘:    print(‘Successful connection‘)    while True:        username = raw_input(‘Your name(press only Enter to quit): ‘)        tcpCliSock.send(username)        if not username:            break        # username is not None        response = tcpCliSock.recv(BUFSIZ)        if response==‘Reuse‘:            print(‘The name is reuse, please set a new one‘)            continue        else:            print(‘Welcome!‘)            break    if not username:        tcpCliSock.close()        recvMessage = threading.Thread(target = Recv, args = (tcpCliSock,None))    sendMessage = threading.Thread(target = Send, args = (tcpCliSock,None))    sendMessage.start()    recvMessage.start()    sendMessage.join()    recvMessage.join()

总结:

功能简陋,后续会有所改进。这里还有很多地方需要注意。

比如说两个客户端A成功连接后,和客户端B聊天。A发送消息时直接输入ctrl+c退出程序(sendMessage线程会结束),我将这种情况模拟成A发送Quit登出。服务器接收到A登出信息之后,会回发一个Quit给A,A成功登出(recvMessage线程结束)。此外如果A和B建立了聊天关系,就要接触这个关系,服务器发送Quit.给B,B会继续接收信息,但是服务器端的chatwith字典中已经不存在A.socket --> B.socket关系。

但是还有很多没有解决的问题,比如说客户端A并没有输入信息,直接点击关闭按钮退出,就会发生异常(与之聊天的B客户端会崩溃)。

如果当前存在 A-->B的聊天关系,这时有一个C登录,并且确定了C-->A的聊天关系,那么A会和C聊天,这时客户端B就会被挂起。

主要的问题还是在于客户端非正常登出时的应对,目前解决了一部分问题,但是应该还有不少缺陷。

Python实现多用户全双工聊天(一对一)

评论关闭