快捷導(dǎo)航

python基礎(chǔ)教程項(xiàng)目五之虛擬茶話會(huì)

更新時(shí)間：2018年04月02日 11:47:34 作者：the5fire

這篇文章主要為大家詳細(xì)介紹了python基礎(chǔ)教程項(xiàng)目五之虛擬茶話會(huì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

幾乎在學(xué)習(xí)、使用任何一種編程語(yǔ)言的時(shí)候，關(guān)于socket的練習(xí)從來(lái)都不會(huì)少，尤其是會(huì)寫一些局域網(wǎng)的通信的東西。所以書上的這個(gè)項(xiàng)目剛好可以練習(xí)一下socket編程。

這個(gè)練習(xí)的整體思路首先有一個(gè)聊天的服務(wù)器，這個(gè)服務(wù)器的功能主要是提供客戶端socket的連接、存儲(chǔ)每個(gè)客戶端的連接session，處理每個(gè)連接發(fā)送的消息、解析客戶端發(fā)送的數(shù)據(jù)。就這些，至于客戶端方面不需要寫代碼，用系統(tǒng)的telnet工具即可。

我覺得有了上面的分析，剩下的這個(gè)程序就沒有什么說(shuō)的了，當(dāng)然，除了那兩個(gè)把socket封裝的類之外。

自己使用python中的socket類嘗試這個(gè)編寫了一個(gè)簡(jiǎn)單的通信程序，不過(guò)不知為什么，通信中總是出現(xiàn)意外。這段簡(jiǎn)單的代碼如下：

server.py

import socket

mysocket = 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 socket
import time

clientsocket = 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()

這個(gè)程序出錯(cuò)的原因沒有去細(xì)揪，因?yàn)閜ython中提供了兩個(gè)封裝好的類來(lái)完成socket通信過(guò)程：asynchat中的async_chat和asyncore中的dispatcher以及asyncore本身。前面的類是用來(lái)處理客戶端同服務(wù)器的每一次會(huì)話，后面的類主要是用來(lái)提供socket連接服務(wù)。并且將每一個(gè)socket連接都托管給前者（async_chat）來(lái)處理。

來(lái)看代碼：

from asyncore import dispatcher
from asynchat import async_chat
import socket, asyncore

PORT = 5005
NAME = 'TestChat'

class EndSession(Exception):pass

class 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 EndSession

class 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: pass

class 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

整個(gè)程序分為我一開始說(shuō)的三個(gè)部分：

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

存儲(chǔ)每個(gè)客戶端的連接session，處理每個(gè)連接發(fā)送的消息：ChatSession類，這個(gè)類的作用很簡(jiǎn)單，接受數(shù)據(jù)，判斷是否有終結(jié)符，如果有調(diào)用found_terminator這個(gè)方法。

解析客戶端發(fā)送的數(shù)據(jù)：就是剩下的room相關(guān)的類，這些類分別用來(lái)處理客戶端發(fā)送的字符串和命令，都是繼承自CommandHandler。

最終截圖：