python之虚拟聊天室服务器详解

来源：互联网发布：夜来风雨声花落知多少编辑：程序博客网时间：2024/05/16 12:43

这个虚拟聊天室还是一个简单的服务器搭建的，运行截图大概是这样的

主要代码如下：

'''''  
Created on 2012-7-19  
 
@author: mars  
''' 
from asyncore import dispatcher  
from asynchat import async_chat  
import socket,asyncore  
 
PORT=5006 
NAME='TestChat2' 
 
class EndSession(Exception):pass 
 
class CommandHandler:  
    def unknown(self,session,cmd):  
        session.push('unknown command: %s\r\n'%cmd)  
    def handle(self,session,line):  
        #del the line from the target chat  
        if not line.strip(): return 
        parts=line.split(' ',1)  
        cmd=parts[0]  
        try:line=parts[1].strip()  
        except IndexError: line='' 
        #try to find the deal programe  
        meth=getattr(self,'do_'+cmd,None)  
        try:  
            meth(session,line)  
        except TypeError:  
            self.unknown(session, cmd)  
class Room(CommandHandler):  
    #including more chat responsible boardcast and deal command  
    def __init__(self,server):  
        self.server=server  
        self.sessions=[]  
    def add(self,session):  
        # one user into the room  
        self.sessions.append(session)  
    def remove(self,session):  
        # one user out the room  
        self.sessions.remove(session)  
    def broadcast(self,line):  
        # to evrytroom one line  
        for session in self.sessions:  
            session.push(line)  
    def do_logout(self,session,line):  
        #make the logout command  
        raise EndSession  
class LoginRoom(Room):  
    #for the user have been connected now to prepare room  
    def add(self,session):  
        Room.add(self, session)  
        #when user come to the room !!!welcome user  
        self.broadcast("welcome to %s\r\n" % self.server.name)  
    def unknown(self,session,cmd):  
        #chul login and loginout can make an advice  
        session.push('please log in\n Use"login<nick>"\r\n')  
      
    def remove(self,session):  
        Room.remove(self, session)  
        #Room.remove(self, session)  
        self.broadcast(session.name+"has out the room.\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)  
        else:  
            session.name=name  
            #print session.name  
              
            #print 'kkk'+name  
            #session.enter(ChatRoom(self))  
            session.enter(self.server.main_room)  
class ChatRoom(Room):  
    #multiuser chatroom  
    def add(self,session):  
        # tell all user  the newuset comein  
        self.broadcast(session.name+"has ented the room.\r\n")  
        self.server.users[session.name]=session  
        Room.add(self, session)  
    def do_say(self,session,line):  
        self.broadcast(session.name+":"+line+'\r\n')  
    def do_look(self,session,line):  
        #check who is in the room  
        session.push('the following in the room:\r\n')  
        for other in self.sessions:  
            session.push(other.name+'\r\n')  
    def do_who(self,session,line):  
        #to check who is logining   
        session.push('the floowing is logining:\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):  
    #responsibel sigle user communicate  
    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):  
    #only have one room for the chating 's server  
    def __init__(self,port,name):  
        dispatcher.__init__(self)  
        #dispatcher.__init__(self)  
        self.create_socket(socket.AF_INET,socket.SOCK_STREAM)  
        self.set_reuse_addr()  
        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

现在我们开始来详细分析其中的代码，说道详细其实也不会细致到每一行代码，毕竟很多跟上一篇文章讲的一样，这里我们主要分析的是这个cmd的基础命令解释，你看最开始的那个截图，你会发现，login用户名为ni，然后输入say 你好，然后就有 ni：你好，而这个方法就是do_say();

class CommandHandler:  
    def unknown(self,session,cmd):  
        session.push('unknown command: %s\r\n'%cmd)  
    def handle(self,session,line):  
        #del the line from the target chat  
        if not line.strip(): return 
        parts=line.split(' ',1)  
        cmd=parts[0]  
        try:line=parts[1].strip()  
        except IndexError: line='' 
        #try to find the deal programe  
        meth=getattr(self,'do_'+cmd,None)  
        try:  
            meth(session,line)  
        except TypeError:  
            self.unknown(session, cmd)

1，这个handle函数的作用：首先你输入一个行，然后 parts = line.strip(" " ,1)将行用空格将行断开成为一个个的，然后取里面的第一个也就是cmd = parts[0],就如刚刚的截图所看到的一样say就被取出来作为cmd了。然后再仔细观察下面的getattr是不是似曾相识，对的就在第一篇文本标记的时候讲过的，这个其实很好用，前面是“do_”+cmd(而cmd就是你输入的指令)这样的话，就可以完成了！这当然这是基础命令解释。

2，接下来就是Room了，功能很简单，基础类Room继承了CommandHanlder,包括一个或者多个用户会话的环境，主要负责基本命令处理，和广播。LoginRoom继承Room，为刚刚连接上的用户准备的房间，LogoutRoom用于将用户从服务器移除。ChatRoom就是多用户聊天的地方了，一旦登陆进去就到这儿了。ChatSession和ChatServer就不多说在前面一篇文章已经说过了，

本文出自 “LuoZhengWu” 博客，请务必保留此出处http://brucemars.blog.51cto.com/5288106/962319

0 0