#python项目练习五：虚拟茶话会

# -*- coding: utf-8 -*-
__author__ = 'Administrator'

#链接地址：     def do_logout(self,s,l):
        name=l.strip()
        if not name:
            s.push('please enter a name\r\n')
        elif name in self.server.users:
            s.push('the name "%s" is taken.\r\n'%name)
            s.push('please try again.\r\n')
        else:
            s.name=name
            s.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,s):
        Room.remove(self,s)
        self.broadcast(s.name+'has left the room\r\n')
    def do_say(self,s,l):
        self.broadcast(s.name+':'+l+'\r\n')
    def do_look(self,s,l):
        s.push('the following are in this room:\r\n')
        for o in self.sessions:
            s.push(o+'\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):
        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