python基礎教程項目五之虛擬茶話會

2020-01-04 15:30:22

字體：大中小

來源：轉載

供稿：網友

幾乎在學習、使用任何一種編程語言的時候，關于socket的練習從來都不會少，尤其是會寫一些局域網的通信的東西。所以書上的這個項目剛好可以練習一下socket編程。

這個練習的整體思路首先有一個聊天的服務器，這個服務器的功能主要是提供客戶端socket的連接、存儲每個客戶端的連接session，處理每個連接發送的消息、解析客戶端發送的數據。就這些，至于客戶端方面不需要寫代碼，用系統的telnet工具即可。

我覺得有了上面的分析，剩下的這個程序就沒有什么說的了，當然，除了那兩個把socket封裝的類之外。

自己使用python中的socket類嘗試這個編寫了一個簡單的通信程序，不過不知為什么，通信中總是出現意外。這段簡單的代碼如下：

server.py

import socketmysocket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)mysocket.bind(('',8888))mysocket.listen(5)while True:    connection,addr = mysocket.accept()    revStr = connection.recv(1024)    connection.send('Server:' + revStr)    connection.close()

clinet.py

import socketimport timeclientsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)clientsocket.connect(('',8888))while True:    time.sleep(2)    clientsocket.send('hello the5fire')    print clientsocket.recv(1024)clientsocket.close()

這個程序出錯的原因沒有去細揪，因為python中提供了兩個封裝好的類來完成socket通信過程：asynchat中的async_chat和asyncore中的dispatcher以及asyncore本身。前面的類是用來處理客戶端同服務器的每一次會話，后面的類主要是用來提供socket連接服務。并且將每一個socket連接都托管給前者（async_chat）來處理。

來看代碼：

from asyncore import dispatcherfrom asynchat import async_chatimport socket, asyncorePORT = 5005NAME = 'TestChat'class EndSession(Exception):passclass CommandHandler:    def unknown(self, session, cmd):        session.push('Unknown command: %s/r/n' % cmd)    def handle(self, session, line):        if not line.strip(): return        parts = 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)class Room(CommandHandler):    def __init__(self, server):        self.server = server        self.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)        self.broadcast('Welcome to %s/r/n' % self.server.name)    def unknown(self, session, cmd):        session.push('Please log in /nUse "login"/r/n')    def do_login(self, session, line):        name = line.strip()        if not name:            session.push('Please enter a name/r/n')        elif name in self.server.users:            session.push('The name "%s" is taken./r/n' % name)            sessoin.push('Please try again./r/n')        else:            session.name = name            session.enter(self.server.main_room)class ChatRoom(Room):    def add(self, session):        self.broadcast(session.name + ' has entered the room./r/n')        self.server.users[session.name] = session        Room.add(self, session)    def remove(self, session):        Room.remove(self, session)        self.broadcast(session.name + ' has left the room./r/n')    def do_say(self, session, line):        self.broadcast(session.name + ': ' + line + '/r/n')    def do_look(self, session, line):        session.push('The following are in this room:/r/n')        for other in self.sessions:            session.push(other.name + '/r/n')    def do_who(self, session, line):        session.push('The following are logged in:/r/n')        for name in self.server.users:            session.push(name + '/r/n')class LogoutRoom(Room):    def add(self, session):        try: del self.server.users[session.name]        except KeyError: passclass ChatSession(async_chat):    def __init__(self, server, sock):        async_chat.__init__(self,sock)        self.server = server        self.set_terminator('/r/n')        self.data = []        self.name = None        self.enter(LoginRoom(server))    def enter(self, room):        try:             cur = self.room        except AttributeError:             pass        else: cur.remove(self)        self.room = room        room.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))class ChatServer(dispatcher):    def __init__(self, port, name):        dispatcher.__init__(self)        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)        self.bind(('',port))        self.listen(5)        self.name = name        self.users = {}        self.main_room = ChatRoom(self)    def handle_accept(self):        conn, addr = self.accept()        ChatSession(self,conn)if __name__ == '__main__':    s = ChatServer(PORT, NAME)    try: asyncore.loop()    except KeyboardInterrupt: print

整個程序分為我一開始說的三個部分：

提供客戶端的socket連接：ChatServer類。

存儲每個客戶端的連接session，處理每個連接發送的消息：ChatSession類，這個類的作用很簡單，接受數據，判斷是否有終結符，如果有調用found_terminator這個方法。

解析客戶端發送的數據：就是剩下的room相關的類，這些類分別用來處理客戶端發送的字符串和命令，都是繼承自CommandHandler。

最終截圖：

python,基礎教程,虛擬茶話會