目录

4.1 确定同一聊天室玩家及房主

4.2 绘制开始或准备按钮

4.3 实现按钮功能

4.4 完整代码下载地址：

在本节，我们会在原有的程序基础上加上房间的功能。当玩家打开游戏窗口后，会自动出现在1号或者2号聊天室。只有当1号聊天室所有玩家准备完毕，房主点击开始聊天按钮后，才可以对话，2号聊天室同理。

本项目结构显示如下：

├── SimHei.ttf        # 字体文件
├── client.py         # 客户端代码
├── pics              # 图片文件夹
│   ├── 1.png
│   ├── 2.png
│   ├── 3.png
│   ├── 4.png
│   ├── 5.png
│   └── 6.png
├── player.py         # 包含Player类
└── server.py         # 服务端代码

4.1 确定同一聊天室玩家及房主

进入1号聊天室的玩家只会看到同样进入1号聊天室的其他玩家，不会在窗口上显示2号聊天室的玩家。另外，如果某个玩家是房主（也就是第一个进入到聊天室的），那在人物图片下方就会显示“房主”文本。

修改Player类：

# player.py
class Player:def __init__(self, p_id, x, y,  pic_num, frame_width, frame_height):...self.is_host = False                # 1self.room = random.randint(1, 2)    # 2...def draw(self, win, pic):...if self.is_host:                    # 3host_text = font.render("房主", True, (150, 150, 150))win.blit(host_text, (self.x + round(self.frame_width/2) - round(host_text.get_width()/2), self.y + self.frame_height + host_text.get_height()))

代码解释如下：

1. is_host变量用来判断当前玩家是不是房主。

2. room变量用来表示玩家当前进入的房间号，要么是1号房间，要么是2号房间。

3. 如果是房主，那就在人物图片下方绘制“房主”两个字。

修改Server类：

# server.py
class Server:...def handle_message(self, conn):while True:try:data = conn.recv(2048)if not data:print("未接收到数据，关闭连接")self.players_data.pop(str(id(conn)))conn.close()breakelse:data = pickle.loads(data)self.update_one_player_data(data)self.set_host(data["player"].room)    # 1conn.sendall(pickle.dumps(self.get_other_players_data(data["id"], data["player"].room)))    # 2except Exception as e:print(repr(e))breakdef update_one_player_data(self, data):key = data["id"]value = data["player"]self.players_data[key] = valuedef get_other_players_data(self, current_player_id, current_player_room):data = {}for key, value in self.players_data.items():if key != current_player_id and value.room == current_player_room:data[key] = valuereturn datadef set_host(self, current_player_room):for key, value in self.players_data.items():if value.room == current_player_room:self.players_data[key].is_host = Truereturn

代码解释如下：

1. set_host()函数用来设置房主，在字典中靠前的键值对都是先进入到房间的玩家数据，因为在Python3.6之后的版本中，字典是有序的。比如A玩家先进入1号聊天室，B玩家之后再进入1号聊天室，那在players_data字典变量中，A玩家的数据就排在B玩家数据之前。假如A玩家退出了游戏，那服务端也会删除A玩家数据，B玩家就成房主了。

2. 通过get_other_players_data()函数获取其他玩家数据时，我们还要传入当前玩家的房间号，因为要返回房间号相同的其他玩家数据，而不是其他所有玩家数据。

注：如果你使用的是Python3.6之前的版本，那服务端在确定房主时会出现错误，因为字典不是有序的。

修改GameWindow类：

# client.py
class GameWindow:...def update_window(self):self.window.fill((255, 255, 255))self.draw_room_text()                         # 1self.player.move()self.player.draw(self.window, self.pic_dict[self.player.pic_num])if randint(0, 1000) < 50:self.player.speak()self.update_messages(self.player)other_players_data = pickle.loads(self.send_player_data())self.update_other_players_data(other_players_data)self.show_host(other_players_data)             # 2pygame.display.update()...def draw_room_text(self):font = pygame.font.Font("SimHei.ttf", 15)room_text = font.render(f"{self.player.room}号房间", True, (150, 150, 150))self.window.blit(room_text, (self.width/2-room_text.get_width()/2, 10))def show_host(self, data):for player in data.values():if player.is_host:returnself.player.is_host = True

代码解释如下：

1. 将当前的房间号绘制到窗口上方。

2. 如果其他某个玩家的is_host值是True，那当前玩家就不是房主；如果其他玩家的is_host值都是False，那当前玩家就是房主。

运行结果如下：

 1号房间有两位玩家，2号房间只有一位玩家。 

4.2 绘制开始或准备按钮

房主玩家的窗口上应该绘制开始聊天按钮，当其他所有玩家都准备完毕后，点击开始聊天按钮就可以显示对话内容。非房主玩家界面上显示的是点击准备和取消准备按钮。

修改Player类：

# player.py
class Player:def __init__(self, p_id, x, y,  pic_num, frame_width, frame_height):...self.message = ""self.is_host = Falseself.is_ready = False       # 1self.is_started = False     # 2self.room = random.randint(1, 2)...def draw(self, win, pic):       # 3...if self.is_host:host_text = font.render("房主", True, (150, 150, 150))win.blit(host_text, (self.x + round(self.frame_width/2) - round(host_text.get_width()/2), self.y + self.frame_height + host_text.get_height()))else:if self.is_ready:text = font.render("已准备", True, (150, 150, 150))else:text = font.render("未准备", True, (150, 150, 150))win.blit(text, (self.x + round(self.frame_width/2) - round(text.get_width()/2), self.y + self.frame_height + text.get_height()))...

代码解释如下：

1. is_ready变量表示当前玩家的准备状态。

2. is_started变量用来表示游戏是否开始，如果房主玩家点击了开始聊天，那其他所有玩家的is_started值都会变为True。

3. 如果is_ready为True，那就在人物图下方绘制“已准备”三个字；如果为False，就绘制“未准备”三个字。

修改GameWindow类：

# client.py
class GameWindow:def __init__(self):...self.all_players_messages = []self.max_messages_shown = 6self.btn_width = 100            # 1self.btn_height = 40self.btn_rect = (self.width/2 - self.btn_width/2, self.height - self.btn_height, self.btn_width, self.btn_height)...def update_window(self):self.window.fill((255, 255, 255))self.draw_room_text()self.draw_button()              # 2self.player.move()self.player.draw(self.window, self.pic_dict[self.player.pic_num])if randint(0, 1000) < 50:self.player.speak()self.update_messages(self.player)other_players_data = pickle.loads(self.send_player_data())self.update_other_players_data(other_players_data)self.show_host(other_players_data)pygame.display.update()...def update_messages(self, player):if not player.is_started:        # 3returnself.player.is_started = True......def draw_button(self):if self.player.is_started:returnfont = pygame.font.Font("SimHei.ttf", 15)pygame.draw.rect(self.window, (150, 150, 150), self.btn_rect)if self.player.is_host:btn_text = font.render("开始聊天", True, (255, 255, 255))else:if self.player.is_ready:btn_text = font.render("取消准备", True, (255, 255, 255))else:btn_text = font.render("点击准备", True, (255, 255, 255))self.window.blit(btn_text, (self.btn_rect[0] + self.btn_width / 2 - btn_text.get_width() / 2,self.btn_rect[1] + self.btn_height / 2 - btn_text.get_height() / 2))...

代码解释如下：

1. 确定按钮的宽高和绘制位置。

2. 将按钮绘制到窗口底部正中位置，如果聊天已经开始，那开始/准备按钮就无需再出现。窗口上一共会出现三种按钮：开始聊天按钮、点击准备按钮以及取消准备按钮。

3. 如果其他玩家数据中的is_started属性值为False，就代表还未开始聊天，于是不绘制聊天内容。

运行结果如下：

4.3 实现按钮功能

非房主玩家点击准备按钮后，人物图下方的文本会从“未准备”变为“已准备”。房主玩家点击开始聊天按钮后，如果其他任何一个玩家还未准备，那在窗口上就会显示xxx玩家还未准备。如果全部都已经准备完毕，房主点击开始聊天按钮后，窗口上就会显示聊天内容了。

修改GameWindow类：

# client.py
class GameWindow:...def update_window(self):...other_players_data = pickle.loads(self.send_player_data())self.update_other_players_data(other_players_data)self.show_host(other_players_data)self.check_click(other_players_data)    # 1pygame.display.update()def update_messages(self, player):if not player.is_started:returnself.player.is_started = True           # 2......def check_click(self, data):buttons = pygame.mouse.get_pressed()if not buttons[0]:returnpos = pygame.mouse.get_pos()if self.btn_rect[0] < pos[0] < self.btn_rect[0]+self.btn_width and \self.btn_rect[1] < pos[1] < self.btn_rect[1]+self.btn_height:if self.player.is_host:self.start_conversation(data)else:if self.player.is_ready:self.get_unready()else:self.get_ready()def get_ready(self):self.player.is_ready = Truedef get_unready(self):self.player.is_ready = Falsedef start_conversation(self, data):     # 2unready_id_list = []for key, value in data.items():if not value.is_ready:unready_id_list.append(value.id)if not unready_id_list:self.player.is_started = Trueelse:font_size = 15font = pygame.font.Font("SimHei.ttf", font_size)for i, p_id in enumerate(unready_id_list):unready_text = font.render(f"玩家{p_id}还未准备", True, (150, 150, 150))self.window.blit(unready_text, (self.width/2-unready_text.get_width()/2, self.height/2 + (font_size*(i+1))))...

代码解释如下：

1. check_click()函数用来检测鼠标左键是否按在了按钮上。如果当前玩家是房主，那点击按钮后就会开始聊天；如果不是房主，那就会在已准备与未准备之间切换。

2. 这行代码的主要目的是为了让非房主玩家的is_started属性设置为True。如果没有这行代码的话，那开始聊天后，非房主玩家的准备按钮就会一直在draw_button()函数中被绘制。

3. 当房主点击开始聊天按钮后，先检查下其他玩家是否都已经准备了，如果是的话，则将Player对象的is_started属性设置为True，否则在窗口上显示还未准备的玩家id。

修改Server类：

class Server:...def handle_message(self, conn):while True:try:data = conn.recv(2048)if not data:print("未接收到数据，关闭连接")self.players_data.pop(str(id(conn)))conn.close()breakelse:data = pickle.loads(data)self.update_one_player_data(data)self.set_host(data["player"].room)self.check_is_started(data["player"].is_started, data["player"].room)               # 1conn.sendall(pickle.dumps(self.get_other_players_data(data["id"], data["player"].room)))except Exception as e:print(repr(e))break...def check_is_started(self, is_started, room):if is_started:for key, value in self.players_data.items():if value.room == room:self.players_data[key].is_started = True...

代码解释如下：

1. 当房主玩家的is_started值被设置为True之后，服务端会在check_is_started()函数中将其他所有玩家的is_started值都设置为True。这样所有玩家的窗口上就都会显示聊天内容了，而且开始/准备按钮也都不再被绘制。

运行结果如下：

4.4 完整代码下载地址：

链接:https://pan.baidu.com/s/1wqLUYwCqFhGrEbZIYvsyJg  

密码:3g7z