pyqt5实现简单的 p2p聊天

import sys
from PyQt5.QtCore import*
from PyQt5.QtWidgets import *
from newClient import*
from PyQt5.QtGui import QIcon
#主页
class MainPage(QWidget):

def __init__(self):
super().__init__()
self.initUI()

def initUI(self):
self.setGeometry(500, 50, 0, 0)
self.setWindowTitle('')
self.setWindowIcon(QIcon('icon.jpg'))
self.resize(300,300)
self.loginButton = QPushButton(self)
self.loginButton.setText("登陆")          #text
self.loginButton.setIcon(QIcon("close.png")) #icon
self.loginButton.setShortcut('Ctrl+D')  #shortcut key
self.loginButton.clicked.connect(self.close)
self.loginButton.setToolTip("Login") #Tool tip
self.loginButton.move(50,100)

self.registerButton = QPushButton(self)
self.registerButton.setText("注册")          #text
self.registerButton.setIcon(QIcon("close.png")) #icon
self.registerButton.setShortcut('Ctrl+D')  #shortcut key
self.registerButton.clicked.connect(self.close)
self.registerButton.setToolTip("Register") #Tool tip
self.registerButton.move(150,100)
self.show()
#跳转到登陆界面
class Login(QWidget):

def __init__(self):
super().__init__()

self.initUI()

def initUI(self):
self.setGeometry(500, 50, 0, 0)
self.setWindowTitle('登陆界面')
self.setWindowIcon(QIcon('icon.jpg'))
self.resize(400,300)
layout = QGridLayout()
nameLabel = QLabel("用户名")
self.nameLineEdit = QLineEdit(" ")
passWordLabel = QLabel("密码")
self.passWordLineEdit = QLineEdit(" ")
self.confirmButton = QPushButton("确定")
self.cancelButton = QPushButton("取消")
self.cancelButton.resize(50,50)
self.confirmButton.clicked.connect(self.addUser)
layout.addWidget(nameLabel,1,0)
layout.addWidget(self.nameLineEdit,1,1)
layout.addWidget(passWordLabel,2,0)
layout.addWidget(self.passWordLineEdit,2,1)
layout.addWidget(self.confirmButton)
layout.addWidget(self.cancelButton)
layout.setColumnStretch(1, 2)
self.setLayout(layout)
def handle_click(self):
if not self.isVisible():
self.show()
def addUser(self):
pList = []
pList.append(self.nameLineEdit.text)
pList.append(self.passWordLineEdit)
loginAndSend(pList)
#注册界面
class Register(QWidget):
registerSignal = pyqtSignal(list)
def __init__(self):
super().__init__()

self.initUI()

def bindSignal(self,func):
self.registerSignal.connect(func)

def initUI(self):
self.setGeometry(500, 50, 0, 0)
self.setWindowTitle('注册界面')
self.setWindowIcon(QIcon('icon.jpg'))
self.resize(300,300)
layout = QGridLayout()
nameLabel = QLabel("用户名")
self.nameLineEdit = QLineEdit("")
passWordLabel1 = QLabel("密码")
self.passWordLineEdit1 = QLineEdit("")
passWordLabel2 = QLabel("确定密码")
self.passWordLineEdit2 = QLineEdit("")
ipAddress = QLabel("IP地址")
self.ipAddressEdit =QLineEdit("")
ipPort = QLabel("端口")
self.ipPortEdit = QLineEdit("")
self.confirmButton = QPushButton("确定")
self.cancelButton = QPushButton("取消")
self.cancelButton.resize(50,50)
layout.addWidget(nameLabel,1,0)
layout.addWidget(self.nameLineEdit,1,1)
layout.addWidget(passWordLabel1,2,0)
layout.addWidget(self.passWordLineEdit1,2,1)
layout.addWidget(passWordLabel2,3,0)
layout.addWidget(self.passWordLineEdit2,3,1)
layout.addWidget(ipAddress,4,0)
layout.addWidget(self.ipAddressEdit,4,1)
layout.addWidget(ipPort,5,0)
layout.addWidget(self.ipPortEdit,5,1)
layout.addWidget(self.confirmButton)
layout.addWidget(self.cancelButton)
layout.setColumnStretch(1, 3)
self.setLayout(layout)

def sendData(self):
pList =[]
pList.append(str(self.nameLineEdit.text()))
pList.append(str(self.passWordLineEdit1.text()))
pList.append(str(self.passWordLineEdit2.text()))
re = registerAndSend(pList)
if re:
print("注册成功")

def handle_click(self):
if not self.isVisible():
self.show()
class chatPage(QMainWindow):
inputSigal = pyqtSignal(str)
sendMesSignal = pyqtSignal(str)
def __init__(self):
super().__init__()

self.initUI()

def initUI(self):
self.setGeometry(800, 50, 0, 0)
self.setWindowTitle('')
self.setWindowIcon(QIcon('icon.jpg'))
self.resize(300,600)
#设置中心窗口
componudWidget = QWidget()
layout = QGridLayout()
self.inputText = QTextEdit("输入聊天信息")
self.inputText.resize(100,100)
self.inputConButton = QPushButton("发送消息")
self.inputCanBUtton = QPushButton("取消发送")
self.showText = QTextEdit("message is here")
self.showText.resize(100,100)
layout.addWidget(self.showText)
layout.addWidget(self.inputText)
layout.addWidget(self.inputConButton)
layout.addWidget(self.inputCanBUtton)
componudWidget.setLayout(layout)
self.setCentralWidget(componudWidget)
#创建触发事件
self.exitAction = QAction(QIcon('exit.png'), '&Exit', self)
self.exitAction.setShortcut('Ctrl+Q')
self.exitAction.setStatusTip('Exit application')
self.exitAction.triggered.connect(qApp.quit)
self.indivChatA = QAction(QIcon('exit.png'), '开始聊天', self)
self.indivChatA .setShortcut("开始聊天")
self.groupChatA = QAction(QIcon('exit.png'), '创建一个群聊', self)
self.groupChatA .setShortcut("创建群聊")
self.eGroupChatA = QAction(QIcon('exit.png'), '开始加入一个群聊', self)
self.eGroupChatA .setShortcut("加入群聊")
self.aGroupChatA = QAction(QIcon('exit.png'), '添加一个群聊', self)
self.aGroupChatA .setShortcut("添加群聊")
self.upFileA = QAction(QIcon('exit.png'), '上传文件', self)
self.upFileA.setShortcut("上传文件")
self.getHelpA = QAction(QIcon('exit.png'), '帮助', self)
self.getHelpA.setShortcut("帮助")
#添加菜单
self.chat =self.menuBar().addMenu("聊天")
self.indivChat =  self.chat.addMenu("选择个体聊天")
self.groupChat = self.chat.addMenu("创建群聊")
self.enterGC = self.chat.addMenu("加入群聊")
self.addGC = self.chat.addMenu("添加群聊")
self.up = self.menuBar().addMenu("上传文件")
self.help = self.menuBar().addMenu("帮助")
self.chooseUser = self.menuBar().addMenu("选择聊天对象")
self.quit = self.menuBar().addMenu("退出系统")
#为菜单添加Action
self.indivChat.addAction(self.indivChatA)   #添加个体聊天Action
self.groupChat.addAction(self.groupChatA)   #群聊Action
self.enterGC.addAction(self.eGroupChatA)    #加入群聊Action
self.addGC.addAction(self.aGroupChatA)      #添加群聊Action
self.up.addAction(self.upFileA)             #上传文件Action
self.help.addAction(self.getHelpA)          #帮助Action
#添加信号槽
self.inputConButton.clicked.connect(self.sendMessage)
self.inputCanBUtton.clicked.connect(self.sendMessage)
self.indivChatA.triggered.connect(lambda:self.emitChatSiganl("cp")) #个体聊天触发事件
self.groupChatA.triggered.connect(lambda:self.emitChatSiganl("cg")) #群聊触发事件
self.eGroupChatA.triggered.connect(lambda:self.emitChatSiganl("eg"))#加入一个群聊
self.aGroupChatA.triggered.connect(lambda:self.emitChatSiganl("ag"))#添加一个群聊
self.upFileA.triggered.connect(self.openFile)    #上传文件触发事件
self.getHelpA.triggered.connect(lambda:self.emitChatSiganl("h"))    #获取帮助触发事件
self.quit.addAction(self.exitAction)

def handle_click(self):
self.showText.setText("聊天内容")
if not self.isVisible():
self.show()
def emitChatSiganl(self,signal):

self.inputSigal.emit(signal)
print("lunched now")

def sendMessage(self):

message = str(self.inputText.toPlainText())
target = "max"
chat(target,message)
print(message)
print ("message has emited")
def setText(self,message):
print(message)
self.showText.setText(message)

print(message)
def openFile(self):
fileName, filetype = QFileDialog.getOpenFileName(self,
"选择文件路径",
"H:/",
"All Files (*);;Text Files (*.txt)")
print(fileName)
target = "max"
file = open(fileName)
print("文件打开成功")
data = file.read(1024)
print("文件读取成功")
sendFile(target,data)
print("文件发送成功")
print(fileName)
if __name__ == '__main__':
#创建页面实例对象
app = QApplication(sys.argv)
mp = MainPage()  #主页
lg = Login()     #登陆界面
re = Register()  #注册界面
cp = chatPage()  #聊天界面
myInputData = inputData()
# myGetData = getData()
mydata = getdata()
#将信号绑定到槽
mydata.getDataSignal.connect(cp.setText)
cp.inputSigal.connect(myInputData.setTarget)
mp.loginButton.clicked.connect(lg.handle_click)
mp.loginButton.clicked.connect(mp.hide)
mp.registerButton.clicked.connect(re.handle_click)
mp.registerButton.clicked.connect(mp.hide)
re.confirmButton.clicked.connect(re.sendData)
re.confirmButton.clicked.connect(cp.handle_click)
lg.cancelButton.clicked.connect(QCoreApplication.quit)
cp.sendMesSignal.connect(myInputData.setMessage)
#与服务器建立连接，获取链接对象
tcpCliSock.connect(ADDR)
print('Connected with server')
mydata.start()
sys.exit(app.exec_())

1 import socketserver,json,time
2 import subprocess
3
4 connLst = []
5 groupLst = []
6 ##  代号 地址和端口 连接对象
7 #optype = {'ag':'group adding','cp':'chat with individual','cg':'chat with group'}
8 class Connector(object):   ##连接对象类
9     def __init__(self,account,password,addrPort,conObj):
10         self.account = account
11         self.password = password
12         self.addrPort = addrPort
13         self.conObj = conObj
14
15 class Group(object):#群组类
16     def __init__(self,groupname,groupOwner):
17         self.groupId = 'group'+str(len(groupLst)+1)
18         self.groupName = 'group'+groupname
19         self.groupOwner = groupOwner
20         self.createTime = time.time()
21         self.members=[groupOwner]
22
23 class MyServer(socketserver.BaseRequestHandler):
24
25     def handle(self):
26         print("got connection from",self.client_address)
27         userIn = False
28         global connLst
29         global groupLst
30         while not userIn:
31             conn = self.request
32             data = conn.recv(1024)
33             if not data:
34                 continue
35             dataobj = json.loads(data.decode('utf-8'))
36             #如果连接客户端发送过来的信息格式是一个列表且注册标识为False时进行用户注册或者登陆
37             ret = '0'
38             if type(dataobj) == list and not userIn:
39                 account = dataobj[0]
40                 password = dataobj[1]
41                 optype = dataobj[2]
42                 existuser = False
43                 if len(connLst) > 0:
44                     for obj in connLst:
45                         if obj.account == account:
46                             existuser = True
47                             if obj.password == password:
48                                 userIn = True
49                                 print('{} has logged in system({})'.format(account,self.client_address))
50                                 break
51                 if optype == 'login' and (not userIn or not existuser):
52                     ret = '1'
53                     print('{} failed to logged in system({})'.format(account, self.client_address))
54                 else:
55                     if existuser:
56                         ret = '1'
57                         print('{} failed to register({}),account existed!'.format(account, self.client_address))
58                     else:
59                         try:
60                             conObj = Connector(account,password,self.client_address,self.request)
61                             connLst.append(conObj)
62                             print('{} has registered to system({})'.format(account,self.client_address))
63                             userIn = True
64                         except:
65                             print('%s failed to register for exception!'%account)
66                             ret = '99'
67             #conn.sendall(ret.encode('utf-8'))
68             if ret == '0':
69                 break
70
71         while True:
72         #除登陆注册之外的请求的监听
73             conn = self.request
74             data = conn.recv(1024)
75             if not data:
76                 continue
77             print(data)
78             dataobj = data.decode('utf-8')
79             dataobj = json.loads(dataobj)
80             if dataobj['type'] == 'ag' and userIn:
81             #如果判断用户操作请求类型为添加群组则进行以下操作
82                 groupName = dataobj['groupName']
83                 groupObj = Group(groupName,self.request)
84                 groupLst.append(groupObj)
85                 conn.sendall('ag0'.encode('utf-8'))
86                 print('%s added'%groupName)
87                 continue
88
89             if dataobj['type'] == 'eg' and userIn:
90             #入群操作
91                 groupName = dataobj['groupName']
92                 ret = 'eg1'
93                 for group in groupLst:
94                     if groupName == group.groupName:
95                         group.members.append(self.request)
96                         print('{} added into {}'.format(self.client_address,groupName))
97                         ret = 'eg0'
98                         break
99                 conn.sendall(ret.encode('utf-8'))
100                 continue
101
102             #客户端将数据发给服务器端然后由服务器转发给目标客户端
103             print('connLst',connLst)
104             print('grouplst',groupLst)
105             if len(connLst) > 1:
106                 sendok = False
107                 if dataobj['type'] == 'cg':
108                 #群内广播（除发消息的人）
109                     print('group',data)
110                     for obj in groupLst:
111                         if obj.groupName == dataobj['to']:
112                             for user in obj.members:
113                                 if user != self.request:
114                                     user.sendall(data)
115                 if dataobj['type'] == 'cp':
116                 #个人信息发送
117                     for obj in connLst:
118                         if dataobj['to'] == obj.account:
119                             obj.conObj.sendall(data)
120                             sendok = True
121                     if sendok == False:
122                         print('no target valid!')
123                 if dataobj['type'] == 'f':
124                     for obj in connLst:
125                         if dataobj['to'] == obj.account:
126                             obj.conObj.sendall(data)
127                             sendok = True
128                     if sendok == False:
129                         print('no target valid!')
130             else:
131                 conn.sendall('-1'.encode('utf-8'))
132                 continue
133
134 if __name__ == '__main__':
135     server = socketserver.ThreadingTCPServer(('localhost',9998),MyServer)
136     print('waiting for connection...')
137     server.serve_forever()

1 from socket import *
2 import threading,sys,json,re
3 from PyQt5.QtCore import*
4 from PyQt5.QtWidgets import *
5 #引入json模块主要是为了数据的封装传输，re的话是做一些合法性的验证
6 HOST = 'localhost'
7 PORT=9998
8 BUFSIZE = 1024  ##缓冲区大小  1K
9 ADDR = (HOST,PORT)
10 myre = r"^[_a-zA-Z]\w{0,}"
11 tcpCliSock = socket(AF_INET,SOCK_STREAM)
12 #创建一个socket连接
13 userAccount = None
14 #用户登录标志，也用来记录登录的用户名称
15 def registerAndSend(list):
16 #用户注册函数
17     print("""
18     Glad to have you a member of us!
19     """)
20     accout = list[0]
21     # if not re.findall(myre, accout):
22     #     print('Account illegal!')
23     #     return None
24     password1  = str(list[1])
25     password2 = str(list[2])
26     if not (password1 and password1.__eq__(password2)):
27         print("password1: %s" %password1)
28         print("password2: %s" %password2)
29         print('Password not illegal!')
30         return None
31     try:
32         global userAccount
33         userAccount = accout
34         regInfo = [accout,password1,'register']
35         datastr = json.dumps(regInfo)
36         tcpCliSock.send(datastr.encode('utf-8'))
37         #data = tcpCliSock.recv(BUFSIZE)
38         data = "1"
39         data = data.decode('utf-8')
40         if data == '0':
41            print('Success to register!')
42            return True
43         elif data == '1':
44             print('Failed to register, account existed!')
45             return False
46         else:
47             print('Failed for exceptions!')
48             return False
49     except Exception as e:
50         print(e)
51
52 def loginAndSend(list):
53 #用户登录函数
54     print("""
55     Welcome to login in!
56     """)
57     accout = list[0]
58     if not re.findall(myre, accout):
59         print('Account illegal!')
60         return None
61     password = list[1]
62     if not password:
63         print('Password illegal!')
64         return None
65     global userAccount
66     userAccount = accout
67     loginInfo = [accout, password,'login']
68     datastr = json.dumps(loginInfo)
69     tcpCliSock.send(datastr.encode('utf-8'))
70     data = tcpCliSock.recv(BUFSIZE)
71     if data == '0':
72         print('Success to login!')
73         return True
74     else:
75         print('Failed to login in(user not exist or username not match the password)!')
76         return False
77
78 def chat(target,msg):
79 #进入聊天（群聊和点对点聊天可以选择）
80         print('{} -> {}: '.format(userAccount,target))
81         print("I am chating now inside the chat function")
82         if len(msg) > 0 and not msg in 'qQ':
83             if 'group' in target:
84                 optype = 'cg'
85             else:
86                 optype = 'cp'
87
88             dataObj = {'type': optype, 'to': target, 'msg': msg, 'froms': userAccount}
89             datastr = json.dumps(dataObj)
90             tcpCliSock.send(datastr.encode('utf-8'))
91 def sendFile(target,file):
92     print("a file send")
93     dataObj = {'type': "f", 'to': target, "file":file,'froms': userAccount}
94     print(dataObj["type"])
95     datastr = json.dumps(dataObj)
96     tcpCliSock.send(datastr.encode('utf-8'))
97     print("a file send")
98
99
100 class inputData(threading.Thread):
101     message = ""
102     targe = ""
103     def setMessage(self,msg):
104         global message
105         message = msg
106         print(message)
107         return message
108     def setTarget(self,tgt):
109         global target
110         targe = tgt
111         print(targe)
112         return targe
113     def run(self):
114         global targe,message
115         print(targe)
116         print(message)
117
118
119 class getdata(QThread):
120 #接收数据线程
121     getDataSignal = pyqtSignal(str)
122     def run(self):
123         while True:
124
125
126               data = tcpCliSock.recv(BUFSIZE).decode('utf-8')
127               dataObj = json.loads(data)
128               if dataObj['type'] == 'cp':
129                  newStr = str('{} -> : {}'.format(dataObj['froms'], dataObj['msg']))
130                  print(newStr)
131                  self.getDataSignal.emit(newStr)
132               elif dataObj['type'] == 'f':
133                  print("I got your file ")
134