本文實例為大家分享了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()
新聞熱點
疑難解答