python实现的聊天室（二）

python实现的聊天室（二）

1、前言

我在python实现的聊天室（一）一文中实现了一个没有界面的聊天室，所有操作都在CMD命令行下进行，用户体验比较差。所以在这一篇文章中，主要实现一个有界面的聊天室。

2、需求分析

我们要实现的分为两部分：

• 服务器：负责与用户建立 Socket 连接，并将某个用户发送的消息广播到所有在线的用户。显示用户进入/退出聊天室信息。
• 客户端：可以输入聊天的内容并发送，同时可以显示其他用户的消息记录。

3、服务端实现

# -*- coding: gbk -*-  
from PyQt4.QtGui import *  
from PyQt4.QtCore import *  
import sys  
from PyQt4.QtNetwork import *  
import locale	  
#QTextCodec.setCodecForTr(QTextCodec.codecForName("utf8"))	  
class TcpClientSocket(QTcpSocket):  
    def __init__(self,parent=None):  
        super(TcpClientSocket,self).__init__(parent)  
        self.connect(self,SIGNAL("readyRead()"),self.dataReceive)  
        self.connect(self,SIGNAL("disconnected()"),self.slotDisconnected)  
        self.length = 0  
        self.msglist = QByteArray()  	      
    def dataReceive(self):  
        while self.bytesAvailable() > 0:  
            length = self.bytesAvailable()  
            msg = self.read(length)  
            self.emit(SIGNAL("updateClients(QString,int)"),msg,length)  	                      
    def slotDisconnected(self):  
        pass  	          
class Server(QTcpServer):  
    def __init__(self,parent=None,port=0):  
        super(Server,self).__init__(parent)  
        self.listen(QHostAddress.Any,port)  
        self.tcpClientSocketList = []  	      
    def incomingConnection(self,socketDescriptor):  
        tcpClientSocket = TcpClientSocket(self)  
        	        self.connect(tcpClientSocket,SIGNAL("updateClients(QString,int)"),self.updateClients)  
        	        	        self.connect(tcpClientSocket,SIGNAL("disconnetcted(int)"),self.slotDisconnected)  
        tcpClientSocket.setSocketDescriptor(socketDescriptor)  
        self.tcpClientSocketList.append(tcpClientSocket)        
    def updateClients(self,msg,length):  
        self.emit(SIGNAL("updateServer(QString,int)"),msg,length)  
        for i in xrange(len(self.tcpClientSocketList)):  
            item = self.tcpClientSocketList[i]
            length_msg = item.writeData(msg)			
            #length_msg = item.writeData(msg.toUtf8())  
            #if length_msg != msg.toUtf8().length():  
            #    continue  					      
    def slotDisconnected(self,descriptor):  
        for i in xrange(len(self.tcpClientSocketList)):  
            item = self.tcpClientSocketList[i]  
            if item.socketDescriptor() == descriptor:  
                self.tcpClientSocketList.remove[i]  
                return  
        return    
class TcpServer(QDialog):  
    def __init__(self,parent=None,f=None):  
        super(TcpServer,self).__init__(parent)  
        self.setWindowTitle("服务端")  
        vbMain = QVBoxLayout(self)  
        self.ListWidgetContent = QListWidget(self)  
        vbMain.addWidget(self.ListWidgetContent)  	          
        hb = QHBoxLayout()  
        LabelPort = QLabel(self)  
        LabelPort.setText(self.tr("Port:"))  
        hb.addWidget(LabelPort)  	          
        LineEditPort = QLineEdit(self)  
        hb.addWidget(LineEditPort)  	          
        vbMain.addLayout(hb)  	          
        self.PushButtonCreate = QPushButton(self)  
        self.PushButtonCreate.setText(self.tr("创建聊天室"))  
        vbMain.addWidget(self.PushButtonCreate)  
          	        self.connect(self.PushButtonCreate,SIGNAL("clicked()"),self.slotCreateServer)  
        self.port  = 8010  
        LineEditPort.setText(QString.number(self.port))  	      
    def slotCreateServer(self):  
        server = Server(self,self.port)  
        self.connect(server,SIGNAL("updateServer(QString,int)"),self.updateServer)  
        self.PushButtonCreate.setEnabled(False)  	      
    def updateServer(self,msg,length):  
        #self.ListWidgetContent.addItem(msg.fromUtf8(msg))  
		self.ListWidgetContent.addItem(msg) 	                                 
if __name__ == '__main__':             
	app=QApplication(sys.argv)	
	mycode = locale.getpreferredencoding()
	code = QTextCodec.codecForName(mycode)
	QTextCodec.setCodecForLocale(code)
	QTextCodec.setCodecForTr(code)
	QTextCodec.setCodecForCStrings(code)		
	dialog=TcpServer()  
	dialog.show()  
	app.exec_()  

4、客户端实现

# -*- coding: gbk -*-  
from PyQt4.QtGui import *  
from PyQt4.QtCore import *  
import sys  
from PyQt4.QtNetwork import *  
import locale  
#QTextCodec.setCodecForTr(QTextCodec.codecForName("utf8"))  	  
class TcpClient(QDialog):  
    def __init__(self,parent=None):  
        super(TcpClient,self).__init__(parent)  
        self.setWindowTitle(self.tr("客户端"))    	          
        self.status = False   
        self.serverIP = QHostAddress()  
        self.port = 8010  
        self.msglist = QByteArray()  	          
        vbMain = QVBoxLayout(self)  
        self.ListWidgetContent = QListWidget(self)  
        vbMain.addWidget(self.ListWidgetContent)  	          
        hb = QHBoxLayout()  
        self.LineEditMessage = QLineEdit(self)  
        hb.addWidget(self.LineEditMessage)  
        self.PushButtonSend = QPushButton(self)  
        self.PushButtonSend.setText(self.tr("发送"))  
        self.PushButtonSend.setEnabled(False)  
        hb.addWidget(self.PushButtonSend)  
        self.connect(self.PushButtonSend,SIGNAL("clicked()"),self.slotSend)  	          
        hb1 = QHBoxLayout()  
        LabelName = QLabel(self)  
        LabelName.setText(self.tr("用户名:"))  
        self.LineEditUser = QLineEdit(self)  
        hb1.addWidget(LabelName)  
        hb1.addWidget(self.LineEditUser)  	          
        hb2 = QHBoxLayout()  
        LabelServerIP = QLabel(self)  
        LabelServerIP.setText(self.tr("服务器地址:"))  
        self.LineEditIP = QLineEdit(self)  
        hb2.addWidget(LabelServerIP)  
        hb2.addWidget(self.LineEditIP)   	          
        hb3 = QHBoxLayout()  
        LabelPort = QLabel(self)  
        LabelPort.setText(self.tr("端口:")) 	          
        self.LineEditPort = QLineEdit(self)  
        hb3.addWidget(LabelPort)  
        hb3.addWidget(self.LineEditPort)   	          
        vbMain.addLayout(hb)  
        vbMain.addLayout(hb1)  
        vbMain.addLayout(hb2)  
        vbMain.addLayout(hb3)  	          
        self.PushButtonLeave = QPushButton(self)  
        self.PushButtonLeave.setText(self.tr("进入聊天室"))  
        vbMain.addWidget(self.PushButtonLeave)  	          
        self.connect(self.PushButtonLeave,SIGNAL("clicked()"),self.slotEnter)  	          
    def slotSend(self):  
        msg = self.userName + ":" + self.LineEditMessage.text(		
        #length = self.tcpSocket.writeData(msg.toUtf8())
        length = self.tcpSocket.writeData(msg)		
        self.LineEditMessage.clear()   
        #if length != msg.toUtf8().length():  
        #    return    	          
    def slotEnter(self):  
        if not self.status:  
            ip = self.LineEditIP.text()  
            if not self.serverIP.setAddress(ip):  
                QMessageBox.information(self,self.tr("error"),self.tr("server ip address error!"))  
                return  
            if self.LineEditUser.text() == "":  
                QMessageBox.information(self,self.tr("error"),self.tr("User Name error!"))  
                return  
            self.userName = self.LineEditUser.text()  
            self.tcpSocket = QTcpSocket(self)  
            self.connect(self.tcpSocket,SIGNAL("connected()"),self.slotConnected)  
            self.connect(self.tcpSocket,SIGNAL("disconnected()"),self.slotDisconnected)  
            self.connect(self.tcpSocket,SIGNAL("readyRead()"),self.dataReceived)              
            self.tcpSocket.connectToHost(self.serverIP.toString(),8010)  
            self.status = True  
        else:  
            msg = self.userName + ":" + self.tr("离开聊天室") 
            length = self.tcpSocket.writeData(msg)			
            #length = self.tcpSocket.writeData(msg.toUtf8()) 			
            #if length != msg.toUtf8().length():  
            #    return                
            self.tcpSocket.disconnectFromHost()  
            self.status = False  	      
    def slotConnected(self):  
        self.PushButtonSend.setEnabled(True)  
        self.PushButtonLeave.setText(self.tr("离开聊天室"))  	          
        msg = self.userName + ":" + self.tr("进入聊天室")  
        #length = self.tcpSocket.writeData(msg.toUtf8())
        length = self.tcpSocket.writeData(msg)		
        #if length != msg.toUtf8().length():  
        #    return  	      
    def slotDisconnected(self):  
        self.PushButtonSend.setEnabled(False)  
        self.PushButtonLeave.setText(self.tr("进入聊天室"))  
    def dataReceived(self):  
        while self.tcpSocket.bytesAvailable() > 0:  
            length = self.tcpSocket.bytesAvailable()  
            msg = QString(self.tcpSocket.read(length))  
            #msg = msg.fromUtf8(msg)  
            #self.ListWidgetContent.addItem(msg.fromUtf8(msg))
            self.ListWidgetContent.addItem(msg)				          	  
if __name__ == '__main__':  
	app=QApplication(sys.argv) 	
	mycode = locale.getpreferredencoding()
	code = QTextCodec.codecForName(mycode)
	QTextCodec.setCodecForLocale(code)
	QTextCodec.setCodecForTr(code)
	QTextCodec.setCodecForCStrings(code)		
	dialog=TcpClient()  
	dialog.show()  
	app.exec_()  

5、运行效果

运行时先启动服务端，再运行客户端。服务器启动后创建聊天室。客户端运行后，输入昵称、服务器的IP及port信息，进入聊天室。此时服务端会提示用户进入聊天室。当客户端关闭，表明该用户退出聊天室，服务器会转发该用户退出消息到各个客户端。
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-TLWfRAnb-1593526750899)(http://odsh9s4s2.bkt.clouddn.com/mypyqtchatserver.png)]
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-XLahNRDt-1593526750901)(http://odsh9s4s2.bkt.clouddn.com/mypyqtchatclientA.png)]
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-xB06qmLF-1593526750903)(http://odsh9s4s2.bkt.clouddn.com/mypyqtchatclientB.png)]

6、参考链接

因为代码中用到了汉字，所以涉及到编码问题。希望该链接能帮助到你。
http://blog.csdn.net/uselym/article/details/51867056
526750901)]
[外链图片转存中…(img-xB06qmLF-1593526750903)]

6、参考链接

因为代码中用到了汉字，所以涉及到编码问题。希望该链接能帮助到你。
http://blog.csdn.net/uselym/article/details/51867056