基于Socket的网络聊天室

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

PORT = 5005
NAME = "ChatRoom"

class EndSession(Exception):pass

class CommandHandler:
def unknow(self, session, cmd):
session.push('Unknow commans %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.unknow(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 unknow(self, session, cmd):
session.push('Please log in \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')
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')
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 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 loffed 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(LoginRoom(self.server))

class ChatServer(dispatcher):
def __init__(self,port ,name):
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