如何把《python基础编程》的24章虚拟茶话会改成多房间的？

python小菜

如题！！求各位大牛指导，或者给点思路

1. #coding:utf-8
   
2. from asyncore import dispatcher  
   
3. from asynchat import async_chat
   
4. import socket
   
5. import asyncore
   
6. 
   
7. HOST = ''
   
8. PORT = 50005
   
9. SERVERCONF = (HOST,PORT)
   
10. NAME = 'TestChat'
    
11. 
    
12. class EndSession(Exception):
    
13.     pass  
    
14. 
    
15. class CommandHandler(object):
    
16.     '''''
    
17.     命令解析类
    
18.     '''
    
19.      
    
20.     def unknown(self,session,cmd):
    
21.         '响应未知的命令'
    
22.         session.push('Unknown command: %s \r\n' %cmd)
    
23. 
    
24.     def handle(self,session,line):
    
25.         '处理从客户端发来的信息 解析命令和内容'
    
26.             
    
27.         if not line.strip():  
    
28.             print 'get a line :',line  
    
29.             return
    
30.          
    
31.         parts = line.split(' ',1)
    
32.         #命令分离  
    
33.         cmd = parts[0]
    
34.         try:
    
35.             line = parts[1].strip()
    
36.         except IndexError:
    
37.             line = ''
    
38.         #调用相应的方法处理命令  
    
39.         meth = getattr(self, 'do_'+cmd, None)
    
40.         try:
    
41.             #调用命令对应的方法  
    
42.             meth(session,line)
    
43.         except TypeError:
    
44.             self.unknown(session, cmd)
    
45.             
    
46. class Room(CommandHandler):
    
47.     '''''
    
48.     包括一个或多个用户（会话）的泛型环境。负责处理基本的命令和广播
    
49.     '''
    
50.      
    
51.     def __init__(self,server):
    
52.         self.server = server
    
53.         self.sessions = []
    
54.          
    
55.     def add(self, session):
    
56.         'add a user'
    
57.         self.sessions.append(session)
    
58.          
    
59.     def remove(self,session):
    
60.         'a user leave'
    
61.         self.sessions.remove(session)
    
62.          
    
63.     def broadcast(self,line):
    
64.         'give all session in the room a line'
    
65.         for session in self.sessions:
    
66.             session.push(line)
    
67.          
    
68.     def do_logout(self,session,line):
    
69.         '响应登出命令'
    
70.         raise EndSession
    
71.      
    
72. class LoginRoom(Room):
    
73.     '''''
    
74.     为刚刚登陆的用户准备的房间
    
75.     '''
    
76.      
    
77.     def add(self,session):
    
78.         Room.add(self, session)
    
79.         self.broadcast('Welcome to %s \r\n' %self.server.name)
    
80.          
    
81.     def unknown(self, session, cmd):
    
82.         session.push('Please login \nUse "login <nick>"\r\n')
    
83.          
    
84.     def do_login(self, session, line):
    
85.         name = line.strip()
    
86.         #登陆时必须指定一个用户名  
    
87.         if not name:
    
88.             session.push('Please enter a name\r\n')
    
89.         elif name in self.server.users: #用户名不可重复 ？  
    
90.             session.push('The name "%s" is taken! \r\n' %name)
    
91.             session.push('Please try again.\r\n')
    
92.         else:
    
93.             session.name = name
    
94.             session.enter(self.server.main_room)
    
95.             
    
96. class ChatRoom(Room):
    
97.     '''''
    
98.     为多用户在一起聊天准备的房间
    
99.     '''
    
100.     def add(self, session):
     
101.         #tell all new one come  
     
102.         self.broadcast(session.name +' has entered the room. \r\n')
     
103.         self.server.users[session.name] = session
     
104.         Room.add(self, session)
     
105.          
     
106.     def remove(self, session):
     
107.         Room.remove(self. session)
     
108.         #tell all the user logout  
     
109.         self.broadcast(session.name +' has left the room \r\n')
     
110.          
     
111.     def do_say(self, session, line):
     
112.         self.broadcast(session.name+': '+line+'\r\n')
     
113.          
     
114.     def do_look(self, session, line):
     
115.         'look who is in the room'
     
116.         session.push('The following are in this room: \r\n')
     
117.         for other in self.sessions:
     
118.             session.push(other.name+'\r\n')
     
119.             
     
120.     def do_who(self, session, line):
     
121.         '查看谁登陆过'
     
122.         session.push('The following are logged in: \r\n')
     
123.         for name in self.server.users:
     
124.             session.push(name+ '\r\n')
     
125. 
     
126. class LogoutRoom(Room):
     
127.     '''''
     
128.     为单个用户准备的房间，只用于将用户从服务器移除
     
129.     '''
     
130.      
     
131.     def add(self, session):
     
132.         try:
     
133.             del self.server.users[session.name]
     
134.         except KeyError:
     
135.             pass  
     
136.          
     
137. class ChatSession(async_chat):
     
138.     '''''
     
139.     单个会话，负责和用户通信
     
140.     '''
     
141.      
     
142.     def __init__(self, server, sock):
     
143.         async_chat.__init__(self, sock)
     
144.         self.server = server
     
145.         self.set_terminator('\r\n')
     
146.         self.data = []
     
147.         self.name = None
     
148.         #所有会话从这里登入  
     
149.         self.enter(LoginRoom(server))
     
150. 
     
151.     def enter(self, room):
     
152.         try:  
     
153.             cur = self.room
     
154.         except AttributeError:
     
155.             pass  
     
156.         self.room = room  
     
157.         room.add(self)
     
158.          
     
159.     def collect_incoming_data(self, data):
     
160.         self.data.append(data)
     
161.          
     
162.     def found_terminator(self):
     
163.         line = ''.join(self.data)
     
164.         self.data = []
     
165.         try:
     
166.             self.room.handle(self, line)
     
167.         except EndSession:
     
168.             self.handle_close()
     
169.          
     
170.     def handle_close(self):
     
171.         async_chat.handle_close(self)
     
172.         self.enter(LogoutRoom(self.server))
     
173.          
     
174. class ChatServer(dispatcher):
     
175.     '''''
     
176.     只有一个房间的聊天服务器
     
177.     '''
     
178.      
     
179.     def __init__(self, serverconf, name):
     
180.         dispatcher.__init__(self)
     
181.         self.create_socket(socket.AF_INET,socket.SOCK_STREAM)
     
182.         self.set_reuse_addr() #可以重用上一次的端口号  
     
183.         self.bind(SERVERCONF)
     
184.         self.listen(5)
     
185.         self.name = name  
     
186.         self.users = {}
     
187.         self.main_room = ChatRoom(self)
     
188.      
     
189.     def handle_accept(self):
     
190.         conn, addr = self.accept()
     
191.         ChatSession(self, conn)
     
192. 
     
193. #主函数        
     
194. if __name__=='__main__':
     
195.     s = ChatServer(SERVERCONF,NAME)
     
196.     try:
     
197.         asyncore.loop()
     
198.     except KeyboardInterrupt:
     
199.         print 'server Exception ...quit'

复制代码

ssfjhh

同学，你至少把题目抄过来呗，那本书不是谁都有，而且谁有功夫帮你翻书？

python小菜

回复 2# ssfjhh

不好意思。。。代码已经贴上去啦！
   

ssfjhh

回复 3# python小菜


    同学，给你跪了。

python小菜

。。。难道还有问题！！这个程序只是一个房间的聊天，现在想改成多房间聊天，不知道如何下手回复 4# ssfjhh


   

python小菜

大牛快出现啊~不要沉了

Linux_manne

多房间聊天？ 没明白

python小菜

源程序就是只能在一个房间里面聊天，而我想把它改成多个房间，就是新来的用户可以选择创建一个房间，或者加入其他存在的房间！回复 7# Linux_manne


   

pitonas

在不同房间里面聊天?
一个房聊 不好?