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、参考链接