Python聊天室
1.簡介
本次項目課是實現簡單聊天室程序的服務器端和客戶端。
2.知識點
服務器端涉及到asyncore、asynchat和socket這幾個模塊,客戶端用到了telnetlib、wx、time和thread這幾個模塊。
二、項目實戰(服務器端)
1.服務器類
首先需要一個聊天服務器,這里繼承asyncore的dispatcher類來實現,代碼如下
class ChatServer(dispatcher):"""聊天服務器"""def __init__(self, port):dispatcher.__init__(self)self.create_socket(socket.AF_INET, socket.SOCK_STREAM)self.set_reuse_addr()self.bind(('', port))self.listen(5)self.users = {}self.main_room = ChatRoom(self)def handle_accept(self):conn, addr = self.accept()ChatSession(self, conn)2.會話類
有了服務器類還需要能維護每個用戶的連接會話,這里繼承asynchat的async_chat類來實現,代碼如下:
class ChatSession(async_chat):"""負責和單用戶通信"""def __init__(self, server, sock):async_chat.__init__(self, sock)self.server = serverself.set_terminator('\n')self.data = []self.name = Noneself.enter(LoginRoom(server))def enter(self, room):'從當前房間移除自身,然后添加到指定房間'try:cur = self.roomexcept AttributeError:passelse:cur.remove(self)self.room = roomroom.add(self)def collect_incoming_data(self, data):'接受客戶端的數據'self.data.append(data)def found_terminator(self):'當客戶端的一條數據結束時的處理'line = ''.join(self.data)self.data = []try:self.room.handle(self, line)except EndSession:self.handle_close()def handle_close(self):async_chat.handle_close(self)self.enter(LogoutRoom(self.server))3.命令解釋器
現在就需要一個命令解釋器能夠解釋用戶的命令,例如登錄、查詢在線用戶和發消息等,代碼如下:
class CommandHandler:"""命令處理類"""def unknown(self, session, cmd):'響應未知命令'session.push('Unknown command: %s\n' % cmd)def handle(self, session, line):'命令處理'if not line.strip():returnparts = line.split(' ', 1)cmd = parts[0]try:line = parts[1].strip()except IndexError:line = ''meth = getattr(self, 'do_' + cmd, None)try:meth(session, line)except TypeError:self.unknown(session, cmd)4.房間
接下來就需要實現聊天室的房間了,這里我們定義了三種房間,分別是用戶剛登錄時的房間、聊天的房間和退出登錄的房間,這三種房間都有一個公共的父類,代碼如下:
class Room(CommandHandler):"""包含多個用戶的環境,負責基本的命令處理和廣播"""def __init__(self, server):self.server = serverself.sessions = []def add(self, session):'一個用戶進入房間'self.sessions.append(session)def remove(self, session):'一個用戶離開房間'self.sessions.remove(session)def broadcast(self, line):'向所有的用戶發送指定消息'for session in self.sessions:session.push(line)def do_logout(self, session, line):'退出房間'raise EndSessionclass LoginRoom(Room):"""剛登錄的用戶的房間"""def add(self, session):'用戶連接成功的回應'Room.add(self, session)session.push('Connect Success')def do_login(self, session, line):'登錄命令處理'name = line.strip()if not name:session.push('UserName Empty')elif name in self.server.users:session.push('UserName Exist')else:session.name = namesession.enter(self.server.main_room)class ChatRoom(Room):"""聊天用的房間"""def add(self, session):'廣播新用戶進入'session.push('Login Success')self.broadcast(session.name + ' has entered the room.\n')self.server.users[session.name] = sessionRoom.add(self, session)def remove(self, session):'廣播用戶離開'Room.remove(self, session)self.broadcast(session.name + ' has left the room.\n')def do_say(self, session, line):'客戶端發送消息'self.broadcast(session.name + ': ' + line + '\n')def do_look(self, session, line):'查看在線用戶'session.push('Online Users:\n')for other in self.sessions:session.push(other.name + '\n')class LogoutRoom(Room):"""用戶退出時的房間"""def add(self, session):'從服務器中移除'try:del self.server.users[session.name]except KeyError:pass5.服務器端完整代碼
(略)
小編生平也最討厭這個“略”字,不過既然各位看官已經辛苦看到了這里,小編不給點福利實在是于心不忍。
友情提示一:功力高深的各位好漢,集齊前四種代碼,可自行召喚服務器端完整代碼和小編香吻一枚。
友情提示二:功力不足的親們,請登錄實驗樓官方網站http://www.shiyanlou.com/courses/?course_type=project&tag=all ,閉關練功。
(喂喂~不許扔雞蛋!大家都是文明人!么么噠)三、項目實戰(客戶端)
完成了服務器端后,就需要實現客戶端了,這里客戶端連接服務器使用了telnetlib模塊。
1.登錄窗口
這里的圖形界面包選擇了wxPython,前面有安裝說明,登錄窗口通過繼承wx.Frame類來實現,代碼如下:
class LoginFrame(wx.Frame):"""登錄窗口"""def __init__(self, parent, id, title, size):'初始化,添加控件并綁定事件'wx.Frame.__init__(self, parent, id, title)self.SetSize(size)self.Center()self.serverAddressLabel = wx.StaticText(self, label = "Server Address", pos = (10, 50), size = (120, 25))self.userNameLabel = wx.StaticText(self, label = "UserName", pos = (40, 100), size = (120, 25))self.serverAddress = wx.TextCtrl(self, pos = (120, 47), size = (150, 25))self.userName = wx.TextCtrl(self, pos = (120, 97), size = (150, 25))self.loginButton = wx.Button(self, label = 'Login', pos = (80, 145), size = (130, 30))self.loginButton.Bind(wx.EVT_BUTTON, self.login)self.Show()def login(self, event):'登錄處理'try:serverAddress = self.serverAddress.GetLineText(0).split(':')con.open(serverAddress[0], port = int(serverAddress[1]), timeout = 10)response = con.read_some()if response != 'Connect Success':self.showDialog('Error', 'Connect Fail!', (95, 20))returncon.write('login ' + str(self.userName.GetLineText(0)) + '\n')response = con.read_some()if response == 'UserName Empty':self.showDialog('Error', 'UserName Empty!', (135, 20))elif response == 'UserName Exist':self.showDialog('Error', 'UserName Exist!', (135, 20))else:self.Close()ChatFrame(None, -2, title = 'ShiYanLou Chat Client', size = (500, 350))except Exception:self.showDialog('Error', 'Connect Fail!', (95, 20))def showDialog(self, title, content, size):'顯示錯誤信息對話框'dialog = wx.Dialog(self, title = title, size = size)dialog.Center()wx.StaticText(dialog, label = content)dialog.ShowModal()2.聊天窗口
聊天窗口中最主要的就是向服務器發消息并接受服務器的消息,這里通過子線程來接受,代碼如下:
class ChatFrame(wx.Frame):"""聊天窗口"""def __init__(self, parent, id, title, size):'初始化,添加控件并綁定事件'wx.Frame.__init__(self, parent, id, title)self.SetSize(size)self.Center()self.chatFrame = wx.TextCtrl(self, pos = (5, 5), size = (490, 310), style = wx.TE_MULTILINE | wx.TE_READONLY)self.message = wx.TextCtrl(self, pos = (5, 320), size = (300, 25))self.sendButton = wx.Button(self, label = "Send", pos = (310, 320), size = (58, 25))self.usersButton = wx.Button(self, label = "Users", pos = (373, 320), size = (58, 25))self.closeButton = wx.Button(self, label = "Close", pos = (436, 320), size = (58, 25))self.sendButton.Bind(wx.EVT_BUTTON, self.send)self.usersButton.Bind(wx.EVT_BUTTON, self.lookUsers)self.closeButton.Bind(wx.EVT_BUTTON, self.close)thread.start_new_thread(self.receive, ())self.Show()def send(self, event):'發送消息'message = str(self.message.GetLineText(0)).strip()if message != '':con.write('say ' + message + '\n')self.message.Clear()def lookUsers(self, event):'查看當前在線用戶'con.write('look\n')def close(self, event):'關閉窗口'con.write('logout\n')con.close()self.Close()def receive(self):'接受服務器的消息'while True:sleep(0.6)result = con.read_very_eager()if result != '':self.chatFrame.AppendText(result)3.客戶端完整代碼
咳咳咳~小編要告訴你們一個不幸的消息,(眾:下去下去!)好吧,我想你們也該知道了,客戶端完整代碼=1(登錄窗口)+2(聊天窗口)+自我調整,若有不明白的看客,請登錄實驗樓官方網站實驗樓
http://www.shiyanlou.com/courses/?course_type=project&tag=all(小編我是穿著鋼鐵盔甲上來滴,親們,溫柔點~)四、小結
最后就可以運行程序進行聊天了,注意需要先啟動服務器再啟動客戶端。這個項目中使用了asyncore的dispatcher來實現服務 器,asynchat的asyn_chat來維護用戶的連接會話,用wxPython來實現圖形界面,用telnetlib來連接服務器,在子線程中接受 服務器發來的消息,由此一個簡單的聊天室程序就完成了。
這里的圖形界面使用的是wxPython,試著換一個圖形界面包來實現客戶端。這個程序非常簡單,你也可以自己擴展想要的功能。
項目效果截圖
登錄窗口
聊天窗口
?寫到這里,小編不得不感謝各位看官,有什么意見和需要歡迎和小編溝通~一躬到底!!
總結
- 上一篇: 基于Server-Sent Event的
- 下一篇: DIY一个高大上带提醒的计时器,简单实用