Python noobie needs help with chatroom development

[sajaisosa]

Hey there
I've been following through the book "Beginning Python from Novice to Professional", and I have learned a great deal. At the moment, I am trying to expand upon the chatroom project near the end of this book.

My problem, is that I can't get my head around how to allow the users to add new rooms. At the moment, there is only one room. My first thought on how to do this was instantiating objects from the main room class, using the name that the user entered in the command. So, entering the command "/room myroom" would create a new object called "myroom" (if it didn't already exist) and then move the user into that room.

However, I can't see how you can specify the name of the new object as the program is running. A google session turned up nothing either.

I think I am attacking this in totally the wrong way, am I?

from asyncore import dispatcher
from asynchat import async_chat
 
import socket, asyncore
 
PORT = 5006
NAME = "Test Chat"
 
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 = ''
 
        if cmd[0] == "/":
            cmd = cmd[1:]
            meth = getattr(self, 'do_'+cmd, None)
 
            if callable(meth):
                meth(session, line)
            else:
                self.unknown(session, cmd)
        else:
            meth = getattr(self, 'do_say', None)
            if callable(meth):
                meth(session, line)
            else:
                self.unknown(session, cmd)
 
class Room(CommandHandler): 
    def __init__(self, server):
        self.server = server
        self.sessions = []
        self.name = "default"
 
    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
 
    def do_room(self, session, line):
        parts = line.split(' ', 1)
        try: line = parts[1].strip()
        except IndexError: line = ''
        roomname = line.strip()
        session.enter(self.server.main_room)
 
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 <nick>"\r\n')
 
    def do_login(self, session, line):
        parts = line.split(' ', 1)
        try: line = parts[1].strip()
        except IndexError: line = ''
        name = line.strip()
        if not name:
            session.push('Please enter a name\r\n')
        elif name in self.server.users:
            session.push('That name is already in use!')
        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 people are in the room:\r\n')
        for other in self.sessions:
            session.push(other.name + '\r\n')
 
    def do_who(self, session, line):
        session.push('The following people 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.set_reuse_addr()
        self.bind(('', PORT))
        self.listen(5)
        self.NAME = NAME
        self.users = {}
        self.main_room = ChatRoom(self)
        self.rooms
 
    def handle_accept(self):
        conn, addr = self.accept()
        ChatSession(self, conn)
 
if __name__ == '__main__':
    s = ChatServer(PORT, NAME)
    try: asyncore.loop()
    except KeyboardInterrupt: print

[bartonc]

Hey there
I've been following through the book "Beginning Python from Novice to Professional", and I have learned a great deal. At the moment, I am trying to expand upon the chatroom project near the end of this book.

My problem, is that I can't get my head around how to allow the users to add new rooms. At the moment, there is only one room. My first thought on how to do this was instantiating objects from the main room class, using the name that the user entered in the command. So, entering the command "/room myroom" would create a new object called "myroom" (if it didn't already exist) and then move the user into that room.

However, I can't see how you can specify the name of the new object as the program is running. A google session turned up nothing either.

I think I am attacking this in totally the wrong way, am I?

You can use exec() to turn strings into variable names. The trick (sometimes) is choosing which namespace to use. Go with default at first, thusly:

>>> class anObject:
... pass
...
>>> aName = "myVaribaleName "

>>> exec("%s = anObject" %(aName))
>>> myVaribaleName
<class __builtin__.anO bject at 0x02CB3A20>
>>>