[Python-*-读书]Python基础教程--虚拟茶话会

总的来说是一个只支持一个聊天室的服务器端 客户端使用telnet

书籍下载地址：Python资料汇总

Python教程  这里是第二十四章的源码，练习研究之用

#coding:utf8 
#python2.7 chatSer2.py
#稍微复杂一点的聊天室服务器
'''
用python启动服务 telnet连接之后就可以变成简单的聊天服务器
第一次登陆：login name 
登出： logout
说话： say 
查看谁登陆过服务器：who 
查看同一个房间的人 : look 

简单例子：
telnet localhost 50005
Welcome to TestChat
login tianshi
diyu has entered the room.
nih
Unknown command: ih
say nihao
tianshi: nihao
diyu: niyehao

'''

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

HOST = ''
PORT = 50005
SERVERCONF = (HOST,PORT)
NAME = 'TestChat'

class EndSession(Exception):
    pass 

class CommandHandler(object):
    '''
    命令解析类
    '''
    
    def unknown(self,session,cmd):
        '响应未知的命令'
        session.push('Unknown command:%s \r\n' %cmd)

    def handle(self,session,line):
        '处理从客户端发来的信息 解析命令和内容'
        if not line:
            print 'please write a command...'
            
        if not line.strip(): 
            print 'get a line :',line 
            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):
        'add a user'
        self.sessions.append(session)
        
    def remove(self,session):
        'a user leave'
        self.sessions.remove(session)
        
    def broadcast(self,line):
        'give all session in the room a 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 login \nUse "login <nick>"\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)
            session.push('Please try again.\r\n')
        else:
            session.name = name
            session.enter(self.server.main_room)
            
class ChatRoom(Room):
    '''
    为多用户在一起聊天准备的房间
    '''
    def add(self, session):
        #tell all new one come
        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)
        #tell all the user logout
        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):
        'look who is in the room'
        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 
        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, serverconf, name):
        dispatcher.__init__(self)
        self.create_socket(socket.AF_INET,socket.SOCK_STREAM)
        self.set_reuse_addr() #可以重用上一次的端口号
        self.bind(SERVERCONF)
        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(SERVERCONF,NAME)
    try:
        asyncore.loop()
    except KeyboardInterrupt:
        print 'server Exception ...quit'
        
                     
        

结果如图所示