本文实例为大家分享了python实现多人聊天室的具体代码,供大家参考,具体内容如下
一、目的
以实现小项目的方式,来巩固之前学过的Python基本语法以及相关的知识。
二、相关技术
1.wxpython GUI编程
2.网络编程
3.多线程编程
4.数据库编程
5.简单的将数据导出到Excel表
三、存在的漏洞以及不足
1.由于数据库编码的问题,无法使用中文。
2.在客户端关闭后,其相关的线程仍然存在于服务器的用户线程队列中,所以服务器会错误地往已关闭的客户端传送信息。
3.客户端初始登录并加载历史记录时,会出现每条历史消息后面的回车键丢失的现象,解决的方法是:在加载相邻两条消息之间加个时间间隔,但效果不佳。
四、源码
服务器Server:
# -*- coding: UTF-8 -*-from socket import *import timeimport threadingimport wximport MySQLdbimport xlwtfrom clientthread import ClientThreadclass Server(wx.Frame): def __init__(self,parent=None,id=-1,title='服务器',pos=wx.DefaultPosition,size=(500,300)): '''窗口''' wx.Frame.__init__(self,parent,id,title,pos,size=(400,470)) pl = wx.Panel(self) con = wx.BoxSizer(wx.VERTICAL) subcon = wx.FlexGridSizer(wx.HORIZONTAL) sta = wx.Button(pl , size=(133, 40),label='启动服务器') end = wx.Button(pl, size=(133, 40), label='关闭服务器') hist = wx.Button(pl,size=(133,40),label='导出聊天记录') subcon.Add(sta, 1, wx.BOTTOM) subcon.Add(hist, 1, wx.BOTTOM) subcon.Add(end, 1, wx.BOTTOM) con.Add(subcon,1,wx.ALIGN_CENTRE|wx.BOTTOM) self.Text = wx.TextCtrl(pl, size=(400,250),style = wx.TE_MULTILINE|wx.TE_READONLY) con.Add(self.Text, 1, wx.ALIGN_CENTRE) self.ttex = wx.TextCtrl(pl, size=(400,100),style=wx.TE_MULTILINE) con.Add(self.ttex, 1, wx.ALIGN_CENTRE) sub2 = wx.FlexGridSizer(wx.HORIZONTAL) clear = wx.Button(pl, size=(200, 40), label='清空') send = wx.Button(pl, size=(200, 40), label='发送') sub2.Add(clear, 1, wx.TOP | wx.LEFT) sub2.Add(send, 1, wx.TOP | wx.RIGHT) con.Add(sub2, 1, wx.ALIGN_CENTRE) pl.SetSizer(con) '''窗口''' '''绑定''' self.Bind(wx.EVT_BUTTON, self.EditClear, clear) self.Bind(wx.EVT_BUTTON, self.SendMessage, send) self.Bind(wx.EVT_BUTTON, self.Start, sta) self.Bind(wx.EVT_BUTTON, self.Break, end) self.Bind(wx.EVT_BUTTON, self.WriteToExcel, hist) '''绑定''' '''服务器准备工作''' self.UserThreadList = [] self.onServe = False addr = ('', 21567) self.ServeSock = socket(AF_INET, SOCK_STREAM) self.ServeSock.bind(addr) self.ServeSock.listen(10) '''服务器准备工作''' '''数据库准备工作,用于存储聊天记录''' self.db = MySQLdb.connect('localhost', 'root', '123456', 'user_info') self.cursor = self.db.cursor() self.cursor.execute("select * from history order by time") self.Text.SetValue('') for data in self.cursor.fetchall(): #加载历史聊天记录 self.Text.AppendText('%s said:/n%s/nwhen %s/n/n' % (data[0], data[2], data[1])) '''数据库准备工作,用于存储聊天记录''' #将聊天记录导出到EXCEl表中 def WriteToExcel(self,event): wbk = xlwt.Workbook() sheet = wbk.add_sheet('sheet 1') self.cursor.execute("select * from history order by time") sheet.write(0, 0, "User") sheet.write(0, 1, "Datetime") sheet.write(0, 5, "Message") index = 0 for data in self.cursor.fetchall(): index = index + 1 Time = '%s'%data[1] #将datetime转成字符形式,否则直接写入Excel会变成时间戳 sheet.write(index,0,data[0]) sheet.write(index,1,Time) #写进EXCEL会变成时间戳 sheet.write(index,5,data[2]) wbk.save(r'D:/History_Dialog.xls') #启动服务器的服务线程 def Start(self,event): if not self.onServe: '''启动服务线程''' self.onServe = True mainThread = threading.Thread(target=self.on_serving, args=()) mainThread.setDaemon(True) # 解决父线程结束,子线程还继续运行的问题 mainThread.start() '''启动服务线程''' #关闭服务器 def Break(self,event): self.onServe = False #服务器主循环 def on_serving(self): print '...On serving...' while self.onServe: UserSocket, UserAddr = self.ServeSock.accept() username = UserSocket.recv(1024).decode(encoding='utf-8') #接收用户名 userthread = ClientThread(UserSocket, username,self) self.UserThreadList.append(userthread) #将用户线程加到队列中 userthread.start() self.ServeSock.close() #绑定发送按钮 def SendMessage(self,event): if self.onServe and cmp(self.ttex.GetValue(),''): data = self.ttex.GetValue() self.AddText('Server',data,time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())) self.ttex.SetValue('') # 向所有客户端(包括自己)发送信息,同时更新到数据库 def AddText(self, source, data,Time): self.cursor.execute("insert into history values(/"%s/",/"%s/",/"%s/")" % (source,Time,data)) #双引号里面有双引号,bug:句子不能有双引号、以及中文 self.db.commit() sendData = '%s said:/n%s/nwhen %s/n' % (source,data,Time) self.Text.AppendText('%s/n'%sendData) for user in self.UserThreadList: #bug:客户端关闭了仍然在队列中。如果客户端关闭了,那怎么在服务器判断是否已经关闭了?客户端在关闭之前发一条信息给服务器? user.UserSocket.send(sendData.encode(encoding='utf-8')) #绑定清空按钮 def EditClear(self,event): self.ttex.Clear()def main(): app = wx.App(False) Server().Show() app.MainLoop()if __name__ == '__main__': main()
新闻热点
疑难解答